Merge pull request #70 from airflow-laminar/tkp/suppyd

Move to supervisor-pydantic
airflow-laminar · Feb 9, 2025 · 99e357d · 99e357d
2 parents 1b8685d + 99f94cc
commit 99e357d
Show file tree

Hide file tree

Showing 61 changed files with 76 additions and 2,279 deletions.
diff --git a/airflow_supervisor/__init__.py b/airflow_supervisor/__init__.py
@@ -1,5 +1,6 @@
+from supervisor_pydantic import *
+
 from .airflow import *
-from .client import *
 from .config import *
 
 __version__ = "1.1.1"
diff --git a/airflow_supervisor/airflow/__init__.py b/airflow_supervisor/airflow/__init__.py
@@ -1,5 +1,3 @@
-from .commands import *
-from .common import SupervisorTaskStep
 from .local import *
 
 try:

diff --git a/airflow_supervisor/airflow/__main__.py b/airflow_supervisor/airflow/__main__.py
diff --git a/airflow_supervisor/airflow/commands.py b/airflow_supervisor/airflow/commands.py
diff --git a/airflow_supervisor/airflow/common.py b/airflow_supervisor/airflow/common.py
@@ -1,25 +1,8 @@
-from typing import Literal
-
-SupervisorTaskStep = Literal[
-    "configure-supervisor",
-    "start-supervisor",
-    "start-programs",
-    "stop-programs",
-    "check-programs",
-    "restart-programs",
-    "stop-supervisor",
-    "unconfigure-supervisor",
-    "force-kill",
-]
+from airflow.exceptions import AirflowSkipException
 
 
 def skip_():
-    from airflow.exceptions import AirflowSkipException
-
     raise AirflowSkipException
 
 
-__all__ = (
-    "SupervisorTaskStep",
-    "skip_",
-)
+__all__ = ("skip_",)