Make mkosi a python module and generate script via entrypoint

.github/workflows/ci.yml

@@ -241,15 +241,15 @@ jobs:
       run: sudo apt-get --assume-yes --no-install-recommends install zypper
 
     - name: Build ${{ matrix.distro }}/${{ matrix.format }}
-      run: sudo ./mkosi
+      run: sudo python3 -m mkosi
             --debug run
             --distribution ${{ matrix.distro }}
             --format ${{ matrix.format }}
 
     - name: Build ${{ matrix.distro }}/${{ matrix.format }} UEFI
       if: matrix.format != 'directory' && matrix.format != 'tar' && matrix.format != 'plain_squashfs' &&
           (matrix.distro != 'clear' || matrix.format != 'gpt_squashfs')
-      run: sudo ./mkosi
+      run: sudo python3 -m mkosi
             --debug run
             --distribution ${{ matrix.distro }}
             --format ${{ matrix.format }}
@@ -259,7 +259,7 @@ jobs:
 
     - name: Build ${{ matrix.distro }}/${{ matrix.format }} BIOS
       if: matrix.format != 'directory' && matrix.format != 'tar' && !contains(matrix.format, 'squashfs')
-      run: sudo ./mkosi
+      run: sudo python3 -m mkosi
             --debug run
             --distribution ${{ matrix.distro }}
             --format ${{ matrix.format }}

mkosi → mkosi/__init__.py

@@ -1,5 +1,3 @@
-#!/usr/bin/env python3
-# PYTHON_ARGCOMPLETE_OK
 # SPDX-License-Identifier: LGPL-2.1+
 
 import argparse
@@ -58,10 +56,6 @@
 __version__ = '5'
 
 
-if sys.version_info < (3, 6):
-    sys.exit("Sorry, we need at least Python 3.6.")
-
-
 # These types are only generic during type checking and not at runtime, leading
 # to a TypeError during compilation.
 # Let's be as strict as we can with the description for the usage we have.
@@ -5278,25 +5272,3 @@ def run_verb(args: CommandLineArguments) -> None:
 
     if args.verb == "qemu":
         run_qemu(args)
-
-
-def main() -> None:
-    try:
-        args = parse_args()
-
-        for job_name, a in args.items():
-            # Change working directory if --directory is passed
-            if a.directory:
-                work_dir = a.directory
-                if os.path.isdir(work_dir):
-                    os.chdir(work_dir)
-                else:
-                    die("Error: %s is not a directory!" % work_dir)
-            with complete_step('Processing ' + job_name):
-                run_verb(a)
-    except MkosiException:
-        sys.exit(1)
-
-
-if __name__ == "__main__":
-    main()

mkosi/__main__.py

@@ -0,0 +1,23 @@
+# SPDX-License-Identifier: LGPL-2.1+
+# PYTHON_ARGCOMPLETE_OK
+import os
+import sys
+
+from . import parse_args, complete_step, run_verb, die, MkosiException
+
+
+try:
+    args = parse_args()
+
+    for job_name, a in args.items():
+        # Change working directory if --directory is passed
+        if a.directory:
+            work_dir = a.directory
+            if os.path.isdir(work_dir):
+                os.chdir(work_dir)
+            else:
+                die(f"Error: {work_dir} is not a directory!")
+        with complete_step(f"Processing {job_name}"):
+            run_verb(a)
+except MkosiException:
+    sys.exit(1)

setup.py

@@ -1,8 +1,6 @@
 #!/usr/bin/python3
 # SPDX-License-Identifier: LGPL-2.1+
 
-import sys
-
 from setuptools import setup, Command
 
 class BuildManpage(Command):
@@ -17,9 +15,6 @@ def finalize_options(self):
     def run(self):
         self.spawn(['pandoc', '-t', 'man', '-o', 'mkosi.1', 'mkosi.md'])
 
-if sys.version_info < (3, 6):
-    sys.exit("Sorry, we need at least Python 3.6.")
-
 
 setup(
     name="mkosi",
@@ -29,6 +24,8 @@ def run(self):
     maintainer="mkosi contributors",
     maintainer_email="systemd-devel@lists.freedesktop.org",
     license="LGPLv2+",
-    scripts=["mkosi"],
-    cmdclass = { "man": BuildManpage }
+    python_requires=">=3.6",
+    packages = ["mkosi"],
+    cmdclass = { "man": BuildManpage },
+    entry_points = {"console_scripts": ["mkosi=mkosi.__main__"]},
 )

tests/conftest.py

@@ -2,13 +2,8 @@
 
 import sys
 import os 
-dir_path = os.path.dirname(os.path.realpath(__file__))
 
-import importlib.util
-spec = importlib.util.spec_from_file_location('mkosi', os.path.join(dir_path, '../mkosi.py'))
-mkosi_module = importlib.util.module_from_spec(spec)
-spec.loader.exec_module(mkosi_module)
-sys.modules['mkosi'] = mkosi_module
+import mkosi
 
 from tests.test_config_parser import MkosiConfig
 

tests/test_config_parser.py

@@ -1,16 +1,12 @@
 # SPDX-License-Identifier: LGPL-2.1+
 
+import configparser
+import copy
 import os
-dir_path = os.path.dirname(os.path.realpath(__file__))
-
-import importlib.util
-spec = importlib.util.spec_from_file_location('mkosi', os.path.join(dir_path, '../mkosi.py'))
-mkosi = importlib.util.module_from_spec(spec)
-spec.loader.exec_module(mkosi)
 
 import pytest
-import configparser
-import copy
+
+import mkosi
 
 
 class ChangeCwd(object):