def citation_page(uuid):
game_ref = dbm.retrieve_game_ref(uuid)
perf_ref = dbm.retrieve_perf_ref(uuid)
derived_performances = dbm.retrieve_derived_performances(uuid)
previous_performances = dbm.retrieve_performance_chain(uuid)[:-1]
save_states = dbm.retrieve_save_state(game_uuid=uuid)
extra_files = dbm.retrieve_file_path(game_uuid=uuid)
if game_ref:
return render_template('citation.html',
citeref=game_ref,
is_game=True,
is_performance=False,
derived_performances=derived_performances,
extra_files=extra_files,
save_states=save_states)
elif perf_ref:
performance_video = "/cite_data/{}/{}".format(perf_ref['replay_source_file_ref'],
perf_ref['replay_source_file_name'])
return render_template('citation.html',
citeref=perf_ref,
is_game=False,
is_performance=True,
previous_performances=previous_performances,
performance_video=performance_video)
return "No record found, sorry."
def performance_info(uuid):
perf_info = {}
perf_ref = dbm.retrieve_perf_ref(uuid)
perf_info['record'] = perf_ref.elements
perf_info['linkedStates'] = dbm.retrieve_all_state_perf_links(uuid)
perf_info['availablePerformances'] = [dict(p.get_element_items()) for p in dbm.retrieve_derived_performances(perf_ref['game_uuid'])]
return jsonify(perf_info)
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()
def delete(ctx, uuid, keep_performances):
no_prompts = ctx.obj['NO_PROMPTS']
verbose = ctx.obj['VERBOSE']
def check_delete(path):
try:
shutil.rmtree(path)
except OSError as e:
print e.message
def delete_from_fts(uuid):
ix = open_dir(LOCAL_FTS_INDEX)
writer = ix.writer()
writer.delete_by_term('id', uuid)
writer.commit()
game_ref = dbm.retrieve_game_ref(uuid)
perf_ref = dbm.retrieve_perf_ref(uuid)
state_ref = dbm.retrieve_save_state(uuid=uuid)
if game_ref:
perfs = dbm.retrieve_derived_performances(uuid)
states = dbm.retrieve_save_state(game_uuid=uuid)
files = dbm.retrieve_file_path(game_uuid=uuid)
for f in files:
check_delete(os.path.join(LOCAL_DATA_ROOT, LOCAL_GAME_DATA_STORE, f['source_data']))
dbm.delete_from_table(dbm.GAME_FILE_PATH_TABLE, ['game_uuid'], [uuid])
for s in states:
if s['save_state_source_data']:
check_delete(os.path.join(LOCAL_DATA_ROOT, LOCAL_CITATION_DATA_STORE, s['save_state_source_data']))
else:
if s['rl_starts_data']:
check_delete(os.path.join(LOCAL_DATA_ROOT, LOCAL_CITATION_DATA_STORE, s['rl_starts_data']))
check_delete(os.path.join(LOCAL_DATA_ROOT, LOCAL_CITATION_DATA_STORE, s['rl_lengths_data']))
if s['has_screen']:
check_delete(os.path.join(LOCAL_DATA_ROOT, LOCAL_CITATION_DATA_STORE, s['uuid']))
dbm.delete_from_table(dbm.GAME_SAVE_TABLE, ['game_uuid'], [uuid])
for p in perfs:
if p['replay_source_file_ref']:
check_delete(os.path.join(LOCAL_DATA_ROOT, LOCAL_CITATION_DATA_STORE, p['replay_source_file_ref']))
dbm.delete_from_table(dbm.SAVE_STATE_PERFORMANCE_LINK_TABLE,['performance_uuid'], [p['uuid']])
dbm.delete_from_table(dbm.PERFORMANCE_CITATION_TABLE, ['game_uuid'], [uuid])
if game_ref['source_data']:
check_delete(os.path.join(LOCAL_DATA_ROOT, LOCAL_GAME_DATA_STORE, game_ref['source_data']))
dbm.delete_from_table(dbm.GAME_CITATION_TABLE, ['uuid'], [uuid])
delete_from_fts(uuid)
elif perf_ref:
dbm.delete_from_table(dbm.SAVE_STATE_PERFORMANCE_LINK_TABLE,['performance_uuid'], [uuid])
if perf_ref['replay_source_file_ref']:
check_delete(os.path.join(LOCAL_DATA_ROOT, LOCAL_CITATION_DATA_STORE, perf_ref['replay_source_file_ref']))
dbm.delete_from_table(dbm.PERFORMANCE_CITATION_TABLE, ['uuid'], [uuid])
delete_from_fts(uuid)
elif state_ref:
state_ref = state_ref[0] # retrieve states returns lists
files = dbm.retrieve_file_path(save_state_uuid=uuid)
for f in files:
check_delete(os.path.join(LOCAL_DATA_ROOT, LOCAL_GAME_DATA_STORE, f['source_data']))
dbm.delete_from_table(dbm.SAVE_STATE_PERFORMANCE_LINK_TABLE,['save_state_uuid'], [uuid])
if state_ref['has_screen']:
check_delete(os.path.join(LOCAL_DATA_ROOT, LOCAL_CITATION_DATA_STORE, state_ref['uuid']))
dbm.delete_from_table(dbm.GAME_SAVE_TABLE, ['uuid'], [uuid])
delete_from_fts(uuid)
else:
click.echo('UUID {} not found.'.format(uuid))
sys.exit(1)
def gif_performance(ctx, uuid, start, end, regenerate):
verbose = ctx.obj['VERBOSE']
# Peformance retrieval
perf = dbm.retrieve_perf_ref(uuid)
if not perf:
click.echo('Invalid performance uuid: {}'.format(uuid))
sys.exit(1)
cond_print(verbose, "Found performance: {}, '{}'".format(uuid, perf['title']))
# GIF creation
gif_source_path = "'{}/{}/{}'".format(os.path.abspath(LOCAL_CITATION_DATA_STORE),
perf['replay_source_file_ref'],
perf['replay_source_file_name'])
gif_abs_path = "{}/{}".format(os.path.abspath(LOCAL_CITATION_DATA_STORE),
perf['replay_source_file_ref'])
gif_palette_name = "{}_{}_{}_palette.png".format(uuid, start, end)
gif_file_name = "{}_{}_{}.gif".format(uuid, start, end)
# All commands are just straight defaults for now, basic settings are:
# FPS: 10
# Height: 320
# Width: Scale to height
ffmpeg_palette = '/usr/local/bin/ffmpeg -y -ss {} -t {} -i {} -vf fps=10,scale=320:-1:flags=lanczos,palettegen {}'.format(
start,
end - start,
gif_source_path,
"'{}/{}'".format(gif_abs_path, gif_palette_name)
)
ffmpeg_gif = '/usr/local/bin/ffmpeg -ss {} -t {} -i {} -i {} -filter_complex "fps=10,scale=320:-1:flags=lanczos[x];[x][1:v]paletteuse" {}'.format(
start,
end - start,
gif_source_path,
"'{}/{}'".format(gif_abs_path, gif_palette_name),
"'{}/{}'".format(gif_abs_path, gif_file_name)
)
# Check if gif is already present, and remove if regenerate flag is set
if os.path.isdir(gif_abs_path) and regenerate:
cond_print(verbose, "--regenerate flag active, deleting previous gif information")
shutil.rmtree(gif_abs_path)
elif os.path.isdir(gif_abs_path):
click.echo('Gif for that time index already present, please use --regenerate to override this message.')
sys.exit(1)
# Create gif directory
cond_print(verbose, "Creating gif directory at {}".format(gif_abs_path))
os.makedirs(gif_abs_path)
# Run gif creation processes
cond_print(verbose, "Running gif creation processes...")
cond_print(verbose, "{}".format(ffmpeg_palette))
try:
subprocess.check_call(ffmpeg_palette, shell=True)
except subprocess.CalledProcessError as e:
click.echo(e.message)
sys.exit(1)
cond_print(verbose, "{}".format(ffmpeg_gif))
try:
subprocess.check_call(ffmpeg_gif, shell=True)
except subprocess.CalledProcessError as e:
click.echo(e.message)
sys.exit(1)
cond_print(verbose, "Success!")
