Add feature: Contact manager

2023-09-21 20:51:21 -07:00
parent f7a47a06ec
commit 270debef67
9 changed files with 300 additions and 42 deletions
@@ -1,34 +1,137 @@
 from typing import List
+import toml

 class Contact:
-    def __init__(self,name:str) -> None:
+    def __init__(self, name, phone=None, email=None, telegram=None,added_by=None, tags=[], notes=""):
        self.name = name
-        self.tags = []
+        self.phone = phone
+        self.email = email
+        self.telegram = telegram
+        self.added_by = added_by
+        self.tags = tags
+        self.notes = notes

-    def is_zone_owner(self,zone_id=None) -> bool:
-        return True
+    def to_dict(self):
+        return {
+            "name": self.name,
+            "phone": self.phone,
+            "email": self.email,
+            "telegram" : self.telegram,

-    def get_tags(self)->List[str]:
-        return self.tags
+            "added_by": self.added_by,
+            "tags": self.tags,
+            "notes": self.notes
+        }

-    def get_name(self)->str:
-        return self.name
+    @classmethod
+    def from_dict(cls, data):
+        return Contact(data.get("name"), data.get("phone"), data.get("email"), data.get("telegram"),data.get("added_by"), data.get("tags"), data.get("notes"))

+class FamilyMember(Contact):
+    def __init__(self, name, relationship,phone=None, email=None,telegram=None):
+        super().__init__(name, phone, email, telegram)
+        self.name = name
+        self.relationship = relationship  
+        self.is_family_member = True
+
+    def to_dict(self):
+        result = super().to_dict()
+        result["relationship"] = self.relationship
+        return result
+
+    @classmethod
+    def from_dict(cls, data):
+        return FamilyMember(data.get("name"),data.get("relationship"),data.get("phone"), data.get("email"),data.get("telegram"))

 class ContactManager:
    _instance = None
    @classmethod
-    def get_instance(cls):
+    def get_instance(cls,filename=None) -> "ContactManager":
        if cls._instance is None:
-            cls._instance = ContactManager()
+            cls._instance = ContactManager(filename)
        return cls._instance

-    def __init__(self) -> None:
-        self.contacts = {}
-        self.contacts["liuzhicong"] = Contact("liuzhicong")
+    def __init__(self, filename="contacts.toml"):
+        self.filename = filename
+        self.contacts = []
+        self.family_members = []    

-    #def get_by_addr(self,addr:str) -> Contact:
-    #    pass
+        self.is_auto_create_contact_from_telegram = True
+    
+    def load_data(self):
+        try:
+            with open(self.filename, "r") as f:
+                config =  toml.load(f)
+                return self.load_from_config(config)
+        except FileNotFoundError:
+            return {}
+        
+    def load_from_config(self,config_data:dict):
+        self.contacts = [Contact.from_dict(item) for item in config_data.get("contacts", [])]
+        self.family_members = [FamilyMember.from_dict(item) for item in config_data.get("family_members", [])]
+    
+    def save_data(self):
+        data = {
+            "contacts": [contact.to_dict() for contact in self.contacts],
+            "family_members": [member.to_dict() for member in self.family_members]
+        }
+        with open(self.filename, "w") as f:
+            toml.dump(data, f)

-    def get_by_name(self,name:str) -> Contact:
-        return self.contacts.get(name)
+    def add_contact(self, name:str, new_contact:Contact):
+        assert name == new_contact.name
+        self.contacts.append(new_contact)
+        self.save_data()
+
+    def remove_contact(self, name:str):
+        self.contacts = [contact for contact in self.contacts if contact.name != name]
+        self.save_data()
+
+    def find_contact_by_name(self, name:str):
+        for contact in self.contacts:
+            if contact.name == name:
+                return contact
+            
+        for member in self.family_members:
+            if member.name == name:
+                return member
+        return None
+    
+    def find_contact_by_telegram(self, telegram:str):
+        for contact in self.contacts:
+            if contact.telegram == telegram:
+                return contact
+        for member in self.family_members:
+            if member.telegram == telegram:
+                return member            
+        return None
+
+    def find_contact_by_email(self, email:str):
+        for contact in self.contacts:
+            if contact.email == email:
+                return contact
+        for member in self.family_members:
+            if member.email == email:
+                return member   
+        return None
+
+    def find_contact_by_phone(self, phone:str):
+        for contact in self.contacts:
+            if contact.phone == phone:
+                return contact
+        for member in self.family_members:
+            if member.phone == phone:
+                return member   
+        return None
+
+
+    def add_family_member(self, name, new_member:FamilyMember):
+        assert name == new_member.name
+        self.family_members.append(new_member)
+        self.save_data()
+
+    def list_contacts(self):
+        return self.contacts
+
+    def list_family_members(self):
+        return self.family_members