daily/Crud/UserCrud.py

22 lines
539 B
Python

from sqlalchemy.orm import Session
from Models.UserModel import User
from Schemas import UserSchemas
def get_user_info(db: Session, openid: str):
user = db.query(User).filter(openid=openid).first()
return user
def create_user(db: Session, openid: str):
user = User(openid=openid)
db.commit(user)
db.refresh(user)
return user
def change_user_info(db: Session, openid: str, user_info_change: UserSchemas.UserInfoChange):
db.query(User).filter(openid=openid).update(user_info_change.dict())
db.commit()