123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869 |
- import getpass
- import inspect
- import sys
- import client
- import connection
- from client import allowed_commands
- from connection import client_request
- def login(username=None, password=None):
- if connection.session_id is not None:
- client.fake_loading_bar('Signing out', delay=0.025)
- connection.session_id = None
- if username is None:
- username = input('Username: ')
- if password is None:
- if sys.stdin.isatty():
- password = getpass.getpass('Password: ')
- else:
- password = input('Password: ')
- response = client_request('login', {"username": username, "password": password})
- success = 'session_id' in response
- if success:
- connection.session_id = response['session_id']
- print('Login successful.')
- else:
- if 'error_message' in response:
- print('Login failed with message:', response['error_message'])
- else:
- print('Login failed.')
- def register(username, password=None):
- if connection.session_id is not None:
- client.fake_loading_bar('Signing out', delay=0.025)
- connection.session_id = None
- if username is None:
- username = input('Username: ')
- if password is None:
- if sys.stdin.isatty():
- password = getpass.getpass('Password: ')
- else:
- password = input('Password: ')
- response = client_request('register', {"username": username, "password": password})
- if 'error_message' in response:
- print('Registration failed with message:', response['error_message'])
- # noinspection PyShadowingBuiltins
- def help():
- print('Allowed commands:')
- for cmd in allowed_commands:
- this_module = sys.modules[__name__]
- method = getattr(this_module, cmd)
- params = inspect.signature(method).parameters
- num_args = len(params)
- if num_args > 0:
- print('`' + cmd + '`', 'takes the following', num_args, 'arguments:')
- for p in params:
- print(' -', p) # TODO print default value
- else:
- print('`' + cmd + '`', 'takes no arguments')
|