Added hooks for proper CC/BCC handling

iandouglas · iandouglas · commit 3fdb3de31c1e · 2012-10-20T23:24:16.000-07:00
diff --git a/sendgrid/header.py b/sendgrid/header.py
@@ -16,6 +16,22 @@ def add_to(self, to):
         else:
             self.data['to'] += to
 
+    def add_cc(self, cc):
+        if 'cc' not in self.data:
+            self.data['cc'] = []
+        if type(cc) is str:
+            self.data['cc'] += [cc]
+        else:
+            self.data['cc'] += cc
+
+    def add_bcc(self, bcc):
+        if 'bcc' not in self.data:
+            self.data['bcc'] = []
+        if type(bcc) is str:
+            self.data['bcc'] += [bcc]
+        else:
+            self.data['bcc'] += bcc
+
     def set_replyto(self, replyto):
         self.data['reply_to'] = replyto
 
diff --git a/sendgrid/message.py b/sendgrid/message.py
@@ -120,6 +120,8 @@ def add_cc(self, recipients):
         else:
             self.cc += recipients
 
+        self.header.add_cc(recipients)
+        
         return self
 
     def add_bcc(self, recipients):
@@ -138,6 +140,8 @@ def add_bcc(self, recipients):
         else:
             self.bcc += recipients
 
+        self.header.add_bcc(recipients)
+        
         return self
 
     def add_attachment(self, name, file, cid=None):
