import hashlib, os, re
from pathlib import Path

salt = '&C0lzEXdZZ^F2Wk5k9&ZztuF2Bz4o7xobux1Gc#$RbPLt'  #salt
salt2 = 'q1h3Mv#KjLlZoeTJ%&ng3by5svan6kk5eNbHIARyfl3#A' #salt2

def register():
    usr = raw_input("Username: ")
    password = raw_input("Password: ")
    pass_hash_salt = salt + password + salt2  #saltPasswordsalt2
    hash1 = hashlib.sha256(pass_hash_salt) #encrypts
    enc = hash1.hexdigest() #print "Encrypted Text: " + enc
    os.system('clear') #clears terminal
    print ("--Register-Portal--")
    db = (password, enc)
    dbs = (usr)
    dbss = (enc)

    def file():
        upload_usr.write(dbs) #adds user
        upload_usr.close() #closes
        upload_sp = open("db.txt", "a")  #opens db.txt
        upload_sp.write("\n") #adds a new line
        upload_sp.close() #closes
        upload_enc = open("db.txt", "a") #opens db.txt
        upload_enc.write(dbss) #writes encrypted password
        upload_enc.close() #closes
        upload_sp = open("db.txt", "a") #opens db.txt
        upload_sp.write("\n") #adds new line
        upload_sp.close() #closes

    dbdir = Path("/home/element/Documents/Pyton/Projects/db.txt") #db.txt path
    if dbdir.is_file(): #checks the existance of db.txt in /home/element/Documents/Pyton/Projects/db.txt
        upload_usr = open("db.txt", "a") #opens db.txt
        if usr in open('db.txt').read(): #searches for username in db.txt
            print "Error! User already entered in Database!"
            nxt = raw_input("Enter to clear")
            os.system("clear")
            print "Login-Page"
            register()
        else:
            file()
    else:
        upload_usr = open("db.txt", "w") #if db.txt doesnt exist it makes it
        file()
    print ("Successfully Registered! Please login.")
    nxt = raw_input('Enter to clear')
    os.system("clear")
    main()

def login():
    usr = raw_input("Username: ")
    pwd = raw_input("Password: ")
    salt_pwd = (salt + pwd +salt2) #saltPasswordSalt2
    salt_pwd_hash = hashlib.sha256(salt_pwd) #encrypts
    enc = salt_pwd_hash.hexdigest() #sets as a hexidecimal
    print (enc)

def main():
    print ("Login-Page")
    answ = raw_input("Register 1: \nLogin 2: \n:")
    if answ == "1":
        register()
    elif answ == "2":
        login()
    else:
        return False
if __name__ == '__main__':
    main()