def set_data_for(self, group_id=None, search=None):
if isinstance(group_id, int):
qs = ContactGroup.select().where(ContactGroup.group ==
Group.get(Group.id == group_id))
self.data = [("", contact_gp.contact.name, contact_gp.contact.number)
for contact_gp in qs]
else:
qs = Contact.select()
if search:
print(search)
qs = qs.where(Contact.number.contains(search)
| Contact.name.contains(search))
print(qs)
self.data = [("", contact.name, contact.number) for contact in qs]
def del_grp(self, grp_ct):
group = Group.get(Group.name == grp_ct.group.name)
contactgrp = ContactGroup.select().where(ContactGroup.group == group,
ContactGroup.contact == self.contact).get()
contactgrp.delete_instance()
self.refresh()
