Python DatabaseManager.update_table方法代码示例

# 需要导入模块: from database import DatabaseManager [as 别名]
# 或者: from database.DatabaseManager import update_table [as 别名]
def add_rl_save_state_data(uuid):
    rl_starts_b64 = request.form.get('rl_starts')
    rl_lengths_b64 = request.form.get('rl_lengths')
    rl_total_length = int(request.form.get('rl_total_length'))
    rl_lengths_length = int(request.form.get('rl_lengths_length'))
    rl_starts_length = int(request.form.get('rl_starts_length'))

    rl_starts_b_array = bytearray(base64.b64decode(rl_starts_b64))
    rl_lengths_b_array = bytearray(base64.b64decode(rl_lengths_b64))

    if len(rl_starts_b_array) != rl_starts_length:
        rl_starts_b_array.extend([0 for _ in range(len(rl_starts_b_array), rl_starts_length)])

    if len(rl_lengths_b_array) != rl_lengths_length:
        rl_lengths_b_array.extend([0 for _ in range(len(rl_lengths_b_array), rl_lengths_length)])

    rl_start_hash, file_name = save_byte_array_to_store(rl_starts_b_array, file_name=uuid)
    rl_lengths_hash, file_name = save_byte_array_to_store(rl_lengths_b_array, file_name=uuid)

    dbm.update_table(dbm.GAME_SAVE_TABLE,
                     ['rl_starts_data','rl_lengths_data','rl_total_length','compressed'],
                     [rl_start_hash, rl_lengths_hash, rl_total_length, True],
                     ['uuid'], [uuid])

    return jsonify({'record': dbm.retrieve_save_state(uuid=uuid)[0]})

# 需要导入模块: from database import DatabaseManager [as 别名]
# 或者: from database.DatabaseManager import update_table [as 别名]
def update_save_state(uuid):
    update_fields = json.loads(request.form.get('update_fields'))

    #   if linking to a performance, do that and then throw away since GAME_SAVE_TABLE doesn't refer to performance
    if 'performance_uuid' in update_fields:
        performance_uuid = update_fields['performance_uuid']
        performance_time_index = update_fields['performance_time_index']
        action = update_fields['action']
        dbm.link_save_state_to_performance(uuid, performance_uuid, performance_time_index, action)
        del update_fields['performance_uuid']
        del update_fields['performance_time_index']
        del update_fields['action']

    #   make sure that there are still fields to update
    if len(update_fields.keys()) > 0:
        dbm.update_table(dbm.GAME_SAVE_TABLE, update_fields.keys(),update_fields.values(), ['uuid'], [uuid])
    return jsonify(dbm.retrieve_save_state(uuid=uuid)[0])

# 需要导入模块: from database import DatabaseManager [as 别名]
# 或者: from database.DatabaseManager import update_table [as 别名]
def add_save_state_data(uuid):
    save_state_data = request.form.get('buffer')
    compressed = True if request.form.get('compressed') == u'true' else False
    data_length = int(request.form.get('data_length'))

    save_state_b_array = bytearray(base64.b64decode(save_state_data))

    #   The base64 library will interpret trailing zeros as padding and remove them, this adds them back in
    if len(save_state_b_array) != data_length:
        save_state_b_array.extend([0 for _ in range(len(save_state_b_array), data_length)])

    source_data_hash, file_name = save_byte_array_to_store(save_state_b_array, file_name=uuid)
    dbm.update_table(dbm.GAME_SAVE_TABLE,
                     ['save_state_source_data', 'compressed'],
                     [source_data_hash, compressed],
                     ['uuid'], [uuid])
    return jsonify({'record': dbm.retrieve_save_state(uuid=uuid)[0]})

# 需要导入模块: from database import DatabaseManager [as 别名]
# 或者: from database.DatabaseManager import update_table [as 别名]
def performance_add_data(uuid):
    sha1_hash = request.form.get('sha1_hash')
    chunk_id = request.form.get('chunk_id')
    total_chunks = int(request.form.get('total_chunks'))
    chunk_data = request.form.get('chunk_data')
    chunk_size = int(request.form.get('chunk_size'))

    save_state_b_array = bytearray(base64.b64decode(chunk_data))

    #   The base64 library will interpret trailing zeros as padding and remove them, this adds them back in
    if len(save_state_b_array) != chunk_size:
        save_state_b_array.extend([0 for _ in range(len(save_state_b_array), chunk_size)])

    #   If there isn't a temp directory for this hash, create one
    temp_path = os.path.join(LOCAL_DATA_ROOT, "tmp_{}".format(sha1_hash))
    if not os.path.exists(temp_path):
        os.makedirs(temp_path)

    #   Write the chunk of data to a temp file
    with open("{}/{}_of_{}".format(temp_path, (int(chunk_id) + 1), total_chunks), "wb") as temp:
        temp.write(save_state_b_array)

    chunk_paths = [chunk for chunk in fnmatch.filter(os.listdir(temp_path), "*_of_*")]

    #   See if all the chunks are written, if so concatenate into final file and add that to storage
    if total_chunks == len(chunk_paths):
        final_name = "final_{}.mp4".format(sha1_hash)
        final_path = os.path.join(temp_path, final_name)
        final_file = open(final_path, "ab")

        for cp in sorted(chunk_paths, key=lambda p: int(p.split("_")[0])):
            with open("{}/{}".format(temp_path, cp), "rb") as cf:
                final_file.write(cf.read())

        final_file.close()
        final_hash = save_file_to_store(final_path)
        shutil.rmtree(temp_path)

        #   Attach data to performance record
        dbm.update_table(dbm.PERFORMANCE_CITATION_TABLE,
                         ['replay_source_file_ref', 'replay_source_file_name'],
                         [final_hash, final_name],
                         ["uuid"], [uuid])
    return "OK"

# 需要导入模块: from database import DatabaseManager [as 别名]
# 或者: from database.DatabaseManager import update_table [as 别名]
def game_update(uuid):
    update_fields = json.loads(request.form.get('update_fields'))
    dbm.update_table(dbm.GAME_CITATION_TABLE, update_fields.keys(), update_fields.values(), ["uuid"], [uuid])
    game_ref = dbm.retrieve_game_ref(uuid)
    #dbm.update_table(dbm.FTS_INDEX_TABLE, ['content'], [game_ref.to_json_string()], ["uuid"], [uuid])
    return game_ref.to_json_string()

# 需要导入模块: from database import DatabaseManager [as 别名]
# 或者: from database.DatabaseManager import update_table [as 别名]
def performance_update(uuid):
    update_fields = json.loads(request.form.get('update_fields'))
    dbm.update_table(dbm.PERFORMANCE_CITATION_TABLE, update_fields.keys(), update_fields.values(), ["uuid"], [uuid])
    perf_ref = dbm.retrieve_perf_ref(uuid)
    #dbm.update_table(dbm.FTS_INDEX_TABLE, ['content'], [perf_ref.to_json_string()],["uuid"], [uuid])
    return perf_ref.to_json_string()