26 lines
744 B
Python
26 lines
744 B
Python
# _*_ coding: utf-8 _*_
|
|
# @Time :2022/5/29 11:29
|
|
# @Email :508737091@qq.com
|
|
# @Author :qiangyanwen
|
|
# @File :user_repository.py
|
|
from sqlalchemy import or_
|
|
from sqlalchemy.orm import Session
|
|
|
|
from config.db_session import db_add
|
|
from config.factory import AutomationResponse
|
|
from entity.user_entity import UserFrom
|
|
from models.model import User
|
|
|
|
|
|
def check_user_email(db: Session, user: UserFrom) -> bool:
|
|
user = db.query(User).filter(or_(User.username == user.username, user.email == user.email)).first()
|
|
if user:
|
|
return True
|
|
return False
|
|
|
|
|
|
def register_user(db: Session, user: UserFrom)->dict:
|
|
us_er = User(**user.dict())
|
|
db_add(db, us_er)
|
|
return AutomationResponse.model_to_dict(us_er, "password")
|