feat: fastapi

This commit is contained in:
hackrobot
2024-03-20 18:30:39 +08:00
parent 8c79bde279
commit 1bbe70806c
12 changed files with 157 additions and 0 deletions

41
crud.py Normal file
View File

@@ -0,0 +1,41 @@
from sqlalchemy.orm import Session
# from . import models, schemas
import models, schemas
# 通过id获取用户
def get_user(db: Session, user_id: int):
return db.query(models.User).filter(models.User.id == user_id).first()
# 通过openId岔村用户
def get_user_by_openId(db: Session, openId: str):
return db.query(models.User).filter(models.User.openId == openId).first()
# 分页查询-用户all
def get_users(db: Session, skip: int = 0, limit: int = 100):
return db.query(models.User).offset(skip).limit(limit).all()
# 创建用户
# user:schemas.UserCreate
def create_user(db: Session, user:schemas.UserCreate):
db_user = models.User(
wxid=user.wxid,
openId=user.openId,
sex=user.sex,
country=user.country,
province=user.province,
sourceStr=user.sourceStr,
usertype=user.usertype,
nickname=user.nickname,
wxaccount=user.wxaccount,
remark=user.remark,
smallhead=user.smallhead,
)
db.add(db_user)
db.commit()
db.refresh(db_user)
return db_user