BITBAWT

#!/bin/python
import time
import os
import shutil
import base64
#Base64 is very unsecure, just an idea for future use.
#Init functions.
def timer(seconds):
    start = time.time()
    time.clock()
    elapsed = 0
    while elapsed < seconds:
        elapsed = time.time() - start

def createUser(username, password):
    if not os.path.exists(username):
        usercreate = open('users/' + username, "a")
        usercreate.close()
        userwrite = open('users/' + username, "w")
        userwrite.write(password)
        userwrite.close()

def gotoMenu():
    while True:
        print("Confirm your admin account.")
        adminuser = raw_input("Username: ")
        aps = raw_input("Password: ")
        if os.path.exists('users/' + adminuser):
            passd = base64.b64encode(aps)
            confirmcheck = open('users/' + adminuser)
            checkedpw = confirmcheck.readline().rstrip()
            if passd == checkedpw:
                gotoRealMenu()
                break
        else:
            print("Unmatched!")
            quit()

def gotoRealMenu():
    print("(--------------------------------------)")
    print("(          M      E     N      U       )")
    print("(--------------------------------------)")
    print("1. Reset users")
    print("2. Add new admin")
    print("3. Quit")

    selection = raw_input("What would you like to do? ")
    if selection == "1":
            shutil.rmtree('users/')
            quit()
    elif selection == "2":
        amuser = raw_input("Enter your desired username: ")
        ampass = raw_input("Enter your desired password: ")
        amnewpass = base64.b64encode(ampass)
        if not os.path.exists(amuser):
            usercr = open('users/' + amuser, "a")
            usercr.close()
            userwr = open('users/' + amuser, "a")
            userwr.write(amnewpass)
            userwr.close()
            print("Quitting to save.")
            quit()
    elif selection == "3":
        quit()
    else:
        print("An error occurred!")
        quit()

print(chr(27) + "[2J")
print("(--------------------------------------)")
print("(       B     I    T    B    O    T    )")
print("(--------------------------------------)")
print(" Hello, welcome to bitbot.  ")
print(" ")


directory = 'users/'
if not os.path.exists(directory):
     print("It's your first time! Let's get you set up!")
     os.makedirs(directory)
     print("Let's make an Administrator.")
     auser = raw_input("Enter your new admin username: ")
     abadpass = raw_input("Enter your new admin password: ")
     apass = base64.b64encode(abadpass)
     createUser(auser, apass)




menuask = raw_input("Would you like to go to the menu? (y/N) ")
if menuask == "y":
    gotoMenu()

print("Please login")
while True:
    username = raw_input("Enter your username: ")
    password = raw_input("Enter your password: ")
    if os.path.exists('users/' + username):
       usercheck = open('users/' + username)
       passwordcheck = usercheck.readline().rstrip()
       pswd = base64.b64encode(password)
       if pswd == passwordcheck:
          print("Logged in as: " + username)
          usercheck.close()
          break
    else:
        print("Incorrect username or password! Quitting!")
        quit()