81 lines
2.3 KiB
Python
81 lines
2.3 KiB
Python
from sqlalchemy.orm import Session
|
|
|
|
# from . import models, schemas
|
|
import models, schemas
|
|
import json
|
|
import time
|
|
|
|
|
|
# 通过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()
|
|
|
|
|
|
# 通过openId岔村用户
|
|
def get_user_by_h5openid(db: Session, h5openid: str):
|
|
return db.query(
|
|
models.User).filter(models.User.h5openid == h5openid).first()
|
|
|
|
|
|
# 通过wxid岔村用户
|
|
def get_user_by_wxid(db: Session, wxid: str):
|
|
return db.query(models.User).filter(models.User.wxid == wxid).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
|
|
# **user
|
|
def create_user(db: Session, user: schemas.User):
|
|
# def create_user(db: Session, user):
|
|
db_user = models.User(wxid=user.wxid,
|
|
openId=user.openId,
|
|
h5openid=user.h5openid,
|
|
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,
|
|
bighead=user.bighead,
|
|
labels=user.labels,
|
|
source=user.source,
|
|
v1=user.v1)
|
|
|
|
db.add(db_user)
|
|
db.commit()
|
|
db.refresh(db_user)
|
|
return db_user
|
|
|
|
|
|
# 更新用户字段
|
|
def update_user_field(db: Session, wxid: str, openId=None, h5openid=None):
|
|
user = get_user_by_wxid(db, wxid)
|
|
if user:
|
|
# 更新字段值
|
|
# 存储小程序openid
|
|
if openId:
|
|
user.openId = openId
|
|
if h5openid:
|
|
user.h5openid = h5openid
|
|
|
|
# 提交更改
|
|
db.commit()
|
|
db.refresh(user)
|
|
print('用户数据更新')
|
|
return user
|
|
print('用户数据查询失败')
|
|
return False
|