Commit a079804c authored by xianyang's avatar xianyang

优化db

parent 2eaa91cb
import pymysql import pymysql
pymysql.install_as_MySQLdb() pymysql.install_as_MySQLdb()
from core.storage.db import SessionLocal
def get_db():
db = SessionLocal()
try:
yield db
finally:
db.close()
from app import get_db
from typing import List from typing import List
from fastapi import Depends, APIRouter, HTTPException, FastAPI
from fastapi.security import OAuth2PasswordRequestForm
from sqlalchemy.orm import Session from sqlalchemy.orm import Session
import models
from app.api.users import crud, schemas from app.api.users import crud, schemas
from core.storage.db import SessionLocal from fastapi import Depends, APIRouter, HTTPException
from libs.result_format import HttpResultResponse, HttpMessage from libs.result_format import HttpResultResponse, HttpMessage
router = APIRouter() router = APIRouter()
def get_db():
db = SessionLocal()
try:
yield db
finally:
db.close()
# @router.post("/", response_model=schemas.User)
# def create_user(user: schemas.UserCreate, db: Session = Depends(get_db)):
# db_user = crud.get_user_by_email(db, email=user.email)
# if db_user:
# raise HTTPException(status_code=400, detail="Email already registered")
# return crud.create_user(db=db, user=user)
@router.get("/", response_model=List[schemas.User]) @router.get("/", response_model=List[schemas.User])
def read_users(skip: int = 0, limit: int = 100, db: Session = Depends(get_db)): def read_users(skip: int = 0, limit: int = 100, db: Session = Depends(get_db)):
users = crud.get_users(db, skip=skip, limit=limit) users = crud.get_users(db, skip=skip, limit=limit)
...@@ -57,7 +39,7 @@ def login(): ...@@ -57,7 +39,7 @@ def login():
# access_token_expires = timedelta(minutes=config.ACCESS_TOKEN_EXPIRE_MINUTES) # access_token_expires = timedelta(minutes=config.ACCESS_TOKEN_EXPIRE_MINUTES)
return { return {
"token_type": "bearer", "token_type": "bearer"
} }
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment