Kosmorro
/
kosmorro
镜像来自 https://github.com/Kosmorro/kosmorro
1
0
複製 0
程式碼 問題管理 版本發佈 36 Activity
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
222 Commit
5 分支
34 MiB
 
 
 
 
目錄樹: e6d7de82f4
分支列表 標籤列表
${ item.name }
Create branch ${ searchTerm }
from 'e6d7de82f4'
${ noResults }
kosmorro/kosmorrolib/gui/panel.py

242 line
9.8 KiB
原始文件 Blame 文件歷史 

  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()