Merge pull request #18 from wulkanowy/bugfix/keyerror_at_userid

Fix KeyError: userid error
2021-01-23 22:20:08 +01:00 · 2021-01-23 22:20:08 +01:00 · ca39721f87
commit ca39721f87
parent a1695fbdd9 1ed58d3174
1 changed files with 9 additions and 1 deletions
--- a/app/views.py
+++ b/app/views.py
@ -41,8 +41,16 @@ def login(request, *args, **kwargs):
    else:
        key = Fernet.generate_key()
        rkey = Fernet(key)
-
+        x = None
        request.session[request.session.session_key] = key.decode('utf-8')
+
+        while x == None:
+            try:
+                x = request.session[request.session.session_key]
+                x = x
+            except KeyError:
+                x = None
+
        sender_return['s'] = json.dumps(sender_return['s'])
        sender_return['s'] = sender_return['s'].encode()
        sender_return['s'] = rkey.encrypt(sender_return['s'])