Restructure project as an installable package

larsks · larsks · commit 67a2d6713d8b · 2025-10-31T14:20:16.000-04:00
diff --git a/batchtools/__init__.py b/batchtools/__init__.py
diff --git a/batchtools/basecommand.py b/batchtools/basecommand.py
diff --git a/batchtools/batchtools.py b/batchtools/batchtools.py
@@ -1,16 +1,16 @@
 import argparse
 import sys
 
-from basecommand import Command
-from basecommand import SubParserFactory
-from bj import ListJobsCommand
-from bd import DeleteJobsCommand
-from bl import LogsCommand
-from bp import PrintJobsCommand
-from bq import GpuQueuesCommand
-from br import CreateJobCommand
-from bps import ListPodsCommand
-from helpers import is_logged_in
+from .basecommand import Command
+from .basecommand import SubParserFactory
+from .bj import ListJobsCommand
+from .bd import DeleteJobsCommand
+from .bl import LogsCommand
+from .bp import PrintJobsCommand
+from .bq import GpuQueuesCommand
+from .br import CreateJobCommand
+from .bps import ListPodsCommand
+from .helpers import is_logged_in
 
 
 class BatchTools:
diff --git a/batchtools/bd.py b/batchtools/bd.py
@@ -6,9 +6,9 @@
 
 import openshift_client as oc
 
-from basecommand import Command
-from basecommand import SubParserFactory
-from helpers import oc_delete
+from .basecommand import Command
+from .basecommand import SubParserFactory
+from .helpers import oc_delete
 
 
 class DeleteJobsCommandArgs(argparse.Namespace):
diff --git a/batchtools/bj.py b/batchtools/bj.py
@@ -4,7 +4,7 @@
 import sys
 import openshift_client as oc
 
-from basecommand import Command
+from .basecommand import Command
 
 
 class ListJobsCommand(Command):
diff --git a/batchtools/bl.py b/batchtools/bl.py
@@ -5,9 +5,9 @@
 import argparse
 import openshift_client as oc
 
-from basecommand import Command
-from basecommand import SubParserFactory
-from helpers import pretty_print
+from .basecommand import Command
+from .basecommand import SubParserFactory
+from .helpers import pretty_print
 
 
 class LogsCommandArgs(argparse.Namespace):
diff --git a/batchtools/bp.py b/batchtools/bp.py
@@ -6,8 +6,8 @@
 
 import openshift_client as oc
 
-from basecommand import Command
-from basecommand import SubParserFactory
+from .basecommand import Command
+from .basecommand import SubParserFactory
 
 
 class PrintJobsCommandArgs(argparse.Namespace):
diff --git a/batchtools/bps.py b/batchtools/bps.py
@@ -6,8 +6,8 @@
 import argparse
 import openshift_client as oc
 
-from basecommand import Command
-from basecommand import SubParserFactory
+from .basecommand import Command
+from .basecommand import SubParserFactory
 
 
 class ListPodsCommandArgs(argparse.Namespace):
diff --git a/batchtools/bq.py b/batchtools/bq.py
@@ -5,7 +5,7 @@
 
 import openshift_client as oc
 
-from basecommand import Command
+from .basecommand import Command
 
 
 class GpuQueuesCommand(Command):
diff --git a/batchtools/br.py b/batchtools/br.py
@@ -11,12 +11,12 @@
 
 import openshift_client as oc
 
-from basecommand import Command
-from basecommand import SubParserFactory
-from build_yaml import build_job_body
-from helpers import pretty_print
-from helpers import oc_delete
-from file_setup import prepare_context
+from .basecommand import Command
+from .basecommand import SubParserFactory
+from .build_yaml import build_job_body
+from .helpers import pretty_print
+from .helpers import oc_delete
+from .file_setup import prepare_context
 
 
 class CreateJobCommandArgs(argparse.Namespace):
diff --git a/batchtools/build_yaml.py b/batchtools/build_yaml.py
diff --git a/batchtools/file_setup.py b/batchtools/file_setup.py
diff --git a/batchtools/helpers.py b/batchtools/helpers.py
diff --git a/pyproject.toml b/pyproject.toml
@@ -1,15 +1,22 @@
+[build-system]
+requires = ["setuptools>=42"]
+build-backend = "setuptools.build_meta"
+
 [project]
 name = "python-batchtools"
 version = "0.1.0"
 description = "Add your description here"
 readme = "README.md"
 requires-python = ">=3.10"
 dependencies = [
-    "kubernetes-typed>=18.20.2",
-    "openshift-client>=2.0.5",
-    "typing-extensions>=4.15.0",
+  "kubernetes-typed>=18.20.2",
+  "openshift-client>=2.0.5",
+  "typing-extensions>=4.15.0",
 ]
 
+[project.scripts]
+batchtools = "batchtools.batchtools:main"
+
 [dependency-groups]
 dev = ["pytest>=8.4.2", "pytest-cov>=7.0.0", "ty>=0.0.1a24"]
 
@@ -32,3 +39,6 @@ addopts = "--cov=. --cov-report=html"
 
 [tool.coverage.report]
 omit = ["tests/*"]
+
+[tool.setuptools.packages.find]
+where = ["batchtools"]
diff --git a/tests/test_batchtools.py b/tests/test_batchtools.py
@@ -1,5 +1,5 @@
 import pytest
-from batchtools import BatchTools
+from batchtools.batchtools import BatchTools
 
 
 def test_no_command():
diff --git a/tests/test_bd.py b/tests/test_bd.py
@@ -5,7 +5,7 @@
 
 import argparse
 
-from bd import DeleteJobsCommand
+from batchtools.bd import DeleteJobsCommand
 from tests.helpers import DictToObject
 
 
diff --git a/tests/test_bj.py b/tests/test_bj.py
@@ -3,7 +3,7 @@
 
 import argparse
 
-from bj import ListJobsCommand
+from batchtools.bj import ListJobsCommand
 from tests.helpers import DictToObject
 
 
diff --git a/tests/test_bl.py b/tests/test_bl.py
@@ -6,7 +6,7 @@
 from typing import Any
 from typing import Callable
 
-from bl import LogsCommand
+from batchtools.bl import LogsCommand
 from tests.helpers import DictToObject
 
 
diff --git a/tests/test_bp.py b/tests/test_bp.py
@@ -5,7 +5,7 @@
 from typing import Any
 import argparse
 
-from bp import PrintJobsCommand
+from batchtools.bp import PrintJobsCommand
 from tests.helpers import DictToObject
 
 
diff --git a/tests/test_br.py b/tests/test_br.py
@@ -4,8 +4,8 @@
 import tempfile
 import argparse
 
-import build_yaml
-from br import CreateJobCommand
+import batchtools.build_yaml
+from batchtools.br import CreateJobCommand
 from tests.helpers import DictToObject
 
 
@@ -120,7 +120,7 @@ def test_create_job_nowait(
         },
     }
 
-    build_yaml.rsync_script = "testcommand {cmdline}"
+    batchtools.build_yaml.rsync_script = "testcommand {cmdline}"
     CreateJobCommand.run(args)
 
     assert mock_create.call_args.args[0] == expected
diff --git a/uv.lock b/uv.lock
