Apply black formatting

Signed-off-by: Philippe Ombredanne <pombredanne@nexb.com>

Author: pombredanne

gemfileparser/__init__.py

@@ -21,12 +21,12 @@ class Dependency(object):
     """
 
     def __init__(self):
-        self.name = ''
+        self.name = ""
         self.requirement = []
-        self.autorequire = ''
-        self.source = ''
+        self.autorequire = ""
+        self.source = ""
         self.parent = []
-        self.group = ''
+        self.group = ""
 
     def to_dict(self):
         return dict(
@@ -43,41 +43,44 @@ class GemfileParser(object):
     """
     Create a GemfileParser object to perform operations.
     """
+
     gemfile_regexes = collections.OrderedDict()
-    gemfile_regexes['source'] = re.compile(r"source:[ ]?(?P<source>[a-zA-Z:\/\.-]+)")
-    gemfile_regexes['git'] = re.compile(r"git:[ ]?(?P<git>[a-zA-Z:\/\.-]+)")
-    gemfile_regexes['platform'] = re.compile(r"platform:[ ]?(?P<platform>[a-zA-Z:\/\.-]+)")
-    gemfile_regexes['path'] = re.compile(r"path:[ ]?(?P<path>[a-zA-Z:\/\.-]+)")
-    gemfile_regexes['branch'] = re.compile(r"branch:[ ]?(?P<branch>[a-zA-Z:\/\.-]+)")
-    gemfile_regexes['autorequire'] = re.compile(r"require:[ ]?(?P<autorequire>[a-zA-Z:\/\.-]+)")
-    gemfile_regexes['group'] = re.compile(r"group:[ ]?(?P<group>[a-zA-Z:\/\.-]+)")
-    gemfile_regexes['name'] = re.compile(r"(?P<name>[a-zA-Z]+[\.0-9a-zA-Z _-]*)")
-    gemfile_regexes['requirement'] = re.compile(r"(?P<requirement>([>|<|=|~>|\d]+[ ]*[0-9\.\w]+[ ,]*)+)")
+    gemfile_regexes["source"] = re.compile(r"source:[ ]?(?P<source>[a-zA-Z:\/\.-]+)")
+    gemfile_regexes["git"] = re.compile(r"git:[ ]?(?P<git>[a-zA-Z:\/\.-]+)")
+    gemfile_regexes["platform"] = re.compile(r"platform:[ ]?(?P<platform>[a-zA-Z:\/\.-]+)")
+    gemfile_regexes["path"] = re.compile(r"path:[ ]?(?P<path>[a-zA-Z:\/\.-]+)")
+    gemfile_regexes["branch"] = re.compile(r"branch:[ ]?(?P<branch>[a-zA-Z:\/\.-]+)")
+    gemfile_regexes["autorequire"] = re.compile(r"require:[ ]?(?P<autorequire>[a-zA-Z:\/\.-]+)")
+    gemfile_regexes["group"] = re.compile(r"group:[ ]?(?P<group>[a-zA-Z:\/\.-]+)")
+    gemfile_regexes["name"] = re.compile(r"(?P<name>[a-zA-Z]+[\.0-9a-zA-Z _-]*)")
+    gemfile_regexes["requirement"] = re.compile(
+        r"(?P<requirement>([>|<|=|~>|\d]+[ ]*[0-9\.\w]+[ ,]*)+)"
+    )
 
     group_block_regex = re.compile(r"group[ ]?:[ ]?(?P<groupblock>.*?) do")
 
     gemspec_add_dvtdep_regex = re.compile(r".*add_development_dependency(?P<line>.*)")
     gemspec_add_rundep_regex = re.compile(r".*add_runtime_dependency(?P<line>.*)")
     gemspec_add_dep_regex = re.compile(r".*add_dependency(?P<line>.*)")
 
-    def __init__(self, filepath, appname=''):
+    def __init__(self, filepath, appname=""):
         self.filepath = filepath
 
-        self.current_group = 'runtime'
+        self.current_group = "runtime"
 
         self.appname = appname
         self.dependencies = {
-            'development': [],
-            'runtime': [],
-            'dependency': [],
-            'test': [],
-            'production': [],
-            'metrics': [],
+            "development": [],
+            "runtime": [],
+            "dependency": [],
+            "test": [],
+            "production": [],
+            "metrics": [],
         }
         with open(filepath) as gf:
             self.contents = gf.readlines()
 
-        self.gemspec = filepath.endswith(('.gemspec', '.podspec'))
+        self.gemspec = filepath.endswith((".gemspec", ".podspec"))
 
     @staticmethod
     def preprocess(line):
@@ -86,7 +89,7 @@ def preprocess(line):
         """
 
         if "#" in line:
-            line = line[:line.index('#')]
+            line = line[: line.index("#")]
         line = line.strip()
         return line
 
@@ -97,7 +100,7 @@ def parse_line(self, line):
 
         # csv requires a file-like object
         linefile = io.StringIO(line)
-        for line in csv.reader(linefile, delimiter=','):
+        for line in csv.reader(linefile, delimiter=","):
             column_list = []
             for column in line:
                 stripped_column = (
@@ -122,7 +125,7 @@ def parse_line(self, line):
                 for criteria, criteria_regex in GemfileParser.gemfile_regexes.items():
                     match = criteria_regex.match(column)
                     if match:
-                        if criteria == 'requirement':
+                        if criteria == "requirement":
                             dep.requirement.append(match.group(criteria))
                         else:
                             setattr(dep, criteria, match.group(criteria))
@@ -139,18 +142,18 @@ def parse_gemfile(self):
 
         for line in self.contents:
             line = self.preprocess(line)
-            if line == '' or line.startswith('source'):
+            if line == "" or line.startswith("source"):
                 continue
 
-            elif line.startswith('group'):
+            elif line.startswith("group"):
                 match = self.group_block_regex.match(line)
                 if match:
-                    self.current_group = match.group('groupblock')
+                    self.current_group = match.group("groupblock")
 
-            elif line.startswith('end'):
-                self.current_group = 'runtime'
+            elif line.startswith("end"):
+                self.current_group = "runtime"
 
-            elif line.startswith('gemspec'):
+            elif line.startswith("gemspec"):
                 # Gemfile contains a call to gemspec
                 gemfiledir = os.path.dirname(self.filepath)
                 gemspec_list = glob.glob(os.path.join(gemfiledir, "*.gemspec"))
@@ -160,7 +163,7 @@ def parse_gemfile(self):
                 gemspec_file = gemspec_list[0]
                 self.parse_gemspec(path=os.path.join(gemfiledir, gemspec_file))
 
-            elif line.startswith('gem '):
+            elif line.startswith("gem "):
                 line = line[3:]
                 self.parse_line(line)
 
@@ -176,17 +179,17 @@ def parse_gemspec(self, path=None):
             line = self.preprocess(line)
             match = self.gemspec_add_dvtdep_regex.match(line)
             if match:
-                self.current_group = 'development'
+                self.current_group = "development"
             else:
                 match = self.gemspec_add_rundep_regex.match(line)
                 if match:
-                    self.current_group = 'runtime'
+                    self.current_group = "runtime"
                 else:
                     match = self.gemspec_add_dep_regex.match(line)
                     if match:
-                        self.current_group = 'dependency'
+                        self.current_group = "dependency"
             if match:
-                line = match.group('line')
+                line = match.group("line")
                 self.parse_line(line)
         return self.dependencies
 
@@ -205,6 +208,7 @@ def command_line():
     A minimal command line entry point.
     """
     import sys
+
     if len(sys.argv) < 2:
         print("Usage : parsegemfile <input file>")
         sys.exit(0)

tests/test_gemfileparser.py

@@ -16,13 +16,12 @@ def check_gemparser_results(test_file, regen=False):
 
     gemparser = GemfileParser(test_file)
     dependencies = {
-        group: [dep.to_dict() for dep in deps]
-            for group, deps in gemparser.parse().items()
+        group: [dep.to_dict() for dep in deps] for group, deps in gemparser.parse().items()
     }
 
-    expected_file = test_file + '-expected.json'
+    expected_file = test_file + "-expected.json"
     if regen:
-        with open(expected_file, 'w') as o:
+        with open(expected_file, "w") as o:
             json.dump(dependencies, o, indent=2)
 
     with open(expected_file) as o:
@@ -32,39 +31,40 @@ def check_gemparser_results(test_file, regen=False):
 
 
 def test_source_only_gemfile():
-    check_gemparser_results('tests/Gemfile')
+    check_gemparser_results("tests/Gemfile")
 
 
 def test_gemfile_1():
-    check_gemparser_results('tests/Gemfile_1')
+    check_gemparser_results("tests/Gemfile_1")
 
 
 def test_gemfile_2():
-    check_gemparser_results('tests/Gemfile_2')
+    check_gemparser_results("tests/Gemfile_2")
 
 
 def test_gemfile_3():
-    check_gemparser_results('tests/Gemfile_3')
+    check_gemparser_results("tests/Gemfile_3")
 
 
 def test_gemfile_4():
-    check_gemparser_results('tests/Gemfile_4')
+    check_gemparser_results("tests/Gemfile_4")
 
 
 def test_gemfile_platforms():
-    check_gemparser_results('tests/Gemfile_5')
+    check_gemparser_results("tests/Gemfile_5")
 
 
 def test_gemspec_1():
-    check_gemparser_results('tests/sample.gemspec')
+    check_gemparser_results("tests/sample.gemspec")
 
 
 def test_gemspec_2():
-    check_gemparser_results('tests/address_standardization.gemspec')
+    check_gemparser_results("tests/address_standardization.gemspec")
 
 
 def test_gemspec_3():
-    check_gemparser_results('tests/arel.gemspec')
+    check_gemparser_results("tests/arel.gemspec")
+
 
 def test_gemspec_4():
-    check_gemparser_results('tests/logstash-mixin-ecs_compatibility_support.gemspec')
+    check_gemparser_results("tests/logstash-mixin-ecs_compatibility_support.gemspec")