update

crud.py

@@ -5,6 +5,7 @@ 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()
@@ -14,9 +15,11 @@ def get_user(db: Session, user_id: int):
 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()
+    return db.query(
+        models.User).filter(models.User.h5openid == h5openid).first()
 
 
 # 通过wxid岔村用户
@@ -31,28 +34,44 @@ def get_users(db: Session, skip: int = 0, limit: int = 100):
 
 # 创建用户
 # 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,
-        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
-    )
-    
+# **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,
+                          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, user_id: str, openId=None, h5openid=None):
+    user = get_user(db, user_id)
+    if user:
+        # 更新字段值
+        # 存储小程序openid
+        if openId:
+            user.openId = openId
+        if h5openid:
+            user.h5openid = h5openid
+
+        # 提交更改
+        db.commit()
+        db.refresh(user)
+        return user
+    return False