production init

front/msim/misc.py

@@ -0,0 +1,66 @@
+from enum import IntEnum
+from typing import Dict
+
+from core.backend import BackendSession
+from core.models import ContactList, Contact
+
+
+class CommandBitFlag(IntEnum):
+	CallbackReply = 256,
+	CallbackAction = 512,
+	CallbackError = 1024
+
+
+def get_grouped_contacts(bs: BackendSession) -> Dict[str, list[Contact]]:
+	user = bs.user
+	detail = user.detail
+
+	contacts = list({
+		*detail.get_contacts_by_list(ContactList.AL),
+		*detail.get_contacts_by_list(ContactList.FL)
+	})
+	grouped_contacts = {}
+
+	# loop through every contact group
+	for group in detail._groups_by_id.values():
+		grouped_contacts[group.name] = []
+
+		# loop through every contact in this group
+		for contact in contacts:
+			for grp in contact._groups:
+				if grp.id == group.id:
+					grouped_contacts[group.name].append(contact)
+
+	# handle any contacts that might not be apart of any group
+	if ungrouped_contacts := [contact for contact in contacts if not contact._groups]:
+		grouped_contacts['(No Group)'] = []
+
+		for contact in ungrouped_contacts:
+			grouped_contacts['(No Group)'].append(contact)
+
+	return grouped_contacts
+
+
+def get_contact_groups(bs: BackendSession) -> list[str]:
+	groups = []
+
+	user = bs.user
+	detail = user.detail
+
+	contacts = list({
+		*detail.get_contacts_by_list(ContactList.AL),
+		*detail.get_contacts_by_list(ContactList.FL)
+	})
+
+	for group in detail._groups_by_id.values():
+		groups.append(group.name)
+
+	# add a fake `(No Group)` group if there is any contacts not apart of any group
+	if any(not contact._groups for contact in contacts):
+		groups.append('(No Group)')
+
+	return groups
+
+
+def unmarshal_msim_dict(data: str) -> Dict:
+	return dict(pair.split('=', 1) for pair in data.split('\x1c'))