Refactor main function

Signed-off-by: Paul Barker <paul@pbarker.dev>
Ref: #26

Author: pbrkr

mirrorshades

@@ -8,6 +8,6 @@ from os.path import dirname, join
 src_path = join(dirname(__file__), "src")
 sys.path.insert(0, src_path)
 
-import mirrorshades  # noqa: E402
+from mirrorshades.__main__ import main  # noqa: E402
 
-mirrorshades.main()
+main()

setup.cfg

@@ -33,7 +33,7 @@ install_requires =
 
 [options.entry_points]
 console_scripts =
-    mirrorshades = mirrorshades:main
+    mirrorshades = mirrorshades.__main__:main
 
 [options.packages.find]
 where=src

src/mirrorshades/__init__.py

@@ -1,44 +1,4 @@
 # Copyright (c) 2020-2021 Paul Barker <paul@pbarker.dev>
 # SPDX-License-Identifier: Apache-2.0
 
-import argparse
-import logging
-import sys
-
 __version__ = "0.2.1-dev"
-
-
-def parse_args():
-    parser = argparse.ArgumentParser(
-        prog="mirrorshades", description="Data mirroring tool"
-    )
-    parser.add_argument(
-        "config_path",
-        default="mirrorshades.yml",
-        nargs="?",
-        help="path to the configuration file (defaults to 'mirrorshades.yml' "
-        "in the current directory)",
-    )
-    parser.add_argument("--source", "-s", help="select a single source to synchronize")
-    parser.add_argument(
-        "--version", action="version", version=f"mirrorshades {__version__}"
-    )
-    return parser.parse_args()
-
-
-def main():
-    from . import config
-
-    logging.basicConfig(format="%(levelname)s: %(message)s", level=logging.INFO)
-    args = parse_args()
-    config.load(args.config_path)
-
-    if args.source:
-        try:
-            config.sources[args.source].mirror()
-        except KeyError:
-            logging.error(f"Source '{args.source}' does not exist in config file")
-            sys.exit(1)
-    else:
-        for source in config.sources.values():
-            source.mirror()

src/mirrorshades/__main__.py

@@ -1,6 +1,43 @@
 # Copyright (c) 2020-2021 Paul Barker <paul@pbarker.dev>
 # SPDX-License-Identifier: Apache-2.0
 
-from . import main
+import argparse
+import logging
+import sys
+from . import __version__, config
 
-main()
+
+def parse_args():
+    parser = argparse.ArgumentParser(
+        prog="mirrorshades",
+        description="An easily extensible tool for mirroring data from "
+        "git repositories, cloud storage, mail servers and other remote sources.",
+    )
+    parser.add_argument(
+        "config_path",
+        default="mirrorshades.yml",
+        nargs="?",
+        help="path to the configuration file "
+        "(defaults to 'mirrorshades.yml' in the current directory)",
+    )
+    parser.add_argument("--source", "-s", help="select a single source to synchronize")
+    parser.add_argument(
+        "--version", action="version", version=f"mirrorshades {__version__}"
+    )
+    return parser.parse_args()
+
+
+def main():
+    logging.basicConfig(format="%(levelname)s: %(message)s", level=logging.INFO)
+    args = parse_args()
+    config.load(args.config_path)
+
+    if args.source:
+        try:
+            config.sources[args.source].mirror()
+        except KeyError:
+            logging.error(f"Source '{args.source}' does not exist in config file")
+            sys.exit(1)
+    else:
+        for source in config.sources.values():
+            source.mirror()