db-lab3/Controller.py at main · mal1k/db-lab3 · GitHub

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
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
import Model
from View import View

class Controller:
    @staticmethod
    def menu():
        flag = 0
        while flag == 0:
            print("   MENU ")
            print("1. show one table")
            print("2. show all tables")
            print("3. insert data")
            print("4. delete data")
            print("5. update data")
            print("6. exit")
            number = int(input('\nMake your number: '))
            if number == 1 or number == 2:
                Model.show_table(number)
            elif number == 3:
                number2 = View.list()
                try:
                    if number2 == 1:
                        Model.Owner.Insert()
                    elif number2 == 2:
                        Model.FootballClub.Insert()
                    elif number2 == 3:
                        Model.Player.Insert()
                    elif number2 == 4:
                        Model.HeadCoach.Insert()
                    elif number2 == 5:
                        Model.Partner.Insert()
                    elif number2 == 6:
                        Model.Assistant.Insert()
                    elif number2 == 7:
                        Model.Contract.Insert()
                except Exception:
                    print("\n... Key Error  ... Please try again ...\n")
            elif number == 4:
                number2 = View.list()
                try:
                    if number2 == 1:
                        Model.Owner.Delete()
                    elif number2 == 2:
                        Model.FootballClub.Delete()
                    elif number2 == 3:
                        Model.Player.Delete()
                    elif number2 == 4:
                        Model.HeadCoach.Delete()
                    elif number2 == 5:
                        Model.Partner.Delete()
                    elif number2 == 6:
                        Model.Assistant.Delete()
                    elif number2 == 7:
                        Model.Contract.Delete()
                except Exception:
                    print("\n... Key Error ... Please try again ...\n")
            elif number == 5:
                number2 = View.list()
                try:
                    if number2 == 1:
                        Model.Owner.Update()
                    elif number2 == 2:
                        Model.FootballClub.Update()
                    elif number2 == 3:
                        Model.Player.Update()
                    elif number2 == 4:
                        Model.HeadCoach.Update()
                    elif number2 == 5:
                        Model.Partner.Update()
                    elif number2 == 6:
                        Model.Assistant.Update()
                    elif number2 == 7:
                        Model.Contract.Update()
                except Exception:
                    print("\n... Key Error ... Please try again ...\n")
            elif number == 6:
                flag = 1