aboutsummaryrefslogtreecommitdiff
path: root/gui/main_window.py
blob: a269f45fe5d36b3b50c46c2edf380e5d2ba1e50d (plain)
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
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
import mariadb
from PySide6.QtGui import *
from PySide6.QtWidgets import *

from tab_drivers import *
from tab_teams import *
from tab_circuits import *
from tab_races import *

class MainWindow(QMainWindow):
  cursor: mariadb.Cursor = None
  db: mariadb.Connection = None
  menu_bar: QMenuBar
  calendar_id: int

  _tab_drivers: TabDrivers
  _tab_teams: TabTeams
  _tab_circuits: TabCircuits
  _tab_races: TabRaces

  main_layout: QTabWidget

  def set_cursor(self, cursor: mariadb.Cursor):
    self.cursor = cursor

  def set_connection(self, db: mariadb.Connection):
    self.db = db

  def commit(self):
    self.db.commit()

  def exit(self, commit=False):
    if commit: self.commit()
    self.db.close()
    self.close()

  def exit_no_commit(self):
    self.exit(False)
  def exit_commit(self):
    self.exit(True)

  def call_update_flags(self):
    folder = QFileDialog().getExistingDirectory(self, "Open directory", "/var/dab2/")
    if not folder: return
    folder += "/"
    self.cursor.execute("call spUpdateFlags(?)", (folder,))

  def call_update_persons(self):
    folder = QFileDialog().getExistingDirectory(self, "Open directory", "/var/dab2/")
    if not folder: return
    folder += "/"
    self.cursor.execute("call spUpdatePersons(?)", (folder,))

  def call_update_circuits(self):
    folder = QFileDialog().getExistingDirectory(self, "Open directory", "/var/dab2/")
    if not folder: return
    folder += "/"
    self.cursor.execute("call spUpdateCircuits(?)", (folder,))

  def call_delete_flags(self):
    self.cursor.execute("call spDeleteFlags()")

  def focus_member(self, id):
    self.set_tab_index(0)
    self._tab_drivers.set_driver_id(id)

  def focus_team(self, id):
    self.set_tab_index(1)
    self._tab_teams.set_team_id(id)

  def set_tab_index(self, index):
    self.main_layout.setCurrentIndex(index)

  def update(self, cascade=True):
    if cascade == False: return
    self._tab_drivers.update()
    self._tab_teams.update()
    self._tab_circuits.update()
    # self._tab_races.update()

  def switch_season(self):
    self.calendar_id = self.sender().property("id")
    self.update()

  def __init__(self, cursor: mariadb.Cursor, db: mariadb.Connection, parent=None):
    super(MainWindow, self).__init__(parent)

    self.set_cursor(cursor)
    self.set_connection(db)

    self.setWindowTitle("[floating] dab2 eindopdracht main window")
    self.setMinimumHeight(500)

    self._tab_drivers = TabDrivers(self.cursor, self)
    self._tab_teams = TabTeams(self.cursor, self)
    self._tab_circuits = TabCircuits(self.cursor, self)
    self._tab_races = TabRaces(self)

    self.main_layout = QTabWidget(self);
    self.main_layout.addTab(self._tab_drivers, "Drivers")
    self.main_layout.addTab(self._tab_teams, "Teams")
    self.main_layout.addTab(self._tab_circuits, "Cirucits")
    self.main_layout.addTab(self._tab_races, "Races")
    
    self.menu_bar = QMenuBar(self)
    menu = self.menu_bar.addMenu("File")
    action = menu.addAction("Exit (commit changes)")
    action.triggered.connect(self.exit_commit)
    action = menu.addAction("Exit (don't commit changes)")
    action.triggered.connect(self.exit_no_commit)
    menu = self.menu_bar.addMenu("Procedures")
    action = menu.addAction("Import/update flags")
    action.triggered.connect(self.call_update_flags)
    action = menu.addAction("Import/update driver portraits")
    action.triggered.connect(self.call_update_persons)
    action = menu.addAction("Import/update circuit maps")
    action.triggered.connect(self.call_update_circuits)
    action = menu.addAction("Delete flags")
    action.triggered.connect(self.call_delete_flags)
    menu = self.menu_bar.addMenu("Database")
    action = menu.addAction("Commit changes")
    action.triggered.connect(self.commit)

    menu = self.menu_bar.addMenu("Seasons")
    group = QActionGroup(menu)
    group.setExclusive(True)
    self.cursor.execute("select ID, year from calendar")
    for i, season in enumerate(self.cursor.fetchall()):
      action = menu.addAction(str(season[1]))
      action.setProperty("id", season[0])
      action.triggered.connect(self.switch_season)
      action.setCheckable(True)
      if i == 0:
        action.setChecked(True)
        self.calendar_id = season[0]
      group.addAction(action)

    self.setMenuBar(self.menu_bar)
    self.setCentralWidget(self.main_layout)