ericlay
/
fuckyourday
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

swap if/else for case statement

This commit is contained in:
Eric Lay 2024-03-27 21:52:53 -05:00
parent 26f2b4f4f2
commit 897f577706
1 changed files with 13 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
app.py
View File

@ -16,15 +16,16 @@ WEBSITE_DIRS = [name for name in os.listdir(ROOT_DIR) if not name.startswith('.'
# Site choosing logic
def current_website_dir():
session.pop('website_dir', None)
if session['requests'] % 5 == 0:
match session['requests']:
case 5:
session['website_dir'] = (os.path.join(ROOT_DIR, 'Marvel'))
elif session['requests'] % 6 == 0:
case 6:
session['website_dir'] = (os.path.join(ROOT_DIR, 'Escape'))
session.pop('requests', None)
else:
case _:
try:
WEBSITE_DIRS.remove(os.path.join(ROOT_DIR, 'Marvel'))
WEBSITE_DIRS.remove(os.path.join(ROOT_DIR, 'Escape'))
WEBSITE_DIRS.remove(os.path.join(ROOT_DIR, 'Marvel'))
except (ValueError, IndexError) as e:
pass
session['website_dir'] = random.choice(WEBSITE_DIRS)