/
main.py
49 lines (40 loc) · 1.13 KB
/
main.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
import Movies
import Users, ratings
def display_menu():
print("COMMAND MENU")
print("L - List all movies")
print("A - Add a movie")
print("R - Rate a movie")
print("S - Search a movie")
print("Q - Quit program")
print()
def home_menu():
a = input(" 1 to login: \n"
" 2 to sign up: ")
if a == "1":
Users.login(User_details=Users.User_details)
elif a == "2":
Users.sign_up(User_details=Users.User_details)
else:
print("Please provide valid input")
home_menu()
def main():
home_menu()
while True:
command = input("Command: ")
if command == "L":
Movies.list(movie_list=Movies.movie_list)
elif command == "A":
Movies.add(movie_list=Movies.movie_list)
elif command == "R":
ratings.rate(ratings)
elif command == "S":
Movies.search(movie_list=Movies.movie_list)
elif command == "Q":
break
else:
print("Not a valid command. Please try again.\n")
home_menu()
print("Bye!")
if __name__ == "__main__":
main()