From b3cdf4d254495d3f2a1b8f08dff869576959c763 Mon Sep 17 00:00:00 2001
From: Anisha Lunawat <lunawat.anisha@gmail.com>
Date: Tue, 31 Oct 2017 11:51:29 +0530
Subject: [PATCH] Update __init__.py

---
 smtpapi/__init__.py | 15 ++++++++-------
 1 file changed, 8 insertions(+), 7 deletions(-)

diff --git a/smtpapi/__init__.py b/smtpapi/__init__.py
index 43cbb59..de93b40 100644
--- a/smtpapi/__init__.py
+++ b/smtpapi/__init__.py
@@ -34,10 +34,13 @@ def add_substitution(self, key, value):
     def set_substitutions(self, subs):
         self.data['sub'] = subs
 
+    def add(self,key,value,arg):
+        if arg not in self.data:
+            self.data[arg] = {}
+        self.data[arg][key] = value
+        
     def add_unique_arg(self, key, value):
-        if 'unique_args' not in self.data:
-            self.data['unique_args'] = {}
-        self.data['unique_args'][key] = value
+        add(self,key,value,'unique_arg')
 
     def set_unique_args(self, value):
         self.data['unique_args'] = value
@@ -50,10 +53,8 @@ def add_category(self, category):
     def set_categories(self, category):
         self.data['category'] = category
 
-    def add_section(self, key, section):
-        if 'section' not in self.data:
-            self.data['section'] = {}
-        self.data['section'][key] = section
+    def add_section(self, key, value):
+        add(self,key,value,'section')
 
     def set_sections(self, value):
         self.data['section'] = value