#! /usr/bin/env python
#Author: 1R3N1CU5

import sys, os, time
from hashlib import md5


class user_management():

    def __init__(self, user="", pw="", set_user="", set_pw="", database={}, choice=""):
      self.user = user
      self.pw = pw
      self.database = database
      self.choice = choice
      self.set_user = set_user
      self.set_pw = set_pw




    def clear(self):    #Clears Screen
        if os.name in ['nt', 'win32', 'dos']:
            os.system('cls')
        else:
            os.system('clear')


    def menu(self):     #Prints Menu
        print '\n'
        print '-'*50
        print 'Please Select:\n'
        print '-'*50
        print '1) Register User'
        print '2) Login'
        print '3) Exit'
        print '='*50,'\n\n'




    def select(self):   #Main Function of the class. User Selects Option.
        while True:
            self.menu()
            try:
                self.choice = raw_input ('You Entered: ').strip()
            except (KeyboardInterrupt, IOError):
                print '\nAborted! Exiting..'
                time.sleep(2)
                break

            if (self.choice in '123'):
                if (self.choice == '1'):
                    self.clear()
                    self.register()
                    self.clear()
                    print '\n'
                    print '*'*50
                    print 'New User Successfully Registered!'
                    print '^'*50

                elif (self.choice == '2'):
                    self.clear()
                    self.error_chk()
                elif (self.choice == '3'):
                    self.clear()
                    self.roll_credits()
                    break
            else:
                self.clear()
                print 'Invalid Input! Try Again..\n\n'


    def register(self):     #Registers New User
        self.set_user = (raw_input ('Please Enter User Name: ')).strip()
        self.set_pw = (raw_input ('\nPlease Enter Password %s: ' % (self.set_user))).strip()
        self.data_entry(self.set_user, self.set_pw)


    def data_entry(self, set_user, set_pw):     #Stores User Data
        self.user = set_user
        self.pw = set_pw
        self.database = {self.user:self.pw}
        return self.database


    def error_chk(self):    #Checks whether User is Authorised
        name = raw_input('Enter Username: ').strip()
        password = raw_input('Enter Password: ').strip()
        while True:
            if (name in self.database.keys()):
                print ('Username Valid! Checking Password for %s..' % (name))
                time.sleep(2)
                if (password == self.database[name]):
                    print '[+] Access Granted!'
                    self.login()
                    break
                else:
                    print 'Password Incorrect!\n'
                    print 'Returning to Main Menu..'
                    time.sleep(2)
                    self.clear()
                    break
            else:
                print '[-] Acess Denied!\n'
                break


    def login(self):
        print '\nFeature not implemented yet!\n\n'
        print 'Returning to Main Menu..'
        time.sleep(2)



    def roll_credits(self):
        print '\n\t', '-'*50, '\n\tQuitting...', '\n\t', '-'*50
        print '\n\tGot Suggestions? Bugs? Email me at : '
        print '\n\tirenicus_timberlake@yahoo.com \n'
        print '\t','='*50
        time.sleep(3)



if __name__ == '__main__':
    global mng
    mng = user_management()
    mng.select()