Some formatting and cleanup

This commit is contained in:
lvrossem 2023-03-28 15:12:11 -06:00
parent 6e39852f49
commit d1c1738917
3 changed files with 4 additions and 9 deletions

View File

@ -1,8 +1,7 @@
from sqlalchemy.orm import Session
import models
import schemas
from models import CourseProgress, HighScore, User
from schemas.users import UserCreate
def get_user(db: Session, user_id: int):
@ -17,8 +16,8 @@ def get_users(db: Session):
return db.query(User).all()
def create_user(db: Session, user: schemas.users.UserCreate):
db_user = models.User(username=user.username, hashed_password=user.password)
def create_user(db: Session, user: UserCreate):
db_user = User(username=user.username, hashed_password=user.password)
db.add(db_user)
db.commit()
db.refresh(db_user)

View File

@ -2,7 +2,6 @@ from sqlalchemy import create_engine
from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy.orm import sessionmaker
# SQLALCHEMY_DATABASE_URL = "sqlite:///./sql_app.db"
SQLALCHEMY_DATABASE_URL = "postgresql://admin:WeSign123!@localhost/wesigndev"
engine = create_engine(SQLALCHEMY_DATABASE_URL)

View File

@ -16,16 +16,13 @@ Base.metadata.create_all(bind=engine)
def get_db():
db = SessionLocal()
try:
print("Yield")
yield db
finally:
print("Close")
db.close()
@app.get("/")
async def root():
print("Hello world")
return {"message": "Hello world!"}
@ -33,7 +30,7 @@ async def root():
async def create_user(user: users.UserCreate, db: Session = Depends(get_db)):
db_user = crud.get_user_by_username(db, username=user.username)
if db_user:
raise HTTPException(status_code=400, detail="Email already registered")
raise HTTPException(status_code=400, detail="Username already registered")
return crud.create_user(db=db, user=user)