Update to v2.1

latest version/server/arcpurchase.py

@@ -1,4 +1,4 @@
-import sqlite3
+from server.sql import Connect
 import time
 import json
 
@@ -41,87 +41,91 @@ def get_item(c, type='pack'):
 
 def get_single_purchase():
     # main里面没开数据库，这里写一下代替
-    conn = sqlite3.connect('./database/arcaea_database.db')
-    c = conn.cursor()
-    re = get_item(c, type='single')
-    conn.commit()
-    conn.close()
+    re = []
+    with Connect() as c:
+        re = get_item(c, type='single')
+
     return re
 
 
+def buy_item(c, user_id, price):
+    # 购买接口，返回成功与否标识符和剩余源点数
+    c.execute('''select ticket from user where user_id = :a''',
+              {'a': user_id})
+    ticket = c.fetchone()
+    if ticket:
+        ticket = ticket[0]
+    else:
+        ticket = 0
+
+    if ticket < price:
+        return False, ticket
+
+    c.execute('''update user set ticket = :b where user_id = :a''',
+              {'a': user_id, 'b': ticket-price})
+
+    return True, ticket - price
+
+
 def buy_pack(user_id, pack_id):
     # 曲包购买，返回字典
-    conn = sqlite3.connect('./database/arcaea_database.db')
-    c = conn.cursor()
-    c.execute('''select price from item where item_id = :a''', {'a': pack_id})
-    price = c.fetchone()
-    if price:
-        price = price[0]
-    else:
-        price = 0
+    re = {"success": False}
+    with Connect() as c:
+        c.execute('''select price from item where item_id = :a''',
+                  {'a': pack_id})
+        price = c.fetchone()
+        if price:
+            price = price[0]
+        else:
+            price = 0
 
-    c.execute('''select ticket from user where user_id = :a''', {'a': user_id})
-    ticket = c.fetchone()
-    if ticket:
-        ticket = ticket[0]
-    else:
-        ticket = 0
+        flag, ticket = buy_item(c, user_id, price)
 
-    if ticket < price:
-        conn.commit()
-        conn.close()
-        return {
-            "success": False
-        }
+        if flag:
+            c.execute('''insert into user_item values(:a,:b,'pack')''',
+                      {'a': user_id, 'b': pack_id})
 
-    c.execute('''update user set ticket = :b where user_id = :a''',
-              {'a': user_id, 'b': ticket-price})
-    c.execute('''insert into user_item values(:a,:b,'pack')''',
-              {'a': user_id, 'b': pack_id})
+            re = {"success": True}
 
-    conn.commit()
-    conn.close()
-    return {
-        "success": True
-    }
+    return re
 
 
 def buy_single(user_id, single_id):
     # 单曲购买，返回字典
-    conn = sqlite3.connect('./database/arcaea_database.db')
-    c = conn.cursor()
-    c.execute('''select price from item where item_id = :a''',
-              {'a': single_id})
-    price = c.fetchone()
-    if price:
-        price = price[0]
+    re = {"success": False}
+    with Connect() as c:
+        c.execute('''select price from item where item_id = :a''',
+                  {'a': single_id})
+        price = c.fetchone()
+        if price:
+            price = price[0]
+        else:
+            price = 0
+
+        flag, ticket = buy_item(c, user_id, price)
+
+        if flag:
+            c.execute('''insert into user_item values(:a,:b,'single')''',
+                      {'a': user_id, 'b': single_id})
+            re = {"success": True}
+
+    return re
+
+
+def get_prog_boost(user_id):
+    # 世界模式源韵强化，扣50源点，返回剩余源点数
+
+    ticket = -1
+    with Connect() as c:
+        flag, ticket = buy_item(c, user_id, 50)
+
+        if flag:
+            c.execute('''update user set prog_boost = 1 where user_id = :a''', {
+                      'a': user_id})
+    if ticket >= 0:
+        return ticket, None
     else:
-        price = 0
-
-    c.execute('''select ticket from user where user_id = :a''', {'a': user_id})
-    ticket = c.fetchone()
-    if ticket:
-        ticket = ticket[0]
-    else:
-        ticket = 0
-
-    if ticket < price:
-        conn.commit()
-        conn.close()
-        return {
-            "success": False
-        }
-
-    c.execute('''update user set ticket = :b where user_id = :a''',
-              {'a': user_id, 'b': ticket-price})
-    c.execute('''insert into user_item values(:a,:b,'single')''',
-              {'a': user_id, 'b': single_id})
-
-    conn.commit()
-    conn.close()
-    return {
-        "success": True
-    }
+        return 0, 108
 
 
 def get_user_present(c, user_id):
@@ -146,82 +150,70 @@ def get_user_present(c, user_id):
 def claim_user_present(user_id, present_id):
     # 确认并删除用户奖励，返回成功与否的布尔值
     flag = False
-    conn = sqlite3.connect('./database/arcaea_database.db')
-    c = conn.cursor()
-
-    c.execute('''select exists(select * from user_present where user_id=:a and present_id=:b)''',
-              {'a': user_id, 'b': present_id})
-    if c.fetchone() == (1,):
-        flag = True
-        c.execute('''delete from user_present where user_id=:a and present_id=:b''',
+    with Connect() as c:
+        c.execute('''select exists(select * from user_present where user_id=:a and present_id=:b)''',
                   {'a': user_id, 'b': present_id})
-        c.execute('''select * from present where present_id=:b''',
-                  {'b': present_id})
-        x = c.fetchone()
-        now = int(time.time() * 1000)
-        if now <= int(x[1]):
-            # 处理memory
-            items = json.loads(x[2])
-            for i in items:
-                if i['id'] == 'memory':
-                    c.execute('''select ticket from user where user_id=:a''', {
-                              'a': user_id})
-                    ticket = int(c.fetchone()[0])
-                    ticket += int(i['amount'])
-                    c.execute('''update user set ticket=:b where user_id=:a''', {
-                              'a': user_id, 'b': ticket})
+        if c.fetchone() == (1,):
+            c.execute('''delete from user_present where user_id=:a and present_id=:b''',
+                      {'a': user_id, 'b': present_id})
+            c.execute('''select * from present where present_id=:b''',
+                      {'b': present_id})
+            x = c.fetchone()
+            now = int(time.time() * 1000)
+            if now <= int(x[1]):
+                # 处理memory
+                items = json.loads(x[2])
+                for i in items:
+                    if i['id'] == 'memory':
+                        c.execute('''select ticket from user where user_id=:a''', {
+                            'a': user_id})
+                        ticket = int(c.fetchone()[0])
+                        ticket += int(i['amount'])
+                        c.execute('''update user set ticket=:b where user_id=:a''', {
+                            'a': user_id, 'b': ticket})
+                flag = True
+            else:
+                # 过期
+                flag = False
 
-        else:
-            # 过期
-            flag = False
-
-    conn.commit()
-    conn.close()
     return flag
 
 
 def claim_user_redeem(user_id, code):
     # 处理兑换码，返回碎片数量和错误码
     fragment = 0
-    conn = sqlite3.connect('./database/arcaea_database.db')
-    c = conn.cursor()
-    c.execute('''select * from redeem where code=:a''', {'a': code})
-    x = c.fetchone()
-    if not x:
-        conn.commit()
-        conn.close()
-        return 0, 504
+    error_code = 108
+    with Connect() as c:
+        c.execute('''select * from redeem where code=:a''', {'a': code})
+        x = c.fetchone()
+        if not x:
+            return 0, 504
 
-    if x[2] == 0:  # 一次性
-        c.execute(
-            '''select exists(select * from user_redeem where code=:a)''', {'a': code})
-        if c.fetchone() == (1,):
-            conn.commit()
-            conn.close()
-            return 0, 505
-    elif x[2] == 1:  # 每个玩家一次
-        c.execute('''select exists(select * from user_redeem where code=:a and user_id=:b)''',
+        if x[2] == 0:  # 一次性
+            c.execute(
+                '''select exists(select * from user_redeem where code=:a)''', {'a': code})
+            if c.fetchone() == (1,):
+                return 0, 505
+        elif x[2] == 1:  # 每个玩家一次
+            c.execute('''select exists(select * from user_redeem where code=:a and user_id=:b)''',
+                      {'a': code, 'b': user_id})
+            if c.fetchone() == (1,):
+                return 0, 506
+
+        c.execute('''insert into user_redeem values(:b,:a)''',
                   {'a': code, 'b': user_id})
-        if c.fetchone() == (1,):
-            conn.commit()
-            conn.close()
-            return 0, 506
 
-    c.execute('''insert into user_redeem values(:b,:a)''',
-              {'a': code, 'b': user_id})
+        items = json.loads(x[1])
+        for i in items:
+            if i['type'] == 'fragment':
+                fragment = i['amount']
+            if i['type'] == 'memory':
+                c.execute('''select ticket from user where user_id=:a''', {
+                    'a': user_id})
+                ticket = int(c.fetchone()[0])
+                ticket += int(i['amount'])
+                c.execute('''update user set ticket=:b where user_id=:a''', {
+                    'a': user_id, 'b': ticket})
+        error_code = None
 
-    items = json.loads(x[1])
-    for i in items:
-        if i['type'] == 'fragment':
-            fragment = i['amount']
-        if i['type'] == 'memory':
-            c.execute('''select ticket from user where user_id=:a''', {
-                'a': user_id})
-            ticket = int(c.fetchone()[0])
-            ticket += int(i['amount'])
-            c.execute('''update user set ticket=:b where user_id=:a''', {
-                'a': user_id, 'b': ticket})
-
-    conn.commit()
-    conn.close()
-    return fragment, None
+    return fragment, error_code