#!/usr/bin/python3
# -*- coding: UTF-8 -*-


import os
from Todooo.lib import menu,helpdoc,yourtodo,usermanager,datatrans

def main():
	print('\033[0m')
	user=usermanager.User()
	try:
		user.usr,user.id=datatrans.loaduser()
	except:
		os.system('clear')
		print('*************************\n\n\t\t\Welcome to TODO!\n\n\n\n')
		print('Please Create a New User at the first time')
		input('Press <Enter> to continue ...')
	while True:
		os.system('clear')
		menu.main_menu()
		select=int(input())
		if select == 1:
			user.add_user()
			datatrans.saveuser(user.usr,user.id)
			input("Press Enter to continue ...")


		elif select == 2:
			while True:
				name_input=input("Please Enter your username: ")

				if user.user_login(name_input):
					while True:
						os.system('clear')
						menu.after_login(name_input)
						option=int(input())
						if option == 1:
							yourtodo.your_todolist(user.id[name_input])
						elif option == 2:
							user.modify_user()
						elif option == 3:
							user.delete_user()
						elif option == 4:
							break	
				else:
					continue
				break


		elif select == 3:
			print(' **** Chenghao Qian is a lazy guy ****')
			input("Press Enter to continue...")
			continue
		elif select == 4:
			print(helpdoc.help)			
			input("Press Enter to continue...")

		elif select == 5:

			break
		else:
			print("Invalid selection, try again please")
			continue





if __name__ == '__main__':
	main()






