Option to read the account number interactively

diff --git a/wg-mullvad.py b/wg-mullvad.py
index 39268b39abfca21c0d6748453d0ae0893be468bc..49ebae02c76d5756615ecd348acab1084878b6b2 100755 (executable)
--- a/wg-mullvad.py
+++ b/wg-mullvad.py
@@ -291,10 +291,9 @@ def main():
             description=f'{__file__}',
             formatter_class=argparse.ArgumentDefaultsHelpFormatter,
     )
-    required = parser.add_argument_group('required arguments')
-    required.add_argument(
-            '--account', dest='account_number', type=validate_account,
-            action='store', required=True, help='mullvad account number')
+    parser.add_argument(
+        '--account', dest='account_number', type=validate_account,
+        action='store', help='mullvad account number')
 
     parser.add_argument(
         '--settings-file', dest='settings_file', action='store',
@@ -323,6 +322,8 @@ def main():
     args = parser.parse_args()
 
     try:
+        if not args.account_number:
+            args.account_number = validate_account(input('Account number: '))
         mullvad = Mullvad(args)
         mullvad.run()
     except Exception as e: