def handle_system_information(username, password):
mac = uuid.getnode().__str__()
system = rest_client.get_system_information(mac)
system_name = None
# Register a new System if this one isn't recognized
if system is None:
hostname = socket.gethostname()
name_input = raw_input("What do you want to call this system? " +
"For example Home, File Server, ect. [%s]: " % hostname)
name = name_input or hostname
system_name = rest_client.register_system(RegisterSystem(
name, mac, hostname, __version__))
if system_name:
print("Registered a new system " + name)
else:
return (None, None)
# Login with this new system
access_token = rest_client.login_user(LoginForm(username, password, mac))
if access_token is None:
print("Failed to login with system.")
return (None, None)
# If this system is already registered
if system is not None:
system_name = system.name
print("Welcome back! Looks like this box is already registered as " +
system.name + ".")
return (access_token, system_name)
def get_user_information_and_login(username=None, password=None, attempts=0):
if attempts == 4:
print("Too many bad attempts.")
return None
# Only collect user information if we don't already have it
# i.e. if we didn't just register a new user.
if username == None and password == None:
print("Please enter your bashhub credentials")
username = raw_input("Username: "******"Password: ")
# login once we have all of our information
access_token = rest_client.login_user(LoginForm(username, password))
# Package our result to include our credentials to later login our system.
if access_token:
result = (username, password, access_token)
else:
result = get_user_information_and_login(attempts=attempts + 1) or (
None,
None,
None,
)
return result
def get_user_information_and_login(username=None, password=None, attempts=0):
if attempts == 4:
print("Too many bad attempts.")
return None
# Only collect user information if we don't already have it
# i.e. if we didn't just register a new user.
if username == None and password == None:
print("Please enter your bashhub credentials")
username = raw_input("Username: "******"Password: ")
# login once we have all of our information
access_token = rest_client.login_user(LoginForm(username, password))
# Package our result to include our credentials to later login our system.
if access_token:
result = (username, password, access_token)
else:
result = get_user_information_and_login(attempts=attempts + 1) or (
None, None, None)
return result
