Kosmorro
/
kosmorro
spegling av https://github.com/Kosmorro/kosmorro
1
0
Förgrening 0
Kod Ärenden Släpp 36 Aktiviteter
Du kan inte välja fler än 25 ämnen Ämnen måste starta med en bokstav eller siffra, kan innehålla bindestreck ('-') och vara max 35 tecken långa.
222 Incheckningar
5 Grenar
34 MiB
 
 
 
 
Träd: e6d7de82f4
Grenar Taggar
${ item.name }
Skapa branchen ${ searchTerm }
från 'e6d7de82f4'
${ noResults }
kosmorro/kosmorrolib/gui/panel.py

242 rader
9.8 KiB
Rå Blame Historik 

  1. import os

  2. from typing import Union

  3. 

  4. import wx

  5. import wx.adv

  6. import wx.grid

  7. import wx.richtext

  8. import wx.lib.newevent

  9. 

  10. from .positionwindow import PositionWindow

  11. 

  12. from ..data import Position, AsterEphemerides, Event, MoonPhase

  13. from ..i18n import _, TIME_FORMAT, FULL_DATE_FORMAT

  14. from datetime import date

  15. 

  16. FPB_HORIZONTAL = 0x4

  17. 

  18. ComputeButtonEvent, EVT_COMPUTE_BUTTON = wx.lib.newevent.NewEvent()

  19. 

  20. ExportButtonEvent, EVT_EXPORT_BUTTON = wx.lib.newevent.NewEvent()

  21. 

  22. 

  23. class ConfigPanel(wx.Panel):

  24.     activate_position: bool

  25.     position: Union[None, Position]

  26.     compute_date: date

  27.     timezone: int

  28. 

  29.     def __init__(self, parent, activate_position: bool = False, position: Position = None,

  30.                  compute_date: date = date.today(), timezone: int = 0):

  31.         super(ConfigPanel, self).__init__(parent)

  32.         self.activate_position = activate_position

  33.         self.position = position

  34.         self.compute_date = compute_date

  35.         self.timezone = timezone

  36. 

  37.         self._position_checkbox = wx.CheckBox(self, label=_('Position:'))

  38.         self._position_change_btn = wx.Button(self)

  39.         self.update_position_btn_label()

  40.         self._position_change_btn.SetToolTip(wx.ToolTip(_('Change the position')))

  41.         self._position_checkbox.SetValue(activate_position)

  42.         self._position_change_btn.Enable(activate_position)

  43. 

  44.         self.Bind(wx.EVT_CHECKBOX, self.on_position_checkbox, self._position_checkbox)

  45.         self.Bind(wx.EVT_BUTTON, self.on_position_button, self._position_change_btn)

  46. 

  47.         date_lbl = wx.StaticText(self, label=_('Date:'))

  48.         self._date_picker = wx.adv.DatePickerCtrl(self, dt=wx.DateTime(compute_date),

  49.                                                   style=wx.adv.DP_DEFAULT)

  50.         self.Bind(wx.adv.EVT_DATE_CHANGED, self.date_changed, self._date_picker)

  51. 

  52.         timezone_lbl = wx.StaticText(self, label=_('Timezone:'))

  53.         self._timezone_spin = wx.SpinCtrl(self, min=-23, max=23, value=str(timezone))

  54.         self.Bind(wx.EVT_SPINCTRL, self.timezone_changed, self._timezone_spin)

  55. 

  56.         sizer = wx.FlexGridSizer(2, 5, 5)

  57.         sizer.AddGrowableCol(1, 1)

  58. 

  59.         self._compute_button = wx.Button(self, label=_('&Compute'))

  60.         self._export_button = wx.Button(self, label=_('E&xport PDF…'))

  61. 

  62.         self.Bind(wx.EVT_BUTTON, self.compute_button_clicked, self._compute_button)

  63.         self.Bind(wx.EVT_BUTTON, self.export_button_clicked, self._export_button)

  64. 

  65.         sizer.AddMany([(self._position_checkbox, 0, wx.EXPAND),

  66.                        (self._position_change_btn, 0, wx.EXPAND),

  67.                        (date_lbl, 0, wx.EXPAND),

  68.                        (self._date_picker, 0, wx.EXPAND),

  69.                        (timezone_lbl, 0, wx.EXPAND),

  70.                        (self._timezone_spin, 0, wx.EXPAND)])

  71. 

  72.         main_sizer = wx.FlexGridSizer(3, 5, 5)

  73.         main_sizer.AddGrowableCol(0, 2)

  74. 

  75.         main_sizer.AddMany([(sizer, 0, wx.EXPAND),

  76.                             (self._compute_button, 0, wx.EXPAND),

  77.                             (self._export_button, 0, wx.EXPAND)])

  78. 

  79.         self.SetSizer(main_sizer)

  80. 

  81.     def update_position_btn_label(self):

  82.         self._position_change_btn.Label = str(self.position) if self.position is not None else _('Unknown')

  83. 

  84.     def enable_buttons(self, enable: bool = True):

  85.         self._compute_button.Enable(enable)

  86.         self._export_button.Enable(enable)

  87. 

  88.     def disable_buttons(self):

  89.         self.enable_buttons(False)

  90. 

  91.     def on_position_checkbox(self, event):

  92.         self.activate_position = event.IsChecked()

  93.         self._position_change_btn.Enable(self.activate_position)

  94. 

  95.         if self.activate_position and self.position is None:

  96.             self.on_position_button()

  97. 

  98.     def on_position_button(self, _=None):

  99.         pos_win = PositionWindow(self, self.position)

  100.         pos_win.ShowModal()

  101.         self.position = pos_win.position

  102.         if self.position is not None:

  103.             self.update_position_btn_label()

  104.         else:

  105.             self._position_checkbox.SetValue(False)

  106.             self._position_change_btn.Disable()

  107. 

  108.     def date_changed(self, _):

  109.         self.compute_date = date.fromisoformat(self._date_picker.GetValue().FormatISODate())

  110. 

  111.     def timezone_changed(self, _):

  112.         self.timezone = self._timezone_spin.GetValue()

  113. 

  114.     def compute_button_clicked(self, _):

  115.         wx.PostEvent(self.GetEventHandler(), wx.PyCommandEvent(EVT_COMPUTE_BUTTON.typeId, self.GetId()))

  116. 

  117.     def export_button_clicked(self, _):

  118.         wx.PostEvent(self.GetEventHandler(), wx.PyCommandEvent(EVT_EXPORT_BUTTON.typeId, self.GetId()))

  119. 

  120. 

  121. class MoonPhasePanel(wx.Panel):

  122.     def __init__(self, parent):

  123.         super(MoonPhasePanel, self).__init__(parent)

  124. 

  125.         self.moon_bitmap = wx.StaticBitmap(self, size=wx.Size(100, 100))

  126. 

  127.         self.moon_txt_l1 = wx.StaticText(self)

  128.         self.moon_txt_l2 = wx.StaticText(self)

  129. 

  130.         self.moon_txt_l1.SetFont(wx.Font(wx.DEFAULT, wx.DEFAULT, wx.NORMAL, wx.BOLD))

  131. 

  132.         label_sizer = wx.GridSizer(3, 1, 0, 0)

  133.         label_sizer.AddMany([(wx.StaticText(self), 0, wx.ALIGN_LEFT),

  134.                              (self.moon_txt_l1, 0, wx.ALIGN_LEFT | wx.EXPAND),

  135.                              (self.moon_txt_l2, 0, wx.ALIGN_LEFT | wx.EXPAND)])

  136. 

  137.         main_sizer = wx.FlexGridSizer(4, 0, 10)

  138.         main_sizer.AddGrowableCol(0)

  139.         main_sizer.AddGrowableCol(2)

  140.         main_sizer.AddMany([(wx.StaticText(self), 0, wx.ALIGN_LEFT),

  141.                             (self.moon_bitmap, 0, wx.ALIGN_RIGHT),

  142.                             (label_sizer, 0, wx.ALIGN_LEFT)])

  143. 

  144.         self.SetSizer(main_sizer)

  145. 

  146.     def set_moon_phase(self, moon_phase: MoonPhase):

  147.         moon_img = wx.Image(os.path.join(os.path.abspath(os.path.dirname(__file__)),

  148.                                          '..', 'assets', 'moonphases', 'png',

  149.                                          '.'.join([moon_phase.identifier.lower().replace('_', '-'),

  150.                                                    'png'])))

  151.         moon_img = moon_img.Scale(100, 100)

  152.         bitmap = moon_img.ConvertToBitmap()

  153.         self.moon_bitmap.SetBitmap(bitmap)

  154. 

  155.         self.moon_txt_l1.SetLabel(moon_phase.get_phase())

  156.         self.moon_txt_l2.SetLabel(_('{next_moon_phase} on {next_moon_phase_date} at {next_moon_phase_time}').format(

  157.             next_moon_phase=moon_phase.get_next_phase_name(),

  158.             next_moon_phase_date=moon_phase.next_phase_date.strftime(FULL_DATE_FORMAT),

  159.             next_moon_phase_time=moon_phase.next_phase_date.strftime(TIME_FORMAT)

  160.         ))

  161. 

  162.         self.Layout()

  163.         self.Fit()

  164. 

  165. 

  166. class ResultPanel(wx.Panel):

  167.     ephemerides: [AsterEphemerides]

  168.     events: [Event]

  169.     moon_phase: Union[None, MoonPhase]

  170. 

  171.     def __init__(self, parent):

  172.         super(ResultPanel, self).__init__(parent)

  173.         self.moon_phase = None

  174.         self.ephemerides = None

  175.         self.events = None

  176. 

  177.         self._moon_phase_panel = MoonPhasePanel(self)

  178. 

  179.         self._grid_ephemerides = wx.grid.Grid(self)

  180.         self._grid_ephemerides.SetDefaultCellAlignment(wx.ALIGN_CENTRE, wx.ALIGN_CENTRE)

  181.         self._grid_ephemerides.CreateGrid(numRows=0, numCols=3)

  182.         self._grid_ephemerides.EnableEditing(False)

  183.         self._grid_ephemerides.SetColLabelValue(0, _('Rise time'))

  184.         self._grid_ephemerides.SetColLabelValue(1, _('Culmination time'))

  185.         self._grid_ephemerides.SetColLabelValue(2, _('Set time'))

  186.         self._grid_ephemerides.EnableEditing(False)

  187. 

  188.         self._list_events = wx.richtext.RichTextCtrl(self, size=wx.Size(0, 300),

  189.                                                      style=wx.richtext.RE_READONLY | wx.richtext.RE_MULTILINE)

  190. 

  191.         main_sizer = wx.FlexGridSizer(2, 1, 5, 5)

  192.         main_sizer.AddGrowableCol(0)

  193.         main_sizer.AddGrowableRow(1, 2)

  194. 

  195.         sizer = wx.FlexGridSizer(2, 5, 5)

  196.         sizer.AddGrowableCol(0, 0)

  197.         sizer.AddGrowableCol(1, 1)

  198.         sizer.AddMany([(self._grid_ephemerides, 0, wx.EXPAND | wx.ALL),

  199.                        (self._list_events, 0, wx.EXPAND | wx.ALL)])

  200. 

  201.         main_sizer.AddMany([(self._moon_phase_panel, 0, wx.EXPAND | wx.ALL, 5),

  202.                             (sizer, 0, wx.EXPAND | wx.ALL)])

  203.         self.SetSizer(main_sizer)

  204. 

  205.     def render(self):

  206.         if self.moon_phase is not None:

  207.             self._moon_phase_panel.set_moon_phase(self.moon_phase)

  208. 

  209.         if self._grid_ephemerides.NumberRows > 0:

  210.             self._grid_ephemerides.DeleteRows(numRows=self._grid_ephemerides.NumberRows)

  211. 

  212.         if self.ephemerides is not None and len(self.ephemerides) > 0:

  213.             for ephemeris in self.ephemerides:

  214.                 rise_time = ephemeris.rise_time.strftime(TIME_FORMAT) if ephemeris.rise_time is not None else ''

  215.                 culmination_time = ephemeris.culmination_time.strftime(TIME_FORMAT) if ephemeris.culmination_time is not None else ''

  216.                 set_time = ephemeris.set_time.strftime(TIME_FORMAT) if ephemeris.set_time is not None else ''

  217. 

  218.                 self._grid_ephemerides.AppendRows()

  219.                 row = self._grid_ephemerides.NumberRows - 1

  220.                 self._grid_ephemerides.SetRowLabelValue(row, ephemeris.object.name)

  221.                 self._grid_ephemerides.SetCellValue(row, 0, rise_time)

  222.                 self._grid_ephemerides.SetCellValue(row, 1, culmination_time)

  223.                 self._grid_ephemerides.SetCellValue(row, 2, set_time)

  224. 

  225.         self._grid_ephemerides.Show(len(self.ephemerides) > 0)

  226.         self._grid_ephemerides.AutoSize()

  227. 

  228.         if self.events is not None and len(self.events) > 0:

  229.             content = ''

  230.             for event in self.events:

  231.                 content += '\n' if content != '' else ''

  232.                 content += _('- {event_time}: {event_description}').format(

  233.                     event_time=event.start_time.strftime(TIME_FORMAT),

  234.                     event_description=event.get_description()

  235.                 )

  236. 

  237.             self._list_events.SetValue(content)

  238.         else:

  239.             self._list_events.SetValue(_('No events for this day.'))

  240. 

  241.         self.Layout()