Run Python black on repo

.editorconfig

@@ -0,0 +1,29 @@
+# This file is for unifying the coding style for different editors and IDEs.
+# More information at http://EditorConfig.org
+root = true
+
+[*]
+charset = utf-8
+end_of_line = lf
+indent_size = 4
+indent_style = space
+insert_final_newline = true
+max_line_length = 120
+trim_trailing_whitespace = true
+
+[*.md]
+trim_trailing_whitespace = false
+
+[*.py]
+include_trailing_comma = True
+indent_size = 4
+max_line_length = 120
+multi_line_output = 5
+not_skip = __init__.py
+sections = FUTURE,STDLIB,THIRDPARTY,FIRSTPARTY,LOCALFOLDER
+
+[*.yml,*.yaml]
+indent_size = 2
+
+[Makefile]
+indent_style = tab

.gitignore

@@ -30,4 +30,4 @@ pip-log.txt
 #Vim swp
 *.swp
 .idea
-.*
+.DS_Store

ach/builder.py

@@ -2,8 +2,12 @@
 from datetime import datetime, timedelta
 
 from .data_types import (
-    Header, FileControl, BatchHeader,
-    BatchControl, EntryDetail, AddendaRecord
+    Header,
+    FileControl,
+    BatchHeader,
+    BatchControl,
+    EntryDetail,
+    AddendaRecord,
 )
 
 
@@ -23,14 +27,16 @@ def __init__(self, file_id_mod, settings):
 
         try:
             company_name = settings.get(
-                'company_name',
-                settings['immediate_org_name'],
+                "company_name",
+                settings["immediate_org_name"],
             )
-            self.settings['company_name'] = company_name
+            self.settings["company_name"] = company_name
             self.header = Header(
-                settings['immediate_dest'],
-                settings['immediate_org'], file_id_mod,
-                settings['immediate_dest_name'], settings['immediate_org_name']
+                settings["immediate_dest"],
+                settings["immediate_org"],
+                file_id_mod,
+                settings["immediate_dest_name"],
+                settings["immediate_org_name"],
             )
         except KeyError:
             raise Exception(
@@ -40,9 +46,17 @@ def __init__(self, file_id_mod, settings):
 
         self.batches = list()
 
-    def add_batch(self, std_ent_cls_code, batch_entries=None,
-                  credits=True, debits=False, eff_ent_date=None,
-                  company_id=None, entry_desc=None, company_name=None):
+    def add_batch(
+        self,
+        std_ent_cls_code,
+        batch_entries=None,
+        credits=True,
+        debits=False,
+        eff_ent_date=None,
+        company_id=None,
+        entry_desc=None,
+        company_name=None,
+    ):
         """
         Use this to add batches to the file. For valid std_ent_cls_codes see:
         http://en.wikipedia.org/wiki/Automated_Clearing_House#SEC_codes
@@ -60,23 +74,23 @@ def add_batch(self, std_ent_cls_code, batch_entries=None,
             eff_ent_date = datetime.today() + timedelta(days=1)
 
         if credits and debits:
-            serv_cls_code = '200'
+            serv_cls_code = "200"
         elif credits:
-            serv_cls_code = '220'
+            serv_cls_code = "220"
         elif debits:
-            serv_cls_code = '225'
+            serv_cls_code = "225"
 
         batch_header = BatchHeader(
             serv_cls_code=serv_cls_code,
             batch_id=batch_count,
-            company_id=company_id or self.settings['company_id'],
+            company_id=company_id or self.settings["company_id"],
             std_ent_cls_code=std_ent_cls_code,
             entry_desc=entry_desc,
-            desc_date='',
-            eff_ent_date=eff_ent_date.strftime('%y%m%d'),  # YYMMDD
-            orig_stat_code='1',
-            orig_dfi_id=self.settings['immediate_dest'][:8],
-            company_name=(company_name or self.settings['company_name'])[:16],
+            desc_date="",
+            eff_ent_date=eff_ent_date.strftime("%y%m%d"),  # YYMMDD
+            orig_stat_code="1",
+            orig_dfi_id=self.settings["immediate_dest"][:8],
+            company_name=(company_name or self.settings["company_name"])[:16],
         )
 
         entries, failed_entry_errors = [], []
@@ -86,24 +100,23 @@ def add_batch(self, std_ent_cls_code, batch_entries=None,
             try:
                 entry = EntryDetail(
                     std_ent_cls_code=std_ent_cls_code,
-                    id_number=record.get('id_number', ''),
+                    id_number=record.get("id_number", ""),
                 )
 
-                entry.transaction_code = record.get('type')
-                entry.recv_dfi_id = record.get('routing_number')
+                entry.transaction_code = record.get("type")
+                entry.recv_dfi_id = record.get("routing_number")
 
-                if len(record['routing_number']) < 9:
+                if len(record["routing_number"]) < 9:
                     entry.calc_check_digit()
                 else:
-                    entry.check_digit = record['routing_number'][8]
+                    entry.check_digit = record["routing_number"][8]
 
-                entry.dfi_acnt_num = record['account_number']
-                entry.amount = int(round(float(record['amount']) * 100))
-                entry.ind_name = record['name'].upper()[:22]
-                entry.trace_num = self.settings['immediate_dest'][:8] \
-                    + entry.validate_numeric_field(entry_counter, 7)
+                entry.dfi_acnt_num = record["account_number"]
+                entry.amount = int(round(float(record["amount"]) * 100))
+                entry.ind_name = record["name"].upper()[:22]
+                entry.trace_num = self.settings["immediate_dest"][:8] + entry.validate_numeric_field(entry_counter, 7)
 
-                entries.append((entry, record.get('addenda', [])))
+                entries.append((entry, record.get("addenda", [])))
                 entry_counter += 1
             except Exception as e:
                 failed_entry_errors.append((record, e))
@@ -122,8 +135,12 @@ def set_control(self):
         credit_amount = self.get_credit_amount(self.batches)
 
         self.control = FileControl(
-            batch_count, block_count, entadd_count,
-            entry_hash, debit_amount, credit_amount
+            batch_count,
+            block_count,
+            entadd_count,
+            entry_hash,
+            debit_amount,
+            credit_amount,
         )
 
     def get_block_count(self, batches):
@@ -138,8 +155,7 @@ def get_lines(self, batches):
 
         entadd_count = self.get_entadd_count(batches)
 
-        lines = header_count + control_count + batch_header_count \
-            + batch_footer_count + entadd_count
+        lines = header_count + control_count + batch_header_count + batch_footer_count + entadd_count
 
         return lines
 
@@ -177,16 +193,15 @@ def get_credit_amount(self, batches):
         credit_amount = 0
 
         for batch in batches:
-            credit_amount = credit_amount + \
-                int(batch.batch_control.credit_amount)
+            credit_amount = credit_amount + int(batch.batch_control.credit_amount)
 
         return credit_amount
 
     def get_nines(self, rows, line_ending):
-        nines = ''
+        nines = ""
 
         for i in range(rows):
-            nines += '9'*94
+            nines += "9" * 94
             if i == rows - 1:
                 continue
             nines += line_ending
@@ -195,12 +210,12 @@ def get_nines(self, rows, line_ending):
 
     def get_entry_desc(self, std_ent_cls_code):
 
-        if std_ent_cls_code == 'PPD':
-            entry_desc = 'PAYROLL'
-        elif std_ent_cls_code == 'CCD':
-            entry_desc = 'DUES'
+        if std_ent_cls_code == "PPD":
+            entry_desc = "PAYROLL"
+        elif std_ent_cls_code == "CCD":
+            entry_desc = "DUES"
         else:
-            entry_desc = 'OTHER'
+            entry_desc = "OTHER"
 
         return entry_desc
 
@@ -252,7 +267,7 @@ def __init__(self, batch_header, entries):
             entadd_count += len(addenda)
             self.entries.append(FileEntry(entry, addenda))
 
-        #set up batch_control
+        # set up batch_control
 
         batch_control = BatchControl(self.batch_header.serv_cls_code)
 
@@ -285,8 +300,7 @@ def get_debit_amount(self, entries):
         debit_amount = 0
 
         for entry in entries:
-            if str(entry.entry_detail.transaction_code) in \
-                    ['27', '37', '28', '38']:
+            if str(entry.entry_detail.transaction_code) in ["27", "37", "28", "38"]:
                 debit_amount = debit_amount + int(entry.entry_detail.amount)
 
         return debit_amount
@@ -295,8 +309,7 @@ def get_credit_amount(self, entries):
         credit_amount = 0
 
         for entry in entries:
-            if str(entry.entry_detail.transaction_code) in \
-                    ['22', '32', '23', '33']:
+            if str(entry.entry_detail.transaction_code) in ["22", "32", "23", "33"]:
                 credit_amount += int(entry.entry_detail.amount)
 
         return credit_amount
@@ -339,9 +352,9 @@ def __init__(self, entry_detail, addenda_record=[]):
             self.addenda_record.append(
                 AddendaRecord(
                     self.entry_detail.std_ent_cls_code,
-                    pmt_rel_info=addenda.get('payment_related_info').upper(),
+                    pmt_rel_info=addenda.get("payment_related_info").upper(),
                     add_seq_num=index + 1,
-                    ent_det_seq_num=entry_detail.trace_num[-7:]
+                    ent_det_seq_num=entry_detail.trace_num[-7:],
                 )
             )