update

app/main.py

@@ -15,6 +15,8 @@ from fastapi.staticfiles import StaticFiles
 # from app import tgMessage  # type: ignore #发送tg消息
 from datetime import datetime
 import requests
+from tentcentSMS import sendSms
+import random
 
 # 初始化数据库
 models.Base.metadata.create_all(bind=engine)
@@ -50,6 +52,7 @@ def get_db():
 # def root():
 #     return {"message": "Hello World"}
 
+
 def sengTgmesage(message):
     url = "http://192.168.31.38:8521/send_message"
     data = {"message": message}
@@ -138,7 +141,6 @@ def generateqr(url="http://nomad.hackrobot.cn",
     if wxid:
         url += f"?wxid={wxid}"
 
-
     # 创建二维码
     qr = qrcode.QRCode(version=1, box_size=5, border=2)  # 调整二维码大小
     qr.add_data(url)
@@ -220,13 +222,15 @@ def add_user(nickname: Optional[Any] = None,
     elif keyword == '活动':
         print("keyword==活动")
         response = f"活动正在筹备中"
-        combined_image_path = generateqr(url="http://nomad.hackrobot.cn",
+        combined_image_path = generateqr(
+            url="http://nomad.hackrobot.cn",
             wxid=wxid,
             group_image_path="images/meetupqr.jpg")
         return FileResponse(combined_image_path)
     elif keyword == 'qrcode':
         print("keyword==qrcode")
-        combined_image_path = generateqr(url="http://nomad.hackrobot.cn",
+        combined_image_path = generateqr(
+            url="http://nomad.hackrobot.cn",
             #  wxid='qrcode',
             group_image_path="images/nomadqr.jpg")
         return FileResponse(combined_image_path)
@@ -276,3 +280,37 @@ def update_user(identifier: str,
     }
 
 
+# 获取手机号=>生成验证码
+@app.post("/get_verification_code/")
+def get_verification_code(wxid: str,
+                          phone_number: str,
+                          db: Session = Depends(get_db)):
+    verification_code = '{:04d}'.format(random.randint(0, 9999))
+    user = db.query(models.User).filter(models.User.wxid == wxid).first()
+    if not user:
+        raise HTTPException(status_code=404, detail="User not found")
+
+    user.phone_number = phone_number
+    user.verification_code = verification_code
+    db.commit()
+
+    return {
+        "message": "Verification code sent",
+        "verification_code": verification_code
+    }
+
+
+# 验证码校验
+@app.post("/verify_code/")
+def verify_code(phone_number: str, code: str, db: Session = Depends(get_db)):
+    user = db.query(
+        models.User).filter(models.User.phone_number == phone_number).first()
+
+    if not user:
+        raise HTTPException(status_code=404, detail="User not found")
+
+    if user.verification_code == code:
+        return {"message": "Verification successful"}
+    else:
+        raise HTTPException(status_code=400,
+                            detail="Invalid verification code")

app/models.py

@@ -20,6 +20,8 @@ class User(Base):
     ebook = Column(Boolean, default=False)  # 新增电子书字段
     activity_status = Column(String, default="0")  # 新增活动报名字段
     vip_member = Column(Boolean, default=False)  # 新增VIP会员字段
+    verification_code = Column(String, index=True)  # 新增验证码字段
+    phone_number = Column(String, index=True, unique=True)  # 新增手机号字段
 
 
 def sengTgmesage(message):

app/schemas.py

@@ -12,6 +12,8 @@ class UserBase(BaseModel):
     ebook: Optional[bool] = None  # 新增电子书字段
     activity_status: Optional[str] = None  # 新增活动报名字段
     vip_member: Optional[bool] = None  # 新增VIP会员字段
+    verification_code: Optional[bool] = None  #新增验证码字段
+
 
 class UserCreate(UserBase):
     pass

app/tentcentSMS.py

@@ -16,7 +16,7 @@ import random
 
 
 # phoneNum="+8613243889782"
-def sendSms(phoneNum):
+def sendSms(phoneNum,verification_code=None):
     try:
         # 必要步骤：
         # 实例化一个认证对象，入参需要传入腾讯云账户密钥对secretId，secretKey。
@@ -77,7 +77,7 @@ def sendSms(phoneNum):
 
         # 模板参数: 模板参数的个数需要与 TemplateId 对应模板的变量个数保持一致，，若无模板参数，则设置为空
         # 随机4位数字
-        verification_code = '{:04d}'.format(random.randint(0, 9999))
+        # verification_code = '{:04d}'.format(random.randint(0, 9999))
         req.TemplateParamSet = [verification_code]
         # 下发手机号码，采用 E.164 标准，+[国家或地区码][手机号]
         # 示例如：+8613711112222， 其中前面有一个+号 ，86为国家码，13711112222为手机号，最多不要超过200个手机号