# Created by Burhanuddin Udaipurwala on 02/01/2019


import hashlib


class Account:
    def __init__(self, name, account_number, pin, initial_balance):
        self.__name = name
        self.__account_number = account_number
        self.__hashed_pin = hashlib.md5(pin.encode()).hexdigest()
        self.__balance = initial_balance

    def __str__(self):
        return str(self.__account_number) + str(self.__hashed_pin)

    def get_balance(self):
        return self.__balance

    def get_name(self):
        return self.__name

    def get_account_number(self):
        return self.__account_number

    def get_pin_hash(self):
        return self.__hashed_pin

    def withdraw(self, withdraw_amount):
        try:
            self.__balance = int(self.__balance) - int(withdraw_amount)
        except ValueError:
            print("Enter a valid integer value")
        print("Account balance: ", self.get_balance())

    def credit(self, credit_amount):
        try:
            self.__balance = int(self.__balance) + int(credit_amount)
        except ValueError:
            print("Enter a valid integer value")
        print("Account balance: ", self.get_balance())


# mimic api using a list
account_list = []

# track if user is logged in
account_logged_in = False


def create_account():
    name = input("Enter your name")
    initial_balance = input("Enter initial balance")
    pin = input("Enter pin")
    account_number = input("Enter 4 digit account number")

    for account in account_list:
        if account.get_account_number() == account_number:
            print("Account number already exists")

    account_number = input("Enter 4 digit account number")

    account_list.append(Account(name, account_number, pin, initial_balance))
    print("Account created, you can now log in")


def login():
    account_number = input("Enter account")
    pin = input("Pin")
    pin = hashlib.md5(pin.encode()).hexdigest()
    for account in account_list:
        if account.get_account_number() == account_number and account.get_pin_hash() == pin:
            return account

    print("Error: Account not found")
    return False


def print_account_balance():
    if not account_logged_in:
        print("Login to get balance")
        return

    print("Account balance: ", account_logged_in.get_balance())


def credit_balance():
    if not account_logged_in:
        print("Login to continue")
        return
    credit_amount = input("Enter amount to credit")
    account_logged_in.credit(credit_amount)


def withdraw_money():
    if not account_logged_in:
        print("Login to continue")
        return
    withdraw_amount = input("Enter amount to withdraw")
    account_logged_in.withdraw(withdraw_amount)


while True:
    print("Enter your choice")
    print("0. Exit")
    print("1. Create Account")
    print("2. Login")
    print("3. Get balance")
    print("4. Deposit money")
    print("5. Withdraw money")
    print("6. Log out")

    choice = int(input("Enter your choice"))

    if choice == 0:
        break
    elif choice == 1:
        create_account()
    elif choice == 2:
        account_logged_in = login()
    elif choice == 3:
        print_account_balance()
    elif choice == 4:
        credit_balance()
    elif choice == 5:
        withdraw_money()
    elif choice == 6:
        account_logged_in = False
    else:
        print("Incorrect choice, Enter a valid input")