import os
import string
import time

version = "1.0 Alfa"

def login():
    print ("----------------------------------------")
    print ("                Login                   ")
    print ("----------------------------------------")
    k_name = input("Enter username: ")
    if os.path.exists(k_name + ".txt") == False:
        print ("Username not found.")
        create()
    else:
        k_pass = input("Enter password: ")
        with open(k_name + ".txt", "r") as f:
            if k_pass == f.read():
                print("Welcome %s!"%k_name)
                f.close()
                input()
            else:
                print("Password is wrong!")
                create()



def create():
    print("You using login system %s" % version)
    print( "----------------------------------------")
    print("|                 Lobby                |")
    print( "----------------------------------------")
    starting = input("To create user type R, to login type L").upper()
    if starting == "R":
        name = input("Enter username: ")
        password = input("Enter password: ")
        password2 = input("Enter password again: ")
        if password == password2:
            newfile = open(name + ".txt", "w")
            newfile.write(password)
            newfile.close()
            print("User created. Redirecting you to login.")
            time.sleep(2)
            login()
        elif password != password2:
            print("Passwords doesn't match.")
            input()
            create()
    elif starting == "L":
        login()
    else:
        print("nWrong buttonn")
        create()

create()