adding --list support. no tests yet

importanize/__main__.py

@@ -7,6 +7,7 @@
 import os
 import sys
 from fnmatch import fnmatch
+from itertools import chain
 from stat import S_ISFIFO
 
 import pathlib2 as pathlib
@@ -177,6 +178,12 @@ def __bool__(self):
     help='When used CI mode will check if file contains expected '
          'imports as per importanize configuration.'
 )
+parser.add_argument(
+    '--list',
+    action='store_true',
+    default=False,
+    help='List all dependencies found in all parsed files'
+)
 parser.add_argument(
     '--version',
     action='store_true',
@@ -222,6 +229,9 @@ def run_importanize_on_text(text, config, args):
         for j in parse_statements([([i], [first_import_line_number])]):
             groups.add_statement_to_group(j)
 
+    if args.list:
+        return groups
+
     formatted_imports = groups.formatted(
         formatter=formatter,
         length=args.length or config.get('length') or DEFAULT_LENGTH,
@@ -275,6 +285,10 @@ def run(source, config, args, path=None):
             raise
 
         else:
+            if args.list:
+                yield organized
+                return
+
             if args.print and args.header and path:
                 print('=' * len(six.text_type(path)))
                 print(six.text_type(path))
@@ -298,7 +312,7 @@ def run(source, config, args, path=None):
                         msg += ' {}'.format(path)
                     log.info(msg)
 
-            return organized
+            yield organized
 
     elif source.is_file():
         if args.subconfig:
@@ -315,7 +329,8 @@ def run(source, config, args, path=None):
                 return
 
         text = source.read_text('utf-8')
-        return run(text, config, args, source)
+        for i in run(text, config, args, source):
+            yield i
 
     elif source.is_dir():
         if config.get('exclude'):
@@ -333,7 +348,8 @@ def run(source, config, args, path=None):
         all_successes = True
         for f in files:
             try:
-                run(f, config, args, f)
+                for i in run(f, config, args, f):
+                    yield i
             except CIFailure:
                 all_successes = False
 
@@ -388,16 +404,31 @@ def main(args=None):
         args.header = False
 
     all_successes = True
+    all_groups = []
 
     for p in to_importanize:
         try:
-            run(p, config, args)
+            all_groups += [i for i in run(p, config, args)]
         except CIFailure:
             all_successes = False
         except Exception:
             log.exception('Error running importanize')
             return 1
 
+    if args.list:
+        groups = ImportGroups()
+        for c in config['groups']:
+            groups.add_group(c)
+        statements = chain(*(i.statements for i in chain(*all_groups)))
+        for s in statements:
+            groups.add_statement_to_group(s)
+        for g in groups:
+            print(g.config['type'])
+            print('-' * len(g.config['type']))
+            for s in g.unique_statements:
+                print(s)
+            print()
+
     return int(not all_successes)
 
 

importanize/groups.py

@@ -16,7 +16,7 @@ class BaseImportGroup(object):
     def __init__(self, config=None, **kwargs):
         self.config = config or {}
 
-        self.statements = []
+        self.statements = kwargs.get('statements', [])
         self.file_artifacts = kwargs.get('file_artifacts', {})
 
     @property
@@ -133,16 +133,23 @@ def should_add_statement(self, statement):
 ))
 
 
+def sort_groups(groups):
+    return sorted(
+        groups,
+        key=lambda i: list(GROUP_MAPPING.values()).index(type(i))
+    )
+
+
 @six.python_2_unicode_compatible
-class ImportGroups(object):
-    def __init__(self, **kwargs):
-        self.groups = []
+class ImportGroups(list):
+    def __init__(self, *args, **kwargs):
+        super(ImportGroups, self).__init__(*args)
         self.file_artifacts = kwargs.get('file_artifacts', {})
 
     def all_line_numbers(self):
         return sorted(list(set(list(
             itertools.chain(*map(operator.methodcaller('all_line_numbers'),
-                                 self.groups))
+                                 self))
         ))))
 
     def add_group(self, config):
@@ -155,13 +162,10 @@ def add_group(self, config):
             msg = ('"{}" is not supported import group'.format(config['type']))
             raise ValueError(msg)
 
-        self.groups.append(GROUP_MAPPING[config['type']](config))
+        self.append(GROUP_MAPPING[config['type']](config))
 
     def add_statement_to_group(self, statement):
-        groups_by_priority = sorted(
-            self.groups,
-            key=lambda i: list(GROUP_MAPPING.values()).index(type(i))
-        )
+        groups_by_priority = sort_groups(self)
 
         added = False
 
@@ -182,7 +186,7 @@ def as_string(self):
         sep = self.file_artifacts.get('sep', '\n') * 2
         return sep.join(filter(
             None, map(operator.methodcaller('as_string'),
-                      self.groups)
+                      self)
         ))
 
     def formatted(self, formatter=DEFAULT_FORMATTER, length=DEFAULT_LENGTH):
@@ -191,7 +195,7 @@ def formatted(self, formatter=DEFAULT_FORMATTER, length=DEFAULT_LENGTH):
             None, map(operator.methodcaller('formatted',
                                             formatter=formatter,
                                             length=length),
-                      self.groups)
+                      self)
         ))
 
     def __str__(self):

tests/test_groups.py

@@ -238,7 +238,7 @@ def test_should_add_statement(self):
 class TestImportGroups(unittest.TestCase):
     def test_init(self):
         groups = ImportGroups()
-        self.assertListEqual(groups.groups, [])
+        self.assertListEqual(groups, [])
 
     def test_all_line_numbers(self):
         groups = ImportGroups()
@@ -248,12 +248,12 @@ def test_all_line_numbers(self):
         g = BaseImportGroup()
         g.statements = [mock.MagicMock(line_numbers=[2, 7],
                                        spec=ImportStatement)]
-        groups.groups.append(g)
+        groups.append(g)
 
         g = BaseImportGroup()
         g.statements = [mock.MagicMock(line_numbers=[1, 2],
                                        spec=ImportStatement)]
-        groups.groups.append(g)
+        groups.append(g)
 
         self.assertListEqual(groups.all_line_numbers(), [1, 2, 7])
 
@@ -268,14 +268,14 @@ def test_add_group(self):
 
         groups.add_group({'type': 'stdlib'})
 
-        self.assertEqual(len(groups.groups), 1)
-        self.assertEqual(groups.groups[0].__class__, StdLibGroup)
+        self.assertEqual(len(groups), 1)
+        self.assertEqual(groups[0].__class__, StdLibGroup)
 
     def test_add_statement_to_group_one(self):
         groups = ImportGroups()
-        groups.groups = [
+        groups.extend([
             LocalGroup()
-        ]
+        ])
 
         with self.assertRaises(ValueError):
             groups.add_statement_to_group(
@@ -287,27 +287,27 @@ def test_add_statement_to_group_one(self):
         )
 
         self.assertListEqual(
-            groups.groups[0].statements,
+            groups[0].statements,
             [ImportStatement([], '.a')]
         )
 
     def test_add_statement_to_group_priority(self):
         groups = ImportGroups()
-        groups.groups = [
+        groups.extend([
             RemainderGroup(),
             LocalGroup(),
-        ]
+        ])
 
         groups.add_statement_to_group(
             ImportStatement([], '.a')
         )
 
         self.assertListEqual(
-            groups.groups[0].statements,
+            groups[0].statements,
             []
         )
         self.assertListEqual(
-            groups.groups[1].statements,
+            groups[1].statements,
             [ImportStatement([], '.a')]
         )
 
@@ -321,10 +321,10 @@ def test_formatted_with_artifacts(self):
         artifacts = {'sep': '\r\n'}
 
         groups = ImportGroups(file_artifacts=artifacts)
-        groups.groups = [
+        groups.extend([
             RemainderGroup(file_artifacts=artifacts),
             LocalGroup(file_artifacts=artifacts),
-        ]
+        ])
 
         groups.add_statement_to_group(
             ImportStatement([], '.a', file_artifacts=artifacts)