pep8 fixes

fecfile/cache.py

@@ -7,6 +7,7 @@
 TYPE_CACHE_KEY = "%s:%s:%s"
 TYPE_CACHE = {}
 
+
 class FecParserMissingMappingError(Exception):
     """when a line in an FEC filing doesn't have a form/version mapping"""
     def __init__(self, opts, msg=None):
@@ -18,10 +19,11 @@ def __init__(self, opts, msg=None):
             )
         super(FecParserMissingMappingError, self).__init__(msg)
 
+
 def getMapping_from_regex(mappings, form, version):
     """ Raises FecParserMissingMappingError if missing"""
 
-    for mapping in mappings.keys(): 
+    for mapping in mappings.keys():
         if re.match(mapping, form, re.IGNORECASE):
             versions = mappings[mapping].keys()
             for v in versions:
@@ -33,16 +35,18 @@ def getMapping_from_regex(mappings, form, version):
         'version': version,
     })
 
+
 def getMapping(mappings, form, version):
     """ Tries to find the mapping from cache before looking it up w regex """
-    key = MAPPING_CACHE_KEY % (form,version)
+    key = MAPPING_CACHE_KEY % (form, version)
     try:
         mapping = MAPPING_CACHE[key]
     except KeyError:
         mapping = getMapping_from_regex(mappings, form, version)
         MAPPING_CACHE[key] = mapping
     return mapping
 
+
 def getTypeMapping_from_regex(types, form, version, field):
     """ Tries to find the mapping from cache before looking it up w regex """
     for mapping in types.keys():
@@ -58,6 +62,7 @@ def getTypeMapping_from_regex(types, form, version, field):
                             return prop
     return None
 
+
 def getTypeMapping(types, form, version, field):
     """ caches the mapping to dict """
     key = TYPE_CACHE_KEY % (form, version, field)

fecfile/fecparser.py

@@ -3,11 +3,11 @@
 import csv
 import json
 import os
-import re
 import warnings
 
 from .cache import getTypeMapping, getMapping
 
+
 class FecParserTypeWarning(UserWarning):
     """when data in an FEC filing doesn't match types.json"""
     pass
@@ -118,6 +118,7 @@ def parse_header(lines):
     parsed = parse_line(lines[0], fields[1], 0)
     return parsed, fields[1], 1
 
+
 def parse_line(line, version, line_num=None):
     ascii_separator = True
     if version is None or version[0] in comma_versions:
@@ -126,7 +127,6 @@ def parse_line(line, version, line_num=None):
     if len(fields) < 2:
         return None
     form = fields[0]
-
     this_version_mapping = getMapping(mappings, form, version)
     out = {}
     for i in range(len(this_version_mapping)):
@@ -135,8 +135,10 @@ def parse_line(line, version, line_num=None):
         out[k] = getTyped(form, version, k, val, line_num)
     return out
 
+
 nones = ['none', 'n/a']
 
+
 def getTyped(form, version, field, value, line_num):
     prop = getTypeMapping(types, form, version, field)
     if prop:
@@ -175,6 +177,7 @@ def getTyped(form, version, field, value, line_num):
             return None
     return value
 
+
 def print_example(parsed):
     out = {'filing': parsed['filing'], 'itemizations': {}}
     for k in parsed['itemizations'].keys():

speedtests.py

@@ -14,22 +14,19 @@ def speed_test(filepath):
     with open(filepath) as file:
         linecount = 0
         version = None
-        header = None
         for line in file:
-            linecount+=1
+            linecount += 1
             if version is None:
                 results = fecfile.parse_header(line)
-                header = results[0]
                 version = results[1]
             else:
                 parsed = fecfile.parse_line(line, version)
-
                 if not parsed:
                     print("** not parsed %s" % line)
-                else:   
+                else:
                     # count the form type, if given
                     try:
-                        formtypecount.update({parsed['form_type'].upper():1})
+                        formtypecount.update({parsed['form_type'].upper(): 1})
                     except KeyError:
                         continue