update
This commit is contained in:
49
app/main.py
49
app/main.py
@@ -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")
|
||||
|
||||
Reference in New Issue
Block a user