WebHost: Give user feedback if generator failed.
This commit is contained in:
parent
39f85aa291
commit
e76887d0bf
|
@ -77,11 +77,6 @@ def handle_generation_failure(result: BaseException):
|
||||||
|
|
||||||
|
|
||||||
def launch_generator(pool: multiprocessing.pool.Pool, generation: Generation):
|
def launch_generator(pool: multiprocessing.pool.Pool, generation: Generation):
|
||||||
@db_session
|
|
||||||
def handle_fail(result: BaseException):
|
|
||||||
generation.state = STATE_ERROR
|
|
||||||
handle_generation_failure(result)
|
|
||||||
|
|
||||||
logging.info(f"Generating {generation.id} for {len(generation.options)} players")
|
logging.info(f"Generating {generation.id} for {len(generation.options)} players")
|
||||||
|
|
||||||
pool.apply_async(gen_game, (generation.options,),
|
pool.apply_async(gen_game, (generation.options,),
|
||||||
|
|
|
@ -48,45 +48,50 @@ def generate(race=False):
|
||||||
|
|
||||||
|
|
||||||
def gen_game(gen_options, race=False, owner=None, sid=None):
|
def gen_game(gen_options, race=False, owner=None, sid=None):
|
||||||
target = tempfile.TemporaryDirectory()
|
try:
|
||||||
playercount = len(gen_options)
|
target = tempfile.TemporaryDirectory()
|
||||||
seed = get_seed()
|
playercount = len(gen_options)
|
||||||
random.seed(seed)
|
seed = get_seed()
|
||||||
|
random.seed(seed)
|
||||||
|
|
||||||
if race:
|
if race:
|
||||||
random.seed() # reset to time-based random source
|
random.seed() # reset to time-based random source
|
||||||
|
|
||||||
seedname = "M" + (f"{random.randint(0, pow(10, seeddigits) - 1)}".zfill(seeddigits))
|
seedname = "M" + (f"{random.randint(0, pow(10, seeddigits) - 1)}".zfill(seeddigits))
|
||||||
|
|
||||||
erargs = parse_arguments(['--multi', str(playercount)])
|
erargs = parse_arguments(['--multi', str(playercount)])
|
||||||
erargs.seed = seed
|
erargs.seed = seed
|
||||||
erargs.name = {x: "" for x in range(1, playercount + 1)} # only so it can be overwrittin in mystery
|
erargs.name = {x: "" for x in range(1, playercount + 1)} # only so it can be overwrittin in mystery
|
||||||
erargs.create_spoiler = not race
|
erargs.create_spoiler = not race
|
||||||
erargs.race = race
|
erargs.race = race
|
||||||
erargs.skip_playthrough = race
|
erargs.skip_playthrough = race
|
||||||
erargs.outputname = seedname
|
erargs.outputname = seedname
|
||||||
erargs.outputpath = target.name
|
erargs.outputpath = target.name
|
||||||
erargs.teams = 1
|
erargs.teams = 1
|
||||||
erargs.progression_balancing = {}
|
erargs.progression_balancing = {}
|
||||||
erargs.create_diff = True
|
erargs.create_diff = True
|
||||||
|
|
||||||
for player, (playerfile, settings) in enumerate(gen_options.items(), 1):
|
for player, (playerfile, settings) in enumerate(gen_options.items(), 1):
|
||||||
for k, v in settings.items():
|
for k, v in settings.items():
|
||||||
if v is not None:
|
if v is not None:
|
||||||
getattr(erargs, k)[player] = v
|
getattr(erargs, k)[player] = v
|
||||||
|
|
||||||
if not erargs.name[player]:
|
if not erargs.name[player]:
|
||||||
erargs.name[player] = os.path.split(playerfile)[-1].split(".")[0]
|
erargs.name[player] = os.path.split(playerfile)[-1].split(".")[0]
|
||||||
|
|
||||||
erargs.names = ",".join(erargs.name[i] for i in range(1, playercount + 1))
|
erargs.names = ",".join(erargs.name[i] for i in range(1, playercount + 1))
|
||||||
del (erargs.name)
|
del (erargs.name)
|
||||||
|
|
||||||
erargs.skip_progression_balancing = {player: not balanced for player, balanced in
|
erargs.skip_progression_balancing = {player: not balanced for player, balanced in
|
||||||
erargs.progression_balancing.items()}
|
erargs.progression_balancing.items()}
|
||||||
del (erargs.progression_balancing)
|
del (erargs.progression_balancing)
|
||||||
ERmain(erargs, seed)
|
ERmain(erargs, seed)
|
||||||
|
|
||||||
return upload_to_db(target.name, owner, sid)
|
return upload_to_db(target.name, owner, sid)
|
||||||
|
except BaseException:
|
||||||
|
with db_session:
|
||||||
|
Generation.get(id=sid).state = STATE_ERROR
|
||||||
|
raise
|
||||||
|
|
||||||
|
|
||||||
@app.route('/wait/<suuid:seed>')
|
@app.route('/wait/<suuid:seed>')
|
||||||
|
@ -121,9 +126,9 @@ def upload_to_db(folder, owner, sid):
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
flash(e)
|
flash(e)
|
||||||
if multidata:
|
if multidata:
|
||||||
seed = Seed(multidata=multidata, spoiler=spoiler, patches=patches, owner=owner, id=sid)
|
with db_session:
|
||||||
for patch in patches:
|
seed = Seed(multidata=multidata, spoiler=spoiler, patches=patches, owner=owner, id=sid)
|
||||||
patch.seed = seed
|
for patch in patches:
|
||||||
Generation.get(id=sid).delete()
|
patch.seed = seed
|
||||||
commit()
|
Generation.get(id=sid).delete()
|
||||||
return seed.id
|
return seed.id
|
||||||
|
|
Loading…
Reference in New Issue