dbworkload 0.9.2.dev1__tar.gz → 0.10.1__tar.gz

{dbworkload-0.9.2.dev1 → dbworkload-0.10.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: dbworkload
-Version: 0.9.2.dev1
+Version: 0.10.1
 Summary: Workload framework
 License: GPLv3+
 Author: Fabio Ghirardello
@@ -14,6 +14,7 @@ Classifier: Programming Language :: Python :: 3.12
 Classifier: Programming Language :: Python :: 3.13
 Provides-Extra: all
 Provides-Extra: cassandra
+Provides-Extra: convert
 Provides-Extra: mariadb
 Provides-Extra: mongo
 Provides-Extra: mysql
@@ -22,13 +23,20 @@ Provides-Extra: oracle
 Provides-Extra: postgres
 Provides-Extra: spanner
 Requires-Dist: cassandra-driver ; extra == "all" or extra == "cassandra"
+Requires-Dist: fastembed (>=0.7.3,<0.8.0) ; extra == "convert"
 Requires-Dist: google-cloud-spanner ; extra == "all" or extra == "spanner"
 Requires-Dist: jinja2
+Requires-Dist: langchain-core (>=1.0.5,<2.0.0) ; extra == "convert"
+Requires-Dist: langchain-ollama (>=1.0.0,<2.0.0) ; extra == "convert"
+Requires-Dist: langchain-openai (>=1.0.2,<2.0.0) ; extra == "convert"
+Requires-Dist: langgraph (>=1.0.3,<2.0.0) ; extra == "convert"
 Requires-Dist: mariadb ; extra == "all" or extra == "mariadb"
 Requires-Dist: mysql-connector-python ; extra == "all" or extra == "mysql"
 Requires-Dist: numpy
+Requires-Dist: openai (>=2.8.0,<3.0.0) ; extra == "convert"
 Requires-Dist: oracledb ; extra == "all" or extra == "oracle"
 Requires-Dist: pandas
+Requires-Dist: pgvector (>=0.4.1,<0.5.0) ; extra == "convert"
 Requires-Dist: plotext
 Requires-Dist: plotly
 Requires-Dist: prometheus-client

{dbworkload-0.9.2.dev1 → dbworkload-0.10.1}/dbworkload/cli/main.py

@@ -42,7 +42,7 @@ app = typer.Typer(
 )
 
 
-app.add_typer(dbworkload.cli.util.app, name="util")
+app.add_typer(dbworkload.cli.util.util_app, name="util")
 
 version: bool = typer.Option(True)
 

{dbworkload-0.9.2.dev1 → dbworkload-0.10.1}/dbworkload/cli/util.py

@@ -1,16 +1,20 @@
 #!/usr/bin/python
 
+import sys
 from enum import Enum
 from pathlib import Path
 from typing import Optional
 
 import typer
 
-import dbworkload.models.run
 import dbworkload.models.util
-import dbworkload.utils.common
 from dbworkload.cli.dep import EPILOG, Param
 
+try:
+    from ..models.convert import ConvertTool
+except:
+    pass
+
 
 class Compression(str, Enum):
     bz2 = "bz2"
@@ -19,14 +23,14 @@ class Compression(str, Enum):
     zip = "zip"
 
 
-app = typer.Typer(
+util_app = typer.Typer(
     epilog=EPILOG,
     no_args_is_help=True,
     help="Various utils.",
 )
 
 
-@app.command(
+@util_app.command(
     "csv",
     epilog=EPILOG,
     no_args_is_help=True,
@@ -99,7 +103,7 @@ def util_csv(
     )
 
 
-@app.command(
+@util_app.command(
     "yaml",
     epilog=EPILOG,
     no_args_is_help=True,
@@ -135,7 +139,7 @@ def util_yaml(
     dbworkload.models.util.util_yaml(input=input, output=output)
 
 
-@app.command(
+@util_app.command(
     "merge_sort",
     epilog=EPILOG,
     no_args_is_help=True,
@@ -178,7 +182,7 @@ def util_sort_merge(
     dbworkload.models.util.util_merge_sort(input, output, csv_max_rows, compress)
 
 
-@app.command(
+@util_app.command(
     "plot",
     epilog=EPILOG,
     no_args_is_help=True,
@@ -201,7 +205,7 @@ def util_plot(
     dbworkload.models.util.util_plot(input)
 
 
-@app.command(
+@util_app.command(
     "html",
     epilog=EPILOG,
     no_args_is_help=True,
@@ -224,7 +228,7 @@ def util_html(
     dbworkload.models.util.util_html(input)
 
 
-@app.command(
+@util_app.command(
     "merge_csvs",
     epilog=EPILOG,
     no_args_is_help=True,
@@ -247,7 +251,7 @@ def util_merge_csvs(
     dbworkload.models.util.util_merge_csvs(input_dir)
 
 
-@app.command(
+@util_app.command(
     "gen_stub",
     epilog=EPILOG,
     no_args_is_help=True,
@@ -268,3 +272,61 @@ def util_gen_stub(
     ),
 ):
     dbworkload.models.util.util_gen_stub(input_file)
+
+
+@util_app.command(
+    name="convert",
+    help="Convert from PL to PL/pgSQL",
+    no_args_is_help=True,
+)
+def cli_convert(
+    base_dir: Optional[Path] = typer.Option(
+        ".",
+        "--dir",
+        "-d",
+        help="Directory path",
+        exists=True,
+        file_okay=False,
+        dir_okay=True,
+        writable=False,
+        readable=True,
+        resolve_path=True,
+    ),
+    uri: str = typer.Option(
+        None,
+        "--uri",
+        help="The connection URI to the database.",
+    ),
+    root_file: Optional[str] = typer.Option(
+        None,
+        "--root-file",
+        "-r",
+        help="The root_file. Leave empty for processing all *.ddl files.",
+    ),
+    generator_llm: Optional[str] = typer.Option(
+        "Ollama:llama3.2:3b",
+        "--generator-llm",
+        "-g",
+        help="The generator provider:model_name",
+    ),
+    refiner_llm: Optional[str] = typer.Option(
+        "OpenAI:gpt-5",
+        "--refiner-llm",
+        "-n",
+        help="The refiner provider:model_name.",
+    ),
+):
+
+    try:
+        ConvertTool(
+            base_dir,
+            uri,
+            root_file,
+            generator_llm,
+            refiner_llm,
+            # seed,
+            # seed_each_time,
+        ).run()
+    except Exception as e:
+        print(e, file=sys.stderr)
+        typer.Exit(1)