42 lines
1.1 KiB
Python
42 lines
1.1 KiB
Python
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
|