open.source/whoogle-search
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Catch nonexistant session file, fix session type check

Browse Source
This commit is contained in:
Ben Busby 2021-10-23 12:02:00 -06:00
parent 30d929f36d
commit 1aee3893d2
No known key found for this signature in database
GPG Key ID: 339B7B7EB5333D14
1 changed files with 9 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
app/routes.py
View File

@ -60,12 +60,15 @@ def session_required(f):
session_path = os.path.join( session_path = os.path.join(
app.config['SESSION_FILE_DIR'], app.config['SESSION_FILE_DIR'],
user_session) user_session)
with open(session_path, 'rb') as session_file: try:
_ = pickle.load(session_file) with open(session_path, 'rb') as session_file:
data = pickle.load(session_file) _ = pickle.load(session_file)
if type(data) == 'dict' and 'valid' in data: data = pickle.load(session_file)
continue if isinstance(data, dict) and 'valid' in data:
invalid_sessions.append(session_path) continue
invalid_sessions.append(session_path)
except FileNotFoundError:
pass
for invalid_session in invalid_sessions: for invalid_session in invalid_sessions:
os.remove(invalid_session) os.remove(invalid_session)