Convert the trustymail script into a sub-module

Bring the command line functionality directly into the package and use
an entry point instead of a script.

Author: mcdonnnj

setup.py

@@ -113,7 +113,6 @@ def get_version(version_file):
             "pytest",
         ]
     },
-    scripts=["scripts/trustymail"],
-    # Conveniently allows one to run the CLI tool as `example`
-    # entry_points={"console_scripts": ["example = example.example:main"]},
+    # Conveniently allows one to run the CLI tool as `trustymail`
+    entry_points={"console_scripts": ["trustymail = trustymail.cli:main"]},
 )

src/trustymail/__main__.py

@@ -0,0 +1,5 @@
+"""Code to run if this package is used as a Python module."""
+
+from .cli import main
+
+main()

scripts/trustymail src/trustymail/cli.pyscripts/trustymail renamed to src/trustymail/cli.py

@@ -1,6 +1,3 @@
-#!/usr/bin/env python
-# -*- coding: utf-8 -*-
-
 """trustymail: A tool for scanning DNS mail records for evaluating security.
 Usage:
   trustymail (INPUT ...) [options]
@@ -58,14 +55,15 @@
 import docopt
 
 # Local imports
-import trustymail
+from . import trustymail
+from ._version import __version__
 
 # The default ports to be checked to see if an SMTP server is listening.
 _DEFAULT_SMTP_PORTS = {25, 465, 587}
 
 
 def main():
-    args = docopt.docopt(__doc__, version=trustymail.__version__)
+    args = docopt.docopt(__doc__, version=__version__)
 
     # Monkey patching trustymail to make it cache the PSL where we want
     if args["--psl-filename"] is not None:
@@ -181,7 +179,3 @@ def mkdir_p(path):
             pass
         else:
             raise
-
-
-if __name__ == "__main__":
-    main()

src/trustymail/domain.py

@@ -4,9 +4,8 @@
 
 import publicsuffix
 
-from trustymail import PublicSuffixListReadOnly
-from trustymail import PublicSuffixListFilename
-from trustymail import trustymail
+from . import PublicSuffixListReadOnly, PublicSuffixListFilename
+from . import trustymail
 
 
 def get_psl():

src/trustymail/trustymail.py

@@ -15,7 +15,7 @@
 import dns.reversename
 
 
-from trustymail.domain import get_public_suffix, Domain
+from .domain import get_public_suffix, Domain
 
 # A cache for SMTP scanning results
 _SMTP_CACHE = {}