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