summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--jfr_playoff/gui/frames/team.py96
1 files changed, 61 insertions, 35 deletions
diff --git a/jfr_playoff/gui/frames/team.py b/jfr_playoff/gui/frames/team.py
index abeb56f..a24aa0b 100644
--- a/jfr_playoff/gui/frames/team.py
+++ b/jfr_playoff/gui/frames/team.py
@@ -72,37 +72,49 @@ class TeamSelectionFrame(ScrollableFrame):
in enumerate(self.values) if value.get()])
self.master.destroy()
- def renderContent(self, container):
+ def renderHeader(self, container):
container.columnconfigure(1, weight=1)
(ttk.Label(container, text=self.title)).grid(
row=0, column=0, columnspan=2)
- row = 1
- for team in self.teams:
- (ttk.Label(container, text='[%d]' % (row))).grid(row=row, column=0)
- self.values.append(tk.IntVar())
- (ttk.Checkbutton(
- container, text=team[0],
- variable=self.values[row-1]
- )).grid(row=row, column=1, sticky=tk.W)
- if self.selected and self.selected(row-1, team):
- self.values[row-1].set(True)
- row += 1
-
-class TeamFetchSettingsFrame(GuiFrame):
- SOURCE_LINK = 0
- SOURCE_DB = 1
-
- def _changeNotify(self, *args):
- self.winfo_toplevel().event_generate(
- '<<TeamSettingsChanged>>', when='tail')
- def _setFinishingPositions(self, positions):
- self.finishingPositions = positions
- self.finishingPositionsBtn.configure(
- text='[wybrano: %d]' % (len(self.finishingPositions)))
- self._changeNotify(None)
+ def renderTeam(self, container, team, idx):
+ (ttk.Label(container, text='[%d]' % (idx+1))).grid(
+ row=idx+1, column=0)
+ (ttk.Checkbutton(
+ container, text=team[0],
+ variable=self.values[idx]
+ )).grid(row=idx+1, column=1, sticky=tk.W)
- def _chooseFinishingPositions(self):
+ def renderContent(self, container):
+ self.renderHeader(container)
+ for idx, team in enumerate(self.teams):
+ self.values.append(tk.IntVar())
+ self.renderTeam(container, team, idx)
+ if self.selected and self.selected(idx, team):
+ self.values[idx].set(True)
+
+class TeamSelectionButton(ttk.Button):
+ def __init__(self, *args, **kwargs):
+ for arg in ['callback', 'prompt', 'dialogclass']:
+ setattr(self, arg, kwargs[arg] if arg in kwargs else None)
+ if arg in kwargs:
+ del kwargs[arg]
+ kwargs['command'] = self._choosePositions
+ if self.dialogclass is None:
+ self.dialogclass = TeamSelectionFrame
+ if self.prompt is None:
+ self.prompt = 'Wybierz teamy:'
+ ttk.Button.__init__(self, *args, **kwargs)
+ self.setPositions([])
+
+ def setPositions(self, values):
+ self.selected = values
+ self.configure(
+ text='[wybrano: %d]' % (len(values)))
+ if self.callback is not None:
+ self.callback(values)
+
+ def _choosePositions(self):
teams = self.winfo_toplevel().getTeams()
if not len(teams):
tkMessageBox.showerror(
@@ -111,14 +123,26 @@ class TeamFetchSettingsFrame(GuiFrame):
else:
dialog = tk.Toplevel(self)
dialog.title('Wybór teamów')
- selectionFrame = TeamSelectionFrame(
- dialog, title='Wybierz teamy, które zakończyły rozgrywki ' + \
- 'na swojej pozycji:',
+ selectionFrame = self.dialogclass(
+ dialog, title=self.prompt,
teams=teams,
- selected=lambda idx, team: idx+1 in self.finishingPositions,
- callback=self._setFinishingPositions, vertical=True)
+ selected=lambda idx, team: idx+1 in self.selected,
+ callback=self.setPositions, vertical=True)
selectionFrame.pack(fill=tk.BOTH, expand=True, padx=5, pady=5)
+
+class TeamFetchSettingsFrame(GuiFrame):
+ SOURCE_LINK = 0
+ SOURCE_DB = 1
+
+ def _setFinishingPositions(self, positions):
+ self.finishingPositions = positions
+ self._changeNotify(None)
+
+ def _changeNotify(self, *args):
+ self.winfo_toplevel().event_generate(
+ '<<TeamSettingsChanged>>', when='tail')
+
def getTeams(self):
teams = {}
if self.fetchSource.get() == self.SOURCE_LINK:
@@ -191,10 +215,12 @@ class TeamFetchSettingsFrame(GuiFrame):
(ttk.Label(self, text='Pozycje końcowe: ')).grid(
row=3, column=0, columnspan=3, sticky=tk.W+tk.E)
- self.finishingPositionsBtn = ttk.Button(
- self, command=self._chooseFinishingPositions)
- self.finishingPositionsBtn.grid(row=3, column=3, sticky=tk.W)
- self._setFinishingPositions([])
+ finishingPositionsBtn = TeamSelectionButton(
+ self, callback=self._setFinishingPositions,
+ prompt='Wybierz teamy, które zakończyły rozgrywki ' + \
+ 'na swojej pozycji:')
+ finishingPositionsBtn.grid(row=3, column=3, sticky=tk.W)
+ finishingPositionsBtn.setPositions([])
class TeamSettingsFrame(ScrollableFrame):
FORMAT_FETCH = 0