Fix UID's with + in them.

2 files changed, 9 insertions, 1 deletions

diff --git a/main.py b/main.py
index 5ed80a1..afd74f5 100644
--- a/main.py
+++ b/main.py
@@ -276,7 +276,8 @@ class Handler(BaseHTTPRequestHandler):
                 self.end_headers()
             else:
                 if id := query.get('id'):
-                    response = response_for(''.join(id), logged_in)
+                    print("id =", id)
+                    response = response_for(''.join(id).replace(' ', '+'), logged_in)
                     self.send_response(200)
                 else:
                     response = index_page(logged_in)
diff --git a/password.py b/password.py
index d5bc046..5e1f37e 100755
--- a/password.py
+++ b/password.py
@@ -62,14 +62,19 @@ if __name__ == '__main__':
 
     import argparse
     parser = argparse.ArgumentParser()
+
     parser.add_argument('--file', default='passwords.json')
+
     subparsers = parser.add_subparsers(dest='cmd')
+
     add_parser = subparsers.add_parser('add')
     add_parser.add_argument('username')
     add_parser.add_argument('password')
+
     val_parser = subparsers.add_parser('validate')
     val_parser.add_argument('username')
     val_parser.add_argument('password')
+
     args = parser.parse_args()
 
 
@@ -79,3 +84,5 @@ if __name__ == '__main__':
         passwords.save()
     elif args.cmd == 'validate':
         print(passwords.validate(args.username, args.password))
+    else:
+        parser.print_help()