konecty-sdk-python 0.3.0__tar.gz → 0.5.0__tar.gz

{konecty_sdk_python-0.3.0 → konecty_sdk_python-0.5.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: konecty-sdk-python
-Version: 0.3.0
+Version: 0.5.0
 Summary: Konecty SDK Python
 Author-email: Leonardo Leal <leonardo.leal@konecty.com>, Derotino Silveira <derotino.silveira@konecty.com>
 License: MIT

{konecty_sdk_python-0.3.0 → konecty_sdk_python-0.5.0}/cli/apply.py

@@ -9,6 +9,7 @@
 # ///
 """Script para aplicar alterações locais ao MongoDB."""
 
+import asyncio
 import json
 from pathlib import Path
 from typing import Any, Dict, List, Literal, Optional, Set, Tuple, TypedDict, cast
@@ -329,7 +330,8 @@ async def apply_namespace(
     help="Tenta reescrever em caso de falha",
 )
 @click.option("--w", default="majority", help="Nível de escrita do MongoDB")
-async def apply_command(
+@click.option("--mongo-url", help="URL do MongoDB")
+def apply_command(
     metadata_dir: str,
     host: str,
     port: int,
@@ -343,30 +345,71 @@ async def apply_command(
     direct_connection: bool,
     retry_writes: bool,
     w: str,
+    mongo_url: Optional[str],
 ) -> None:
     """Aplica alterações locais ao MongoDB."""
+    asyncio.run(
+        _apply_command(
+            metadata_dir=metadata_dir,
+            host=host,
+            port=port,
+            database=database,
+            username=username,
+            password=password,
+            replicaset=replicaset,
+            document=document,
+            prune=prune,
+            dry_run=dry_run,
+            direct_connection=direct_connection,
+            retry_writes=retry_writes,
+            w=w,
+            mongo_url=mongo_url,
+        )
+    )
+
+
+async def _apply_command(
+    metadata_dir: str,
+    host: str,
+    port: int,
+    database: str,
+    username: Optional[str],
+    password: Optional[str],
+    replicaset: Optional[str],
+    document: Optional[str],
+    prune: bool,
+    dry_run: bool,
+    direct_connection: bool,
+    retry_writes: bool,
+    w: str,
+    mongo_url: Optional[str],
+) -> None:
+    """Internal async implementation of apply command."""
     metadata_path = Path(metadata_dir).resolve()
     if not metadata_path.exists():
         console.print(f"[red]Diretório {metadata_dir} não encontrado[/red]")
         return
 
-    uri_params = []
-    if replicaset:
-        uri_params.extend(
-            [
-                f"replicaSet={replicaset}",
-                f"directConnection={'true' if direct_connection else 'false'}",
-                f"retryWrites={'true' if retry_writes else 'false'}",
-                f"w={w}",
-            ]
-        )
+    uri = mongo_url
+
+    if not uri:
+        uri_params = []
+        if replicaset:
+            uri_params.extend(
+                [
+                    f"replicaSet={replicaset}",
+                    f"directConnection={'true' if direct_connection else 'false'}",
+                    f"retryWrites={'true' if retry_writes else 'false'}",
+                    f"w={w}",
+                ]
+            )
 
-    uri_suffix = f"?{'&'.join(uri_params)}" if uri_params else ""
+        uri_suffix = f"?{'&'.join(uri_params)}" if uri_params else ""
 
-    if username and password:
-        uri = f"mongodb://{username}:{password}@{host}:{port}/admin{uri_suffix}"
-    else:
-        uri = f"mongodb://{host}:{port}{uri_suffix}"
+        if username and password:
+            uri = f"mongodb://{username}:{password}@{host}:{port}/admin{uri_suffix}"
+        else:
+            uri = f"mongodb://{host}:{port}{uri_suffix}"
 
     client = MongoClient(
         uri,
@@ -497,9 +540,13 @@ def main():
     if sys.platform == "win32":
         asyncio.set_event_loop_policy(asyncio.WindowsSelectorEventLoopPolicy())
 
-    asyncio.run(apply_command())
+    apply_command()
 
 
 if __name__ == "__main__":
     main()
     main()
+    main()
+    main()
+    main()
+    main()

{konecty_sdk_python-0.3.0 → konecty_sdk_python-0.5.0}/cli/backup.py

@@ -1,5 +1,6 @@
 """Script para fazer backup dos documentos do MongoDB."""
 
+import asyncio
 import json
 import tarfile
 import tempfile
@@ -24,7 +25,8 @@ console = Console()
 @click.option("--password", help="Senha do MongoDB")
 @click.option("--replicaset", help="Nome do ReplicaSet do MongoDB (ex: rs0)")
 @click.option("--version", help="Rótulo de versão para o arquivo de backup")
-async def backup_command(
+@click.option("--mongo-url", help="URL do MongoDB")
+def backup_command(
     host: str,
     port: int,
     database: str,
@@ -33,8 +35,36 @@ async def backup_command(
     password: Optional[str],
     replicaset: Optional[str],
     version: Optional[str],
+    mongo_url: Optional[str],
 ) -> None:
     """Gera backup dos documentos do MongoDB."""
+    asyncio.run(
+        _backup_command(
+            host=host,
+            port=port,
+            database=database,
+            output=output,
+            username=username,
+            password=password,
+            replicaset=replicaset,
+            version=version,
+            mongo_url=mongo_url,
+        )
+    )
+
+
+async def _backup_command(
+    host: str,
+    port: int,
+    database: str,
+    output: str,
+    username: Optional[str],
+    password: Optional[str],
+    replicaset: Optional[str],
+    version: Optional[str],
+    mongo_url: Optional[str],
+) -> None:
+    """Internal async implementation of backup command."""
     output_dir = Path(output).resolve()
     output_dir.mkdir(parents=True, exist_ok=True)
 
@@ -43,23 +73,26 @@ async def backup_command(
     version_label = f"_{version}" if version else ""
     backup_file = output_dir / f"backup_{timestamp}{version_label}.tar.gz"
 
-    uri_params = []
-    if replicaset:
-        uri_params.extend(
-            [
-                f"replicaSet={replicaset}",
-                "directConnection=false",
-                "retryWrites=true",
-                "w=majority",
-            ]
-        )
+    uri = mongo_url
+
+    if not uri:
+        uri_params = []
+        if replicaset:
+            uri_params.extend(
+                [
+                    f"replicaSet={replicaset}",
+                    "directConnection=false",
+                    "retryWrites=true",
+                    "w=majority",
+                ]
+            )
 
-    uri_suffix = f"?{'&'.join(uri_params)}" if uri_params else ""
+        uri_suffix = f"?{'&'.join(uri_params)}" if uri_params else ""
 
-    if username and password:
-        uri = f"mongodb://{username}:{password}@{host}:{port}/admin{uri_suffix}"
-    else:
-        uri = f"mongodb://{host}:{port}{uri_suffix}"
+        if username and password:
+            uri = f"mongodb://{username}:{password}@{host}:{port}/admin{uri_suffix}"
+        else:
+            uri = f"mongodb://{host}:{port}{uri_suffix}"
 
     client = MongoClient(
         uri,
@@ -121,7 +154,7 @@ def main():
     if sys.platform == "win32":
         asyncio.set_event_loop_policy(asyncio.WindowsSelectorEventLoopPolicy())
 
-    asyncio.run(backup_command())
+    backup_command()
 
 
 if __name__ == "__main__":

{konecty_sdk_python-0.3.0 → konecty_sdk_python-0.5.0}/cli/pull.py

@@ -1,5 +1,6 @@
 """Script para extrair dados do MongoDB e gerar arquivos JSON."""
 
+import asyncio
 import json
 from pathlib import Path
 from typing import Any, Dict, List, Literal, Optional, TypedDict, Union, cast
@@ -84,13 +85,14 @@ async def write_file(file_path: str, content: str) -> None:
 @click.option("--pivot", help="Nome do pivot específico para extrair")
 @click.option("--access", help="Nome do access específica para extrair")
 @click.option("--hook", help="Nome do hook específico para extrair")
+@click.option("--mongo-url", help="URL do MongoDB")
 @click.option(
     "--all",
     "extract_all",
     is_flag=True,
     help="Extrair todas as collections sem perguntar",
 )
-async def pull_command(
+def pull_command(
     host: str,
     port: int,
     database: str,
@@ -105,28 +107,71 @@ async def pull_command(
     access: Optional[str],
     hook: Optional[str],
     extract_all: bool,
+    mongo_url: Optional[str],
 ) -> None:
     """Baixa metadados do MongoDB."""
+    asyncio.run(
+        _pull_command(
+            host=host,
+            port=port,
+            database=database,
+            output=output,
+            username=username,
+            password=password,
+            replicaset=replicaset,
+            document=document,
+            view=view,
+            list_param=list_param,
+            pivot=pivot,
+            access=access,
+            hook=hook,
+            extract_all=extract_all,
+            mongo_url=mongo_url,
+        )
+    )
+
+
+async def _pull_command(
+    host: str,
+    port: int,
+    database: str,
+    output: str,
+    username: Optional[str],
+    password: Optional[str],
+    replicaset: Optional[str],
+    document: Optional[str],
+    view: Optional[str],
+    list_param: Optional[str],
+    pivot: Optional[str],
+    access: Optional[str],
+    hook: Optional[str],
+    mongo_url: Optional[str],
+    extract_all: bool,
+) -> None:
+    """Internal async implementation of pull command."""
     output_dir = Path(output).resolve()
     output_dir.mkdir(parents=True, exist_ok=True)
 
-    uri_params = []
-    if replicaset:
-        uri_params.extend(
-            [
-                f"replicaSet={replicaset}",
-                "directConnection=false",
-                "retryWrites=true",
-                "w=majority",
-            ]
-        )
+    uri = mongo_url
+
+    if not uri:
+        uri_params = []
+        if replicaset:
+            uri_params.extend(
+                [
+                    f"replicaSet={replicaset}",
+                    "directConnection=false",
+                    "retryWrites=true",
+                    "w=majority",
+                ]
+            )
 
-    uri_suffix = f"?{'&'.join(uri_params)}" if uri_params else ""
+        uri_suffix = f"?{'&'.join(uri_params)}" if uri_params else ""
 
-    if username and password:
-        uri = f"mongodb://{username}:{password}@{host}:{port}/admin{uri_suffix}"
-    else:
-        uri = f"mongodb://{host}:{port}{uri_suffix}"
+        if username and password:
+            uri = f"mongodb://{username}:{password}@{host}:{port}/admin{uri_suffix}"
+        else:
+            uri = f"mongodb://{host}:{port}{uri_suffix}"
 
     client = MongoClient(
         uri,
@@ -258,7 +303,7 @@ def main():
     if sys.platform == "win32":
         asyncio.set_event_loop_policy(asyncio.WindowsSelectorEventLoopPolicy())
 
-    asyncio.run(pull_command())
+    pull_command()
 
 
 if __name__ == "__main__":

{konecty_sdk_python-0.3.0 → konecty_sdk_python-0.5.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "konecty-sdk-python"
-version = "0.3.0"
+version = "0.5.0"
 description = "Konecty SDK Python"
 readme = "README.md"
 requires-python = ">=3.11"