Update to v1.4

latest version/server/arcworld.py

@@ -0,0 +1,259 @@
+import json
+import sqlite3
+import os
+
+
+def int2b(x):
+    # int与布尔值转换
+    if x is None or x == 0:
+        return False
+    else:
+        return True
+
+
+def get_world_name(file_dir='./database/map'):
+    # 获取所有地图名称，返回列表
+    L = []
+    for root, dirs, files in os.walk(file_dir):
+        for file in files:
+            if os.path.splitext(file)[1] == '.json':
+                L.append(os.path.splitext(file)[0])
+    return L
+
+
+def get_world_info(map_id):
+    # 读取json文件内容，返回字典
+    f = open('./database/map/'+map_id+'.json', 'r')
+    world_info = json.load(f)
+    f.close()
+    return world_info
+
+
+def get_user_world_info(user_id, map_id):
+    # 读取json文件内容，加上用户信息，返回字典
+    info = get_world_info(map_id)
+    conn = sqlite3.connect('./database/arcaea_database.db')
+    c = conn.cursor()
+    c.execute('''select * from user_world where map_id = :a and user_id = :b''',
+              {'a': map_id, 'b': user_id})
+    x = c.fetchone()
+    if x:
+        info['curr_position'] = x[2]
+        info['curr_capture'] = x[3]
+        info['is_locked'] = int2b(x[4])
+    else:
+        c.execute('''insert into user_world values(:a,:b,0,0,0)''', {
+            'a': user_id, 'b': map_id})
+    conn.commit()
+    conn.close()
+    return info
+
+
+def get_current_map(user_id):
+    # 得到user的当前图，返回字符串
+    conn = sqlite3.connect('./database/arcaea_database.db')
+    c = conn.cursor()
+    c.execute('''select current_map from user where user_id = :a''',
+              {'a': user_id})
+    x = c.fetchone()
+    re = ''
+    if x:
+        re = x[0]
+
+    conn.commit()
+    conn.close()
+    return re
+
+
+def get_world_all(user_id):
+    # 读取所有地图信息并处理，返回字典列表
+    conn = sqlite3.connect('./database/arcaea_database.db')
+    c = conn.cursor()
+    worlds = get_world_name()
+    re = []
+    for map_id in worlds:
+        info = get_world_info(map_id)
+        steps = info['steps']
+        del info['steps']
+        rewards = []
+        for step in steps:
+            if 'items' in step:
+                rewards.append(
+                    {'items': step['items'], 'position': step['position']})
+        info['rewards'] = rewards
+        c.execute('''select * from user_world where map_id = :a and user_id = :b''',
+                  {'a': map_id, 'b': user_id})
+        x = c.fetchone()
+        if x:
+            info['curr_position'] = x[2]
+            info['curr_capture'] = x[3]
+            info['is_locked'] = int2b(x[4])
+        else:
+            c.execute('''insert into user_world values(:a,:b,0,0,0)''', {
+                      'a': user_id, 'b': map_id})
+
+        re.append(info)
+
+    conn.commit()
+    conn.close()
+    return re
+
+
+def get_user_world(user_id, map_id):
+    # 获取用户图信息，返回字典
+    conn = sqlite3.connect('./database/arcaea_database.db')
+    c = conn.cursor()
+    c.execute('''select * from user_world where map_id = :a and user_id = :b''',
+              {'a': map_id, 'b': user_id})
+    x = c.fetchone()
+    re = {
+        "user_id": user_id,
+        "curr_position": 0,
+        "curr_capture": 0,
+        "is_locked": False,
+        "map_id": map_id
+    }
+    if x:
+        re['curr_position'] = x[2]
+        re['curr_capture'] = x[3]
+        re['is_locked'] = int2b(x[4])
+    else:
+        c.execute('''insert into user_world values(:a,:b,0,0,0)''', {
+            'a': user_id, 'b': map_id})
+
+    conn.commit()
+    conn.close()
+    return re
+
+
+def change_user_current_map(user_id, map_id):
+    # 改变用户当前图
+    conn = sqlite3.connect('./database/arcaea_database.db')
+    c = conn.cursor()
+    c.execute('''update user set current_map = :a where user_id=:b''', {
+              'a': map_id, 'b': user_id})
+    conn.commit()
+    conn.close()
+    return None
+
+
+def play_world_song(user_id, args):
+    # 声明是世界模式的打歌，并且记录加成信息
+    conn = sqlite3.connect('./database/arcaea_database.db')
+    c = conn.cursor()
+
+    stamina_multiply = 1
+    fragment_multiply = 100
+    prog_boost_multiply = 0
+    if 'stamina_multiply' in args:
+        stamina_multiply = int(args['stamina_multiply'])
+    if 'fragment_multiply' in args:
+        fragment_multiply = int(args['fragment_multiply'])
+    if 'prog_boost_multiply' in args:
+        prog_boost_multiply = int(args['prog_boost_multiply'])
+
+    c.execute('''delete from world_songplay where user_id=:a and song_id=:b and difficulty=:c''', {
+              'a': user_id, 'b': args['song_id'], 'c': args['difficulty']})
+    c.execute('''insert into world_songplay values(:a,:b,:c,:d,:e,:f)''', {
+              'a': user_id, 'b': args['song_id'], 'c': args['difficulty'], 'd': stamina_multiply, 'e': fragment_multiply, 'f': prog_boost_multiply})
+
+    conn.commit()
+    conn.close()
+    return None
+
+
+def climb_step(user_id, map_id, step, prev_capture, prev_position):
+    # 爬梯子，返回奖励列表，台阶列表，当前的位置和坐标，图信息
+    info = get_world_info(map_id)
+    step_count = int(info['step_count'])
+
+    restrict_ids = [[]] * step_count
+    capture = [0] * step_count
+    reward_bundle = [""] * step_count  # 暂且不用
+    restrict_id = [""] * step_count
+    restrict_type = [""] * step_count
+    items = [[]] * step_count
+    step_type = [[]] * step_count
+    speed_limit_value = [0] * step_count
+    plus_stamina_value = [0] * step_count
+
+    for i in info['steps']:
+        capture[i['position']] = i['capture']
+        if 'items' in i:
+            items[i['position']] = i['items']
+        if 'restrict_id' in i:
+            restrict_id[i['position']] = i['restrict_id']
+        if 'restrict_ids' in i:
+            restrict_ids[i['position']] = i['restrict_ids']
+        if 'restrict_type' in i:
+            restrict_type[i['position']] = i['restrict_type']
+        if 'step_type' in i:
+            step_type[i['position']] = i['step_type']
+            if "speedlimit" in i['step_type']:
+                speed_limit_value[i['position']] = i['speed_limit_value']
+            if "plusstamina" in i['step_type']:
+                plus_stamina_value[i['position']] = i['plus_stamina_value']
+
+    if info['is_beyond']:  # beyond判断
+        dt = info['beyond_health'] - prev_capture
+        if dt >= step:
+            curr_capture = prev_capture + step
+        else:
+            curr_capture = info['beyond_health']
+        i = 0
+        t = prev_capture + step
+        while i < step_count and t > 0:
+            dt = capture[i]
+            if dt > t:
+                t = 0
+            else:
+                t -= dt
+                i += 1
+        if i >= step_count:
+            curr_position = step_count - 1
+        else:
+            curr_position = i
+
+    else:
+        i = prev_position
+        j = prev_capture
+        t = step
+        while t > 0 and i < step_count:
+            dt = capture[i] - j
+            if dt > t:
+                j += t
+                t = 0
+            else:
+                t -= dt
+                j = 0
+                i += 1
+        if i >= step_count:
+            curr_position = step_count - 1
+            curr_capture = 0
+        else:
+            curr_position = i
+            curr_capture = j
+
+    rewards = []
+    steps = []
+    for i in range(prev_position, curr_position+1):
+        if items[i]:
+            rewards.append({'position': i, 'items': items[i]})
+        x = {
+            "map_id": map_id,
+            "position": i,
+            "restrict_ids": restrict_ids[i],
+            "capture": capture[i],
+            "reward_bundle": reward_bundle[i],
+            "restrict_id": restrict_id[i],
+            "restrict_type": restrict_type[i]
+        }
+        if step_type[i]:
+            x['step_type'] = step_type[i]
+        if speed_limit_value[i]:
+            x['speed_limit_value'] = speed_limit_value[i]
+        if plus_stamina_value[i]:
+            x['plus_stamina_value'] = plus_stamina_value[i]
+        steps.append(x)
+
+    return rewards, steps, curr_position, curr_capture, info