Arcaea-server/latest version/server/setme.py

import sqlite3
import server.info


def b2int(x):
    # int与布尔值转换
    if x:
        return 1
    else:
        return 0


def int2b(x):
    # int与布尔值转换
    if x is None or x == 0:
        return False
    else:
        return True


def change_char(user_id, character_id, skill_sealed):
    # 角色改变，包括技能封印的改变，返回成功与否的布尔值
    conn = sqlite3.connect('./database/arcaea_database.db')
    c = conn.cursor()
    c.execute('''select is_uncapped, is_uncapped_override from user_char where user_id = :a and character_id = :b''',
              {'a': user_id, 'b': character_id})
    x = c.fetchone()
    if x is not None:
        if skill_sealed == 'false':
            skill_sealed = False
        else:
            skill_sealed = True
        c.execute('''update user set is_skill_sealed = :a, character_id = :b, is_char_uncapped = :c, is_char_uncapped_override = :d where user_id = :e''', {
                  'a': b2int(skill_sealed), 'b': character_id, 'c': x[0], 'd': x[1], 'e': user_id})

        conn.commit()
        conn.close()
        return True

    conn.commit()
    conn.close()
    return False


def change_char_uncap(user_id, character_id):
    # 角色觉醒改变，返回字典
    conn = sqlite3.connect('./database/arcaea_database.db')
    c = conn.cursor()
    c.execute('''select is_uncapped, is_uncapped_override from user_char where user_id = :a and character_id = :b''',
              {'a': user_id, 'b': character_id})
    x = c.fetchone()
    r = None
    if x is not None and x[0] == 1:
        c.execute('''update user set is_char_uncapped_override = :a where user_id = :b''', {
                  'a': b2int(x[1] == 0), 'b': user_id})
        c.execute('''update user_char set is_uncapped_override = :a where user_id = :b and character_id = :c''', {
                  'a': b2int(x[1] == 0), 'b': user_id, 'c': character_id})
        c.execute('''select * from user_char where user_id = :a and character_id = :b''',
                  {'a': user_id, 'b': character_id})
        y = c.fetchone()
        c.execute(
            '''select name from character where character_id = :x''', {'x': y[1]})
        z = c.fetchone()
        if z is not None:
            char_name = z[0]
        if y is not None:
            r = {
                "is_uncapped_override": int2b(y[14]),
                "is_uncapped": int2b(y[13]),
                "uncap_cores": [],
                "char_type": y[12],
                "skill_id_uncap": y[11],
                "skill_requires_uncap": int2b(y[10]),
                "skill_unlock_level": y[9],
                "skill_id": y[8],
                "overdrive": y[7],
                "prog": y[6],
                "frag": y[5],
                "level_exp": y[4],
                "exp": y[3],
                "level": y[2],
                "name": char_name,
                "character_id": y[1]
            }

    conn.commit()
    conn.close()
    return r


def arc_sys_set(user_id, value, set_arg):
    # 三个设置，PTT隐藏、体力满通知、最爱角色，无返回
    conn = sqlite3.connect('./database/arcaea_database.db')
    c = conn.cursor()
    if 'favorite_character' in set_arg:
        value = int(value)
        c.execute('''update user set favorite_character = :a where user_id = :b''', {
            'a': value, 'b': user_id})

    else:
        if value == 'false':
            value = False
        else:
            value = True

        if 'is_hide_rating' in set_arg:
            c.execute('''update user set is_hide_rating = :a where user_id = :b''', {
                      'a': b2int(value), 'b': user_id})
        if 'max_stamina_notification_enabled' in set_arg:
            c.execute('''update user set max_stamina_notification_enabled = :a where user_id = :b''', {
                      'a': b2int(value), 'b': user_id})

    conn.commit()
    conn.close()
    return None


def arc_add_friend(user_id, friend_id):
    # 加好友，返回好友列表，或者是错误码602、604
    if user_id == friend_id:
        return 604
    conn = sqlite3.connect('./database/arcaea_database.db')
    c = conn.cursor()
    c.execute('''select exists(select * from friend where user_id_me = :x and user_id_other = :y)''',
              {'x': user_id, 'y': friend_id})
    r = None
    if c.fetchone() == (0,):
        c.execute('''insert into friend values(:a, :b)''',
                  {'a': user_id, 'b': friend_id})
        r = server.info.get_user_friend(c, user_id)

    else:
        return 602

    conn.commit()
    conn.close()
    return r


def arc_delete_friend(user_id, friend_id):
    # 删好友，返回好友列表
    conn = sqlite3.connect('./database/arcaea_database.db')
    c = conn.cursor()
    c.execute('''select exists(select * from friend where user_id_me = :x and user_id_other = :y)''',
              {'x': user_id, 'y': friend_id})
    r = None
    if c.fetchone() == (1,):
        c.execute('''delete from friend where user_id_me = :x and user_id_other = :y''',
                  {'x': user_id, 'y': friend_id})

        r = server.info.get_user_friend(c, user_id)

    conn.commit()
    conn.close()
    return r