airflow-cli 0.1.1__py3-none-any.whl → 0.1.11__py3-none-any.whl

airflow_cli/docker_utils.py

@@ -3,6 +3,7 @@ import logging
 from glob import glob
 import yaml
 import os
+import shutil
 
 log = logging.getLogger(__name__)
 
@@ -11,14 +12,32 @@ def docker_up():
     env = os.environ.copy()
     env["AIRFLOW_UID"] = "50000"
     
-    compose_file = os.path.join(os.path.dirname(__file__), "docker-compose.yml")
-    subprocess.run(["docker", "compose", "-f", compose_file, "up", "-d"], env=env, check=True)
-    log.info("✅ Docker environment is ready: http://localhost:8080")
+    # Verifica se já existe docker-compose.yml no diretório atual
+    local_compose_file = "docker-compose.yml"
+    if not os.path.exists(local_compose_file):
+        log.info("📋 Copiando docker-compose.yml para o projeto...")
+        # Copia o arquivo do pacote para o diretório atual
+        package_compose_file = os.path.join(os.path.dirname(__file__), "docker-compose.yml")
+        shutil.copy2(package_compose_file, local_compose_file)
+        log.info("✅ docker-compose.yml copiado com sucesso")
+    
+    # Verifica se a pasta dags existe
+    dags_path = "dags"
+    if not os.path.exists(dags_path):
+        log.info("📁 Criando pasta 'dags'...")
+        os.makedirs(dags_path, exist_ok=True)
+    
+    try:
+        subprocess.run(["docker", "compose", "up", "-d"], env=env, check=True)
+        log.info("✅ Docker environment is ready: http://localhost:8080")
+    except subprocess.CalledProcessError as e:
+        log.error(f"❌ Erro ao iniciar Docker: {e}")
+        log.error("Verifique se o Docker está rodando e funcionando corretamente")
+        raise
 
 def docker_down():
     log.info("🐳 Stopping Docker environment...")
-    compose_file = os.path.join(os.path.dirname(__file__), "docker-compose.yml")
-    subprocess.run(["docker", "compose", "-f", compose_file, "down"], check=False)
+    subprocess.run(["docker", "compose", "down"], check=False)
 
 def run_dag():
     log.info("🚀 Running DAG in Docker...")

{airflow_cli-0.1.1.dist-info → airflow_cli-0.1.11.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: airflow-cli
-Version: 0.1.1
+Version: 0.1.11
 Summary: CLI para facilitar o setup de Airflow com Docker.
 Author-email: LEMA-UFPB <ufpb.lema@gmail.com>
 License-Expression: MIT
@@ -135,7 +135,7 @@ This will run flake8 linter on the `dags/` folder to check for Python syntax iss
 python -m build
 
 # Install built package
-pip install dist/airflow_docker_helper-*.whl
+pip install dist/airflow_cli-*.whl
 ```
 
 ## Publishing to PyPI

airflow_cli-0.1.11.dist-info/RECORD

@@ -0,0 +1,12 @@
+airflow_cli/__init__.py,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
+airflow_cli/cli.py,sha256=CxmcgmV2txkravhZAlimA_r2swPnGHQqm7ppxkahEBM,1507
+airflow_cli/docker_utils.py,sha256=fhSadDaNyURW-9pu7ZtVt2muh2kJQJRpEOyyOpu5B74,2319
+airflow_cli/env_utils.py,sha256=f0L9RBwylypGx5U8BjZlLEtX3ldYNYma7xrLwF0aOU8,1110
+airflow_cli/menu.py,sha256=syKVdaSCxpkt_hPeces3F-wjZdSPZuqOsDyTbqV4gkg,1405
+airflow_cli/os_utils.py,sha256=2iLi9atS94Rp0YMduChH3poNrXWTSeGSTnSWO5v_9W8,1003
+airflow_cli-0.1.11.dist-info/licenses/LICENSE,sha256=UrTV1prTxEoB9MSB9MG4BkGhNDk9EQx5tDTmuQ2C4u4,1065
+airflow_cli-0.1.11.dist-info/METADATA,sha256=VaTXUcMPiY7M9YkCUzjn8ZriRZOq6msc0gGsEb5NDJA,2870
+airflow_cli-0.1.11.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+airflow_cli-0.1.11.dist-info/entry_points.txt,sha256=TyX3EFPfUxoxQ8aazxJREZl6VsOobW_JJlgA7jGpIBU,53
+airflow_cli-0.1.11.dist-info/top_level.txt,sha256=oD1Su7hc6_0veTZoMGIZph94T9S-dtsL-G_IOLuqqVM,12
+airflow_cli-0.1.11.dist-info/RECORD,,

airflow_cli-0.1.1.dist-info/RECORD

@@ -1,12 +0,0 @@
-airflow_cli/__init__.py,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
-airflow_cli/cli.py,sha256=CxmcgmV2txkravhZAlimA_r2swPnGHQqm7ppxkahEBM,1507
-airflow_cli/docker_utils.py,sha256=fxetHBJv5h8PeiVsUPEwkybxySOJBRXOwqFzSrGVBjA,1576
-airflow_cli/env_utils.py,sha256=f0L9RBwylypGx5U8BjZlLEtX3ldYNYma7xrLwF0aOU8,1110
-airflow_cli/menu.py,sha256=syKVdaSCxpkt_hPeces3F-wjZdSPZuqOsDyTbqV4gkg,1405
-airflow_cli/os_utils.py,sha256=2iLi9atS94Rp0YMduChH3poNrXWTSeGSTnSWO5v_9W8,1003
-airflow_cli-0.1.1.dist-info/licenses/LICENSE,sha256=UrTV1prTxEoB9MSB9MG4BkGhNDk9EQx5tDTmuQ2C4u4,1065
-airflow_cli-0.1.1.dist-info/METADATA,sha256=HJtmkDTGCGy9n1v0WAAes-TudDhsykkSUVkKqqQy_oc,2879
-airflow_cli-0.1.1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-airflow_cli-0.1.1.dist-info/entry_points.txt,sha256=TyX3EFPfUxoxQ8aazxJREZl6VsOobW_JJlgA7jGpIBU,53
-airflow_cli-0.1.1.dist-info/top_level.txt,sha256=oD1Su7hc6_0veTZoMGIZph94T9S-dtsL-G_IOLuqqVM,12
-airflow_cli-0.1.1.dist-info/RECORD,,