增加定价

chatgpt-next-web/OnSalesData.py

@@ -0,0 +1,5 @@
+class OnSalesData:
+    def __init__(self, currentPackage, packages):
+        self.currentPackage = currentPackage
+        self.packages = packages
+

chatgpt-next-web/pay_package.py

@@ -10,70 +10,31 @@ packages = [
     },
     {
         'id': '2',
-        'title': '无限制基础访问套餐',
+        'title': '无限制月套餐',
         'basic_chat_limit': -1,  # -1 表示无限次
         'advanced_chat_limit': 0,
-        'price': 60000,   # 以分计算
+        'price': 6000,   # 以分计算
         'expiration': 30 * 24 * 60 * 60  # 过期时间为30天（以秒为单位）
-
+    },
+    {
+        'id': '3',
+        'title': '无限制季套餐',
+        'basic_chat_limit': -1,  # -1 表示无限次
+        'advanced_chat_limit': 0,
+        'price': 16200,   # 以分计算
+        'expiration': 90 * 24 * 60 * 60  # 过期时间为30天（以秒为单位）
+    },
+    {
+        'id': '4',
+        'title': '无限制年套餐',
+        'basic_chat_limit': -1,  # -1 表示无限次
+        'advanced_chat_limit': 0,
+        'price': 57600,   # 以分计算
+        'expiration': 365 * 24 * 60 * 60  # 过期时间为30天（以秒为单位）
     }
 ]
 
 
-test_packages = [{
-    "id": 8,
-    "uuid": "3a07df99-e0d7-4855-835d-ff957e118946",
-    "title": "李振超级无敌套餐",
-    "subTitle": "",
-                "tokens": -1,
-                "chatCount": -1,
-                "advancedChatCount": -1,
-                "drawCount": -1,
-                "days": 99999,
-                "state": 10,
-                "calcTypeId": 2,
-                "calcType": "Daily",
-                "price": 9999.00,
-                "top": 0,
-                "createTime": "2023-07-03 22:54:22",
-                "updateTime": "2023-07-03 22:54:27"
-}, {
-    "id": 9,
-    "uuid": "b2cadc5e-0a18-4aef-99c5-8375f5867534",
-    "title": "X套餐",
-    "subTitle": "",
-                "tokens": 10000,
-                "chatCount": 0,
-                "advancedChatCount": 0,
-                "drawCount": 0,
-                "days": 30,
-                "state": 10,
-                "calcTypeId": 1,
-                "calcType": "Total",
-                "price": 10.00,
-                "top": 0,
-                "createTime": "2023-07-14 22:01:54",
-                "updateTime": "2023-07-14 22:02:07"
-}, {
-    "id": 10,
-    "uuid": "e1623644-caf0-43a2-a298-9894a68290a3",
-    "title": "小时套餐",
-    "subTitle": "小时套餐",
-                "tokens": 0,
-                "chatCount": 2,
-                "advancedChatCount": 0,
-                "drawCount": 0,
-                "days": 31,
-                "state": 10,
-                "calcTypeId": 4,
-                "calcType": "ThreeHourly",
-                "price": 0.01,
-                "top": 0,
-                "createTime": "2023-07-23 12:55:30",
-                "updateTime": "2023-07-23 12:55:33"
-}]
-
-
 # 根据套餐ID获取套餐信息
 def get_package_by_id(package_id):
     for package in packages:

chatgpt-next-web/service.py

@@ -13,6 +13,7 @@ from flask import Flask, request, jsonify, Request, Response
 from redis import Redis
 from utils import nowtime
 import pay_package
+from OnSalesData import OnSalesData
 from ApiResponse import ApiResponse
 from flask_cors import CORS
 from mail import MyEmail
@@ -584,10 +585,16 @@ def packageOnSales():
 
     if not user_id:
         return jsonify({'code': 400, 'message': 'User not found'})
+    
+    current_package = get_user_package(user_id)
+    onSalesData = {
+        'currentPackage': int(current_package.get(b'id', 0)),
+        'packages': pay_package.packages
+    }
     response = ApiResponse(
         code=0,
         message='Login successful',
-        data=pay_package.packages)
+        data=onSalesData)
     return jsonify(response.to_json())