From b0121146329edebce2421855b6d27b709d787614 Mon Sep 17 00:00:00 2001 From: Tranquilled Date: Tue, 31 Oct 2017 23:28:56 -0400 Subject: [PATCH] Created skeleton functions from which to create subfunctions; in reality, all of these functions should be replaced. However, changing function names throughout should be its own issue. --- smtpapi/__init__.py | 27 +++++++++++++++------------ 1 file changed, 15 insertions(+), 12 deletions(-) diff --git a/smtpapi/__init__.py b/smtpapi/__init__.py index 43cbb59..75a04b0 100644 --- a/smtpapi/__init__.py +++ b/smtpapi/__init__.py @@ -34,34 +34,37 @@ def add_substitution(self, key, value): def set_substitutions(self, subs): self.data['sub'] = subs + + def add_attr(self, key, value, attr): + if attr not in self.data: + self.data[attr] = {} + self.data[attr][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_attr(self, key, value, 'unique_args') def set_unique_args(self, value): self.data['unique_args'] = value + def append_attr(self, value, attr): + if attr not in self.data: + self.data[attr] = [] + self.data[attr].append(value) + def add_category(self, category): - if 'category' not in self.data: - self.data['category'] = [] - self.data['category'].append(category) + append_attr(self, category, '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 + add_attr(self, key, section, 'section') def set_sections(self, value): self.data['section'] = value def add_send_each_at(self, time): - if 'send_each_at' not in self.data: - self.data['send_each_at'] = [] - self.data['send_each_at'].append(time) + append_attr(self, time, 'send_each_at') def set_send_each_at(self, time): self.data['send_each_at'] = time