import gi gi.require_version('Gtk', '3.0') from gi.repository import Gtk, GObject, Gdk from mods.db import db, Reception, Patient from mods.files import new_reception_win_file, reception_row_ui_str, reception_list_settings_win_file, open_reception_win_file from mods.settings import s_get_reception_list, s_set_reception_list from mods.root import builder from datetime import datetime, timedelta def add_reception(patient_id, reception_dt): with db.atomic(): patient = Patient.get_by_id(patient_id) return Reception.create(patient=patient, time=reception_dt) def get_reception_timelist(rec_date): s = s_get_reception_list() dstart = datetime(rec_date.year, rec_date.month, rec_date.day, s.day_start[0], s.day_start[1]) dend = datetime(rec_date.year, rec_date.month, rec_date.day, s.day_end[0], s.day_end[1]) work_day_minutes_total = (dend.hour - dstart.hour) * 60 + dend.minute - dstart.minute shift_minutes_range = range(0, work_day_minutes_total, s.interval) return [dstart + timedelta(minutes=x) for x in shift_minutes_range] class ReceptionRow(Gtk.ListBoxRow): @GObject.Property def db_id(self): return self._db_id @db_id.setter def db_id_setter(self, value): self._db_id = value @GObject.Property def datetime(self): return self._datetime @datetime.setter def datetime_setter(self, value): self._datetime = value @GObject.Property def scheduled(self): return self._scheduled @scheduled.setter def scheduled_setter(self, value): self._scheduled = value class NewReceptionPatientLabel(Gtk.Label): @GObject.Property def db_id(self): return self._db_id @db_id.setter def db_id_setter(self, value): self._db_id = value def get_reception(reception_id): with db.atomic(): return Reception.get_by_id(reception_id) def build_reception_row(reception_datetime): b = Gtk.Builder() b.add_from_string(reception_row_ui_str) win = b.get_object('win') box = b.get_object('reception_box') b.get_object('hour').set_text(reception_datetime.strftime('%H')) b.get_object('minute').set_text(reception_datetime.strftime('%M')) with db.atomic(): q = Reception.select().where(Reception.time == reception_datetime) row = ReceptionRow() if len(q): reception = q.get() row.props.scheduled = True row.props.db_id = reception.id reception_cont = b.get_object('reception_cont') reception_cont.props.border_width = 2 reception_cont.override_background_color(Gtk.StateFlags(0), Gdk.RGBA(red=0.6, green=0.8, blue=0.6, alpha=1.0)) reception_patient = Gtk.Label() reception_patient.set_markup(f'{" ".join([reception.patient.last_name, reception.patient.first_name, reception.patient.middle_name])}') #reception_patient.set_text(' '.join([reception.patient.last_name, reception.patient.first_name, reception.patient.middle_name])) reception_cont.add(reception_patient) else: row.props.scheduled = False win.remove(win.get_children()[0]) row.props.datetime = reception_datetime row.add(box) return row def create_open_reception_win(): b = Gtk.Builder() reception_list = builder.get_object('reception_list') reception_row = reception_list.get_selected_row() reception = get_reception(reception_row.props.db_id) class OpenReceptionWinHandler: pass b.add_from_file(open_reception_win_file) b.connect_signals(OpenReceptionWinHandler()) # patient_fio = b.get_object('patient_fio') patient_fio.set_markup(f'{" ".join([reception.patient.last_name, reception.patient.first_name, reception.patient.middle_name])}') reception_dt = b.get_object('reception_datetime') reception_dt.set_markup(f'{reception_row.props.datetime.strftime("%d.%m.%Y - %H:%M")}') # w = b.get_object('open_reception_window') return w def create_new_reception_win(): b = Gtk.Builder() reception_list = builder.get_object('reception_list') row = reception_list.get_selected_row() class NewReceptionWinHandler: def show_choose_patient_win(self, button): from mods.patients import create_choose_patient_win choose_patient_win = create_choose_patient_win(b) choose_patient_win.show_all() def save_reception(self, button): pat_label = patient_cont.get_children()[0] if pat_label.props.db_id < 0: return _ = add_reception(pat_label.props.db_id, row.props.datetime) w.destroy() redraw_reception_list(row.props.datetime) b.add_from_file(new_reception_win_file) b.connect_signals(NewReceptionWinHandler()) ##### patient_label = NewReceptionPatientLabel() patient_label.props.db_id = -1 #patient_label.set_markup('test') ##### patient_cont = b.get_object('patient') patient_cont.pack_start(patient_label, True, True, 0) patient_cont.reorder_child(patient_label, 0) w = b.get_object('new_reception_window') dt_label = b.get_object('datetime') dt_label.set_text(row.props.datetime.strftime('%d.%m.%Y - %H:%M')) return w def create_reception_list_settings_win(): b = Gtk.Builder() class ReceptionListSettingsHandler: def leading_zeroes(self, entry): adj = entry.get_adjustment() val = int(adj.get_value()) entry.set_text(f'{val:02d}') return True def save_settings(self, *a): start_hour = int(b.get_object('start_hour').get_value()) start_minute = int(b.get_object('start_minute').get_value()) end_hour = int(b.get_object('end_hour').get_value()) end_minute = int(b.get_object('end_minute').get_value()) interval = int(b.get_object('interval').get_value()) s_set_reception_list([start_hour, start_minute], [end_hour, end_minute], interval) redraw_reception_list(datetime.now()) b.get_object('reception_list_settings_win').close() b.add_from_file(reception_list_settings_win_file) b.connect_signals(ReceptionListSettingsHandler()) s = s_get_reception_list() b.get_object('start_hour').set_value(s.day_start[0]) b.get_object('start_minute').set_value(s.day_start[1]) b.get_object('end_hour').set_value(s.day_end[0]) b.get_object('end_minute').set_value(s.day_end[1]) b.get_object('interval').set_value(s.interval) w = b.get_object('reception_list_settings_win') return w def redraw_reception_list(selected_date): reception_timelist = get_reception_timelist(selected_date) reception_list = builder.get_object('reception_list') for c in reception_list.get_children(): reception_list.remove(c) for d in reception_timelist: reception_list.add(build_reception_row(d)) reception_list.show_all()