add egg updater script which fixes pytest initialization issue

cve_bin_tool/cli.py

@@ -12,34 +12,41 @@
 
 # Python 2 compatibility stuff
 from __future__ import print_function
-import sys
+
+import argparse
+import logging
+import multiprocessing
 import os
-import csv
 import platform
 import subprocess
-import logging
-import argparse
+import sys
 import textwrap
 import threading
-import pkg_resources
-import multiprocessing
 from collections import defaultdict
 
-from .version import VERSION
-from .util import DirWalk, inpath
-from .extractor import Extractor
-from .strings import Strings
-from .file import is_binary
-from .OutputEngine import OutputEngine
+import pkg_resources
 
+from .OutputEngine import OutputEngine
 from .cvedb import CVEDB, OLD_CACHE_DIR
+from .extractor import Extractor
+from .file import is_binary
 from .log import LOGGER
+from .strings import Strings
+from .util import DirWalk, inpath
+from .version import VERSION
 
 try:
     import queue
 except ImportError:
     import Queue as queue
 
+try:
+    from egg_updater import update_egg
+
+    DEVELOP = True
+except ModuleNotFoundError:
+    DEVELOP = False
+
 
 class InvalidFileError(Exception):
     """ Filepath is invalid for scanning."""
@@ -55,6 +62,11 @@ class Scanner(object):
     def __init__(self, cvedb, checkers=None, logger=None):
         if logger is None:
             logger = LOGGER.getChild(self.__class__.__name__)
+        # Update egg if installed in development mode
+        if DEVELOP:
+            logger.info("Updating egg_info")
+            update_egg()
+
         # Load checkers if not given
         if checkers is None:
             checkers = self.load_checkers()

egg_updater.py

@@ -0,0 +1,46 @@
+import os
+import sys
+from io import StringIO
+
+from setuptools.dist import Distribution
+
+
+def update_egg():
+    with StringIO() as f:
+        sys.stdout = f
+        dist = Distribution(
+            dict(
+                script_name="setup.py",
+                script_args=["egg_info"],
+                name="cve-bin-tool",
+                entry_points={
+                    "cve_bin_tool.checker": [
+                        "{} = cve_bin_tool.checkers.{}:{}".format(
+                            filename.replace(".py", ""),
+                            filename.replace(".py", ""),
+                            "".join(
+                                (filename.replace(".py", "") + " checker")
+                                .replace("_", " ")
+                                .title()
+                                .split()
+                            ),
+                        )
+                        for filename in os.listdir(
+                            os.path.join(
+                                os.path.abspath(os.path.dirname(__file__)),
+                                "cve_bin_tool",
+                                "checkers",
+                            )
+                        )
+                        if filename.endswith(".py") and "__init__" not in filename
+                    ],
+                },
+            )
+        )
+        dist.parse_command_line()
+        dist.run_commands()
+        sys.stdout = sys.__stdout__
+
+
+if __name__ == "__main__":
+    update_egg()

setup.py

@@ -39,7 +39,7 @@
         "Programming Language :: Python :: Implementation :: PyPy",
     ],
     install_requires=["jsonschema>=3.0.2", "pytest"],
-    packages=find_packages(),
+    packages=find_packages(exclude=["egg_updater.py"]),
     entry_points={
         "console_scripts": [
             "cve-bin-tool = cve_bin_tool.cli:main",