Add OrbiterLivyOperator (#40)

* feat: LivyOperator * chore: version bump
astronomer · Jan 10, 2025 · 951e494 · 951e494
1 parent 7d4f7e6
commit 951e494
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 1 deletion.
diff --git a/orbiter/__init__.py b/orbiter/__init__.py
@@ -3,7 +3,7 @@
 import re
 from typing import Any, Tuple
 
-__version__ = "1.4.0"
+__version__ = "1.4.1"
 
 version = __version__
 

diff --git a/orbiter/objects/operators/livy.py b/orbiter/objects/operators/livy.py
@@ -0,0 +1,16 @@
+from orbiter.objects import ImportList, OrbiterRequirement
+from orbiter.objects.task import OrbiterOperator, RenderAttributes
+
+
+class OrbiterLivyOperator(OrbiterOperator):
+    imports: ImportList = [
+        OrbiterRequirement(
+            package="apache-airflow-providers-apache-livy",
+            module="airflow.providers.apache.livy.operators.livy",
+            names=["LivyOperator"],
+        )
+    ]
+    operator: str = "LivyOperator"
+
+    # noinspection Pydantic
+    render_attributes: RenderAttributes = OrbiterOperator.render_attributes