This commit is contained in:
hackrobot
2024-07-16 07:53:56 +08:00
parent 83edc822a2
commit 8ef8418ff4

View File

@@ -1,5 +1,5 @@
from fastapi import FastAPI, HTTPException, Depends
from fastapi.responses import JSONResponse
from fastapi.responses import JSONResponse, PlainTextResponse
from sqlalchemy.orm import Session
from typing import List, Optional, Any
@@ -23,6 +23,8 @@ app.add_middleware(
allow_methods=["*"], # 允许所有方法
allow_headers=["*"], # 允许所有请求头
)
# 依赖注入
def get_db():
db = SessionLocal()
@@ -32,46 +34,49 @@ def get_db():
db.close()
@app.get("/")
def root():
return {"message": "Hello World"}
@app.get("/hello")
def hello():
return {"message": "Hello World"}
# @app.get("/add", response_model=schemas.User)
@app.get("/add")
def add_user(
nickname: Optional[Any] = None,
def add_user(nickname: Optional[Any] = None,
remark: Optional[Any] = None,
wechat_number: Optional[Any] = None,
wxid: Optional[Any] = None,
gender: Optional[Any] = None,
friendaddtime: Optional[Any] = None,
openid: Optional[Any] = None,
db: Session = Depends(get_db)
):
crud.create_user(db=db, user=schemas.UserCreate(
nickname=nickname, remark=remark, wechat_number=wechat_number,
wxid=wxid, gender=gender, friendaddtime=friendaddtime,
db: Session = Depends(get_db)):
crud.create_user(db=db,
user=schemas.UserCreate(nickname=nickname,
remark=remark,
wechat_number=wechat_number,
wxid=wxid,
gender=gender,
friendaddtime=friendaddtime,
openid=openid))
response=f'''{nickname}你好,\n我是机器人助理\n一名开发者,副业:跨境电商,#公众号:异度世界'''
# content = {
# "answer": response,
# "answer_type": "text",
# }
headers = {"Content-Type": "application/json"}
return JSONResponse(content=response, headers=headers)
# return f'''{nickname}你好,\n我是机器人助理\n一名开发者,副业:跨境电商,#公众号:异度世界'''
# response=f'''{nickname}你好,\n我是机器人助理\n一名开发者,副业:跨境电商,#公众号:异度世界'''
# headers = {"Content-Type": "application/json"}
# return JSONResponse(content=response, headers=headers)
response = f"{nickname}你好,\n我是机器人助理\n一名开发者,副业:跨境电商,#公众号:异度世界"
return PlainTextResponse(content=response)
# return '{nickname}你好,\n我是机器人助理\n一名开发者,副业:跨境电商,#公众号:异度世界'
@app.get("/users/")
def read_users(skip: int = 0, limit: int = 10, db: Session = Depends(get_db)):
return crud.get_users(db=db, skip=skip, limit=limit)
@app.get("/user/{identifier}")
def read_user(identifier: str, db: Session = Depends(get_db)):
user = crud.get_user_by_openid_or_wxid(db, identifier=identifier)
@@ -79,6 +84,7 @@ def read_user(identifier: str, db: Session = Depends(get_db)):
raise HTTPException(status_code=404, detail="User not found")
return user
@app.delete("/user/{user_id}")
def delete_user(user_id: int, db: Session = Depends(get_db)):
user = crud.get_user(db=db, user_id=user_id)
@@ -86,8 +92,11 @@ def delete_user(user_id: int, db: Session = Depends(get_db)):
raise HTTPException(status_code=404, detail="User not found")
return crud.delete_user(db=db, user=user)
@app.put("/user/{identifier}")
def update_user(identifier: str, user_update: schemas.UserUpdate, db: Session = Depends(get_db)):
def update_user(identifier: str,
user_update: schemas.UserUpdate,
db: Session = Depends(get_db)):
user = crud.get_user_by_openid_or_wxid(db, identifier=identifier)
if user is None:
raise HTTPException(status_code=404, detail="User not found")