添加每天领取免费次数

chatgpt-next-web/service.py

@@ -9,6 +9,7 @@ import requests
 import os
 from flask import Flask, request, jsonify, Request, Response
 from redis import Redis
+from utils import nowtime
 
 SECERT_KEY = "8U2LL1"
 MY_OPENAI_API_KEY = os.environ.get('MY_OPENAI_API_KEY')
@@ -284,6 +285,29 @@ def proxy_chat_completions():
         return result, response.status_code
 
 
+# 每天领取免费次数
+@app.route('/pickUpFreeChatCount', methods=['POST'])
+def handle_pick_up_free_chat_count():
+    token = parse_token(request)
+
+    # 验证令牌
+    if not validate_token(token):
+        return jsonify({'code': 401, 'message': 'Invalid token'})
+
+    user_id = get_user_id_from_token(token)
+
+    if not user_id:
+        return jsonify({'code': 400, 'message': 'User not found'})
+
+    # 领取赠送的聊天次数
+    success = pick_up_free_chat_count(user_id)
+
+    if success:
+        return {'message': '领取成功'}
+    else:
+        return {'message': '您今天已经领取了'}
+
+
 def parse_token(request: Request):
     token_with_bearer = request.headers.get('Authorization')
 
@@ -301,7 +325,7 @@ def generate_token(user_id, username):
     payload = {
         'user_id': user_id,
         'username': username,
-        'exp': datetime.datetime.utcnow() + datetime.timedelta(hours=1)
+        'exp': nowtime() + datetime.timedelta(days=30)
     }
 
     # 在这里，您可以使用您的密钥（secret key）来签署令牌
@@ -320,7 +344,7 @@ def validate_token(token):
         print(payload)
 
         # 检查令牌的过期时间
-        if 'exp' in payload and datetime.datetime.utcnow() > datetime.datetime.fromtimestamp(payload['exp']):
+        if 'exp' in payload and nowtime() > datetime.datetime.fromtimestamp(payload['exp']):
             return False
 
         return True
@@ -466,5 +490,61 @@ def update_password(email, new_password):
     return False
 
 
+def get_user_free_data(user_id):
+    user_free_key = f'user:{user_id}:free'
+    free_data = redis.hgetall(user_free_key)
+    return free_data
+
+
+def initialize_user_free_data(user_id):
+    free_data = {
+        b'basic_chat_count': 0,
+        b'last_gift_time': str(datetime.date.today()),
+        b'has_pick_up_free': b'false'  # 用户是否领取了赠送的聊天次数
+    }
+    user_free_name = f'user:{user_id}:free'
+    redis.hset(user_free_name,
+               b'basic_chat_count',
+               free_data.get(b'basic_chat_count', 0))
+    redis.hset(user_free_name,
+               b'last_gift_time',
+               free_data.get(b'last_gift_time'))
+    redis.hset(user_free_name,
+               b'has_pick_up_free',
+               free_data.get(b'has_pick_up_free', b'false'))
+    return free_data
+
+
+# 每天领取免费次数
+def pick_up_free_chat_count(user_id):
+    free_data = get_user_free_data(user_id)
+    if not free_data:
+        free_data = initialize_user_free_data(user_id)
+    # 获取用户的聊天次数和最后赠送时间
+    basic_chat_count = free_data.get(b'basic_chat_count', 0)
+    last_gift_time = free_data.get(
+        b'last_gift_time',
+        str(datetime.date.today()))
+    has_pick_up_free = (
+        free_data.get(b'has_pick_up_free', b'false')
+        .decode()
+        .lower() == 'true')
+
+    print('free_data3', basic_chat_count, last_gift_time, has_pick_up_free)
+
+    # 检查用户是否已经领取过赠送的聊天次数今天
+    if has_pick_up_free and last_gift_time.decode() == str(datetime.date.today()):
+        return False
+
+    basic_free_count = 5
+    # 更新用户聊天数据和领取状态
+    user_free_name = f'user:{user_id}:free'
+    redis.hset(user_free_name, b'basic_chat_count', basic_free_count)
+    redis.hset(user_free_name, b'last_gift_time', str(datetime.date.today()))
+    redis.hset(user_free_name, b'has_pick_up_free', 'true')
+
+    return True
+
+
 if __name__ == '__main__':
     app.run(debug=True)

chatgpt-next-web/utils.py

@@ -0,0 +1,5 @@
+import datetime
+
+
+def nowtime():
+    return datetime.datetime.utcnow() + datetime.timedelta(hours=8)