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