98 lines
3.5 KiB
Python
98 lines
3.5 KiB
Python
import gi
|
|
gi.require_version('Gtk', '3.0')
|
|
from gi.repository import Gtk, GObject
|
|
from mods.files import choose_diagnosis_win_file, diagnosis_row_ui_str
|
|
from mods.utils import ConditionalFilter
|
|
from mods.db import DiagnosisCategory, Diagnosis, DiagnosisIndex, db
|
|
import re
|
|
|
|
class DiagnosisFilter(ConditionalFilter):
|
|
def filter(self, cat, query):
|
|
if query != self.fstr:
|
|
self.fstr = query
|
|
self.ids = list(map(lambda x:x.id, self.search_func(cat, query)))
|
|
return self.ids
|
|
|
|
class DiagnosisRow(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
|
|
|
|
def cats_subcats(cat):
|
|
cats = [cat]
|
|
def append_cat(cat_obj):
|
|
cats.append(cat_obj)
|
|
for c in cat_obj.subcat:
|
|
append_cat(c)
|
|
for c in cat.subcat:
|
|
append_cat(c)
|
|
return cats
|
|
|
|
def get_categories(parent=None):
|
|
with db.atomic():
|
|
return DiagnosisCategory.select().where(DiagnosisCategory.parent == parent)
|
|
def get_diagnosises(cat=None):
|
|
with db.atomic():
|
|
if cat:
|
|
d = list(Diagnosis.select().where(Diagnosis.category.in_(cats_subcats(cat))).order_by(Diagnosis.code))
|
|
return d
|
|
else:
|
|
return Diagnosis.select()
|
|
def get_category(cat_id):
|
|
with db.atomic():
|
|
return DiagnosisCategory.get_by_id(cat_id)
|
|
def search_diagnosises(cat, q):
|
|
q = re.sub(r'\s+', ' ', q).strip()
|
|
q = ' '.join([ f'{x}*' for x in q.split(' ')])
|
|
with db.atomic():
|
|
return (Diagnosis.select()
|
|
.join(
|
|
DiagnosisIndex,
|
|
on=(Diagnosis.id == DiagnosisIndex.rowid))
|
|
.where((Diagnosis.category.in_(cats_subcats(cat))) & (DiagnosisIndex.match(q)))
|
|
.order_by(DiagnosisIndex.bm25()))
|
|
diagnosis_filter = DiagnosisFilter(search_diagnosises)
|
|
########################################################
|
|
def build_diagnosis_row(diagnosis):
|
|
b = Gtk.Builder()
|
|
b.add_from_string(diagnosis_row_ui_str)
|
|
win = b.get_object('win')
|
|
box = b.get_object('diagnosis_box')
|
|
b.get_object('title').set_markup(f'<span size="x-large">{diagnosis.code:<8} {diagnosis.title}</span>')
|
|
win.remove(win.get_children()[0])
|
|
row = DiagnosisRow()
|
|
row.props.db_id = diagnosis.id
|
|
row.add(box)
|
|
return row
|
|
|
|
def create_choose_diagnosis_win(reception):
|
|
b = Gtk.Builder()
|
|
class ChooseDiagnosisWinHandler:
|
|
def cat_selection_changed(self, cat_selection):
|
|
sel_iter = cat_selection.get_selected()[1]
|
|
model = cat_selection.get_tree_view().get_model()
|
|
category = get_category(model.get_value(sel_iter, 1))
|
|
self.redraw_diagnosis_list(category)
|
|
def redraw_diagnosis_list(self, cat):
|
|
for c in diagnosis_list.get_children():
|
|
diagnosis_list.remove(c)
|
|
for d in get_diagnosises(cat):
|
|
diagnosis_list.add(build_diagnosis_row(d))
|
|
diagnosis_list.show_all()
|
|
b.add_from_file(choose_diagnosis_win_file)
|
|
b.connect_signals(ChooseDiagnosisWinHandler())
|
|
cat_liststore = b.get_object('category_list')
|
|
def insert_cat(cat, parent=None):
|
|
cur_parent = cat_liststore.append(parent, [cat.title, cat.id])
|
|
for c in get_categories(cat):
|
|
insert_cat(c, cur_parent)
|
|
for c in get_categories():
|
|
insert_cat(c)
|
|
diagnosis_list = b.get_object('diagnosis_list')
|
|
diagnosis_list.set_sort_func(None, None)
|
|
w = b.get_object('choose_diagnosis_window')
|
|
return w
|