adding mails query, tags and contacts to py-fake-service

1 files changed, 19 insertions, 0 deletions

diff --git a/py-fake-service/app/adapter/contacts.py b/py-fake-service/app/adapter/contacts.py
new file mode 100644
index 00000000..45aa2177
--- /dev/null
+++ b/py-fake-service/app/adapter/contacts.py
@@ -0,0 +1,19 @@
+import re   
+
+class Contacts:
+    def __init__(self):
+        self.contacts = []
+
+    def add(self, mbox_mail):
+        contact = mbox_mail.get_from()
+        self.contacts.append(Contact(contact))
+
+    def search(self, query):
+        contacts_query = re.compile(query)
+        return [contact.__dict__ for contact in self.contacts if contacts_query.match(contact.addresses[0])]
+
+class Contact:
+    def __init__(self, contact):
+        self.addresses = [contact]
+        self.name = ''
+