From 197961a6a328717771e6b15c981b36c041a8548d Mon Sep 17 00:00:00 2001
From: Lefteris Chatzimparmpas <lefteris@doist.com>
Date: Wed, 29 Mar 2017 09:41:19 +0000
Subject: [PATCH] Add emails to objects functions.

---
 todoist/api.py             |  2 ++
 todoist/managers/emails.py | 20 ++++++++++++++++++++
 2 files changed, 22 insertions(+)
 create mode 100644 todoist/managers/emails.py

diff --git a/todoist/api.py b/todoist/api.py
index 91a63bf..453bbff 100644
--- a/todoist/api.py
+++ b/todoist/api.py
@@ -26,6 +26,7 @@
 from todoist.managers.templates import TemplatesManager
 from todoist.managers.backups import BackupsManager
 from todoist.managers.quick import QuickManager
+from todoist.managers.emails import EmailsManager
 
 
 class SyncError(Exception):
@@ -82,6 +83,7 @@ def __init__(self,
         self.templates = TemplatesManager(self)
         self.backups = BackupsManager(self)
         self.quick = QuickManager(self)
+        self.emails = EmailsManager(self)
 
         if cache:  # Read and write user state on local disk cache
             self.cache = os.path.expanduser(cache)
diff --git a/todoist/managers/emails.py b/todoist/managers/emails.py
new file mode 100644
index 0000000..c670790
--- /dev/null
+++ b/todoist/managers/emails.py
@@ -0,0 +1,20 @@
+# -*- coding: utf-8 -*-
+from .generic import Manager
+
+
+class EmailsManager(Manager):
+    def get_or_create(self, obj_type, obj_id, **kwargs):
+        """
+        Get or create email to an object.
+        """
+        params = {'token': self.token, 'obj_type': obj_type, 'obj_id': obj_id}
+        params.update(kwargs)
+        return self.api._get('emails/get_or_create', params=params)
+
+    def disable(self, obj_type, obj_id, **kwargs):
+        """
+        Disable email to an object.
+        """
+        params = {'token': self.token, 'obj_type': obj_type, 'obj_id': obj_id}
+        params.update(kwargs)
+        return self.api._get('emails/disable', params=params)