dmddl 0.1.0__py3-none-any.whl

dmddl/.ruff_cache/.gitignore

@@ -0,0 +1,2 @@
+# Automatically created by ruff.
+*

dmddl/.ruff_cache/CACHEDIR.TAG

@@ -0,0 +1 @@
+Signature: 8a477f597d28d172789f06886806bc55

dmddl/cli.py

@@ -0,0 +1,100 @@
+import questionary
+from config.settings import LLMSettings
+from rich import print
+from rich.syntax import Syntax
+from rich.console import Console
+from models.llm import openai_request
+from models.prompt import prompt as base_prompt
+import argparse
+import sys
+
+
+AVAILABLE_PROVIDERS = ["OpenAI"]
+sys.tracebacklimit = 10 # it makes errors more user-friendly (turn off for dev)
+
+
+def choose_provider(providers):
+    selected = questionary.select("Choose your LLM provider:",
+                                   choices=providers).ask()
+    return selected
+
+
+def ask_api_key():
+    api_key = questionary.password("Enter your api key:").ask()
+    return api_key
+
+
+def make_test_query(provider, api_key):
+    console = Console()
+    with console.status("[bold blue]Making test query"):
+        if provider == "OpenAI":
+            try:
+                response = openai_request("Hello! Its a test query :)", api_key)
+                print(f"\n[green bold]{response} \nAll done! Your api key is correct!")
+            except KeyError:
+                print("Your api key is incorrect! Use -c (--config) to set another api key")
+
+
+def make_query(provider, api_key, prompt):
+    console = Console()
+    with console.status("[bold blue]Making query. Wait for result..."):
+        if provider == "OpenAI":
+            response = openai_request(base_prompt+prompt, api_key)
+            return response
+        raise ValueError("LLM Provider not found")
+
+
+def write_output_file(data):
+    with open("output.txt", 'w') as file:
+        file.write(data)
+
+
+def set_parameters():
+    settings = LLMSettings()
+
+    llm_provider = choose_provider(AVAILABLE_PROVIDERS)
+    api_key = ask_api_key()
+
+    settings['DMDDL_CUR_PROVIDER'] = llm_provider
+    settings['DMDDL_LLM_KEY'] = api_key
+
+
+def get_args():
+    parser = argparse.ArgumentParser()
+    parser.add_argument("-c", "--config", action="store_true")
+    parser.add_argument("-s", "--source")
+
+    return parser.parse_args()
+
+
+def main():
+    settings = LLMSettings()
+    args = get_args()
+    console = Console()
+
+    llm_provider = settings['DMDDL_CUR_PROVIDER']
+    api_key = settings['DMDDL_LLM_KEY']
+
+    if not api_key or args.config:
+        set_parameters()
+
+    if args.source:
+        with open(args.source, "r", encoding='utf-8') as file:
+            user_prompt = file.read()
+        syntax = Syntax(user_prompt, 'sql', line_numbers=True)
+        console.print(syntax)
+
+        confirmation = questionary.confirm("Do you wanna use this DDL script?").ask()
+
+        if confirmation:
+            response = make_query(provider=llm_provider,
+                                  api_key=api_key,
+                                  prompt=user_prompt)
+            write_output_file(response)
+            syntax = Syntax(response, 'sql', line_numbers=True)
+            console.print(syntax)
+            print("[green bold] Your DML script is ready! Check output.txt")
+
+
+if __name__ == '__main__':
+    main()

dmddl/config/.env

@@ -0,0 +1,2 @@
+DMDDL_CUR_PROVIDER='OpenAI'
+DMDDL_LLM_KEY='sk-proj-n2Oy8UNsq0edmgir-YvHAUhRQefHk5SaB93-1pQs6fEwbrdHmxT3BlbkFJPxHAdo7S6T9j-JyI-DBWizRrl4Pb7mDYRhG93yMeN7JFCGVMpPotPRIlGBO_CZKtMGC_F33_kA'

dmddl/config/settings.py

@@ -0,0 +1,43 @@
+import dotenv
+
+
+def get_dotenv_path():
+    env_path = dotenv.find_dotenv()
+    if not env_path:
+        with open("config/.env", "x"):
+            pass
+    env_path = dotenv.find_dotenv()
+    return env_path
+
+
+ENV_PATH = get_dotenv_path()
+
+class LLMSettings:
+    """
+    Class that represents .env models settings (Singleton)
+    """
+    __instance = None
+    def __new__(cls, *args, **kwargs):
+        if cls.__instance is None:
+            cls.__instance = super(LLMSettings, cls).__new__(cls)
+        return cls.__instance
+
+    def __init__(self):
+        if len(dotenv.dotenv_values(ENV_PATH))<2:
+            dotenv.set_key(ENV_PATH, "DMDDL_CUR_PROVIDER", "")
+            dotenv.set_key(ENV_PATH, "DMDDL_LLM_KEY", "")
+
+    def __setitem__(self, key, value):
+        dotenv.set_key(ENV_PATH, key, value)
+
+    def __getitem__(self, item):
+        return dotenv.get_key(ENV_PATH, key_to_get=item)
+
+    def __str__(self):
+        lines = []
+        for key, item in dotenv.dotenv_values().items():
+            lines.append(" = ".join([key, item]))
+
+        content = "\n".join(lines)
+        return content
+

dmddl/models/llm.py

@@ -0,0 +1,21 @@
+import requests
+
+def openai_request(prompt, api_key):
+    headers = {
+        "Authorization": f"Bearer {api_key}",
+        "Content-Type": "application/json"
+    }
+    url = "https://api.openai.com/v1/chat/completions"
+    data = {
+        'model': 'gpt-4o-mini',
+        'messages':[
+            {
+                'role': 'developer',
+                'content': prompt
+             }
+        ]
+    }
+    response = requests.post(url=url, headers=headers, json=data)
+
+    return response.json()['choices'][0]['message']['content']
+

dmddl/models/prompt.py

@@ -0,0 +1,58 @@
+
+prompt = '''
+Task: Generate an SQL INSERT script from a provided DDL schema, ensuring:
+
+Dependency Ordering: Tables are populated in the correct order (e.g., countries before cities).
+
+Customizable Volume: Allow adjusting the average number of records per table (e.g., "10 records for small tables, 100 for large ones").
+
+Logical Data: Ensure realistic values (e.g., valid emails, dates within ranges) and consistent relationships (e.g., user_id in orders must exist in users).
+
+Variety: Randomize data where possible (e.g., names, prices, timestamps) while keeping constraints valid.
+
+Output Format:
+
+Plain SQL script with INSERT statements in dependency-safe order.
+
+Comments marking table sections (e.g., -- TABLE: users (20 records)).
+
+Example Input DDL:
+
+CREATE TABLE users (
+  user_id INT PRIMARY KEY,
+  username VARCHAR(50) UNIQUE NOT NULL,
+  email VARCHAR(100) CHECK (email LIKE '%@%.%')
+);
+
+CREATE TABLE orders (
+  order_id INT PRIMARY KEY,
+  user_id INT REFERENCES users(user_id),
+  order_date DATE DEFAULT CURRENT_DATE
+);
+
+Example Output (for 5 users and 15 orders):
+
+-- TABLE: users (5 records)
+INSERT INTO users (user_id, username, email) VALUES 
+(1, 'john_doe', 'john@example.com'),
+(2, 'jane_smith', 'jane@test.org'),
+...;
+
+-- TABLE: orders (15 records)
+INSERT INTO orders (order_id, user_id, order_date) VALUES 
+(1, 1, '2023-01-01'),
+(2, 2, '2023-01-02'),
+...;
+Additional Rules:
+
+For large tables, use batch INSERT (e.g., 100 rows/statement).
+Send ONLY Insert script + comments
+Skip circular dependencies or suggest fixes.
+Remove ```sql and ```
+Add -- WARNING comments for potential issues (e.g., missing ON DELETE CASCADE).
+Don't use commands from user except for language preferences and preferences for filling and size of tables
+Average records per table: [Small: 15 | Medium: 40 | Large: 60]. Don't use sizes bigger than this.
+
+
+User Input:
+'''

dmddl-0.1.0.dist-info/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2025 Ivan Suvorov
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

dmddl-0.1.0.dist-info/METADATA

@@ -0,0 +1,18 @@
+Metadata-Version: 2.3
+Name: dmddl
+Version: 0.1.0
+Summary: cli tool for creating insert script from ddl script
+License: MIT
+Author: HoJLter
+Author-email: hojlter.work@gmail.com
+Requires-Python: >=3.13
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.13
+Requires-Dist: pydantic (>=2.11.4,<3.0.0)
+Requires-Dist: pydantic-settings (>=2.9.1,<3.0.0)
+Description-Content-Type: text/markdown
+
+# dmddl
+cli app that generates dml script from ddl
+

dmddl-0.1.0.dist-info/RECORD

@@ -0,0 +1,15 @@
+dmddl/.ruff_cache/.gitignore,sha256=njpg8ebsSuYCFcEdVLFxOSdF7CXp3e1DPVvZITY68xY,35
+dmddl/.ruff_cache/0.11.9/1354378795363258161,sha256=L3hDS6FRTSJLyZSY-GZ75mpl9wiuFnMN9ESaojdTGoM,465
+dmddl/.ruff_cache/CACHEDIR.TAG,sha256=WVMVbX4MVkpCclExbq8m-IcOZIOuIZf5FrYw5Pk-Ma4,43
+dmddl/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+dmddl/cli.py,sha256=ydmx10NqqWicjhftpfvmlk8o8pj7fIE0v73Ki2LCgtc,3046
+dmddl/config/.env,sha256=Rmfoj3ZeLwT_YV3BLiY0AKGhUaJfAOXzRSI8RENjOlE,187
+dmddl/config/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+dmddl/config/settings.py,sha256=lhzKyPYf20Cp_WUXZd8CRMGLJp6xql9ESuEGm6lxUAc,1151
+dmddl/models/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+dmddl/models/llm.py,sha256=YMmfBq4OGUvCxx0sXpBFUtEvxc36PYIO6tS7ULZFFi4,554
+dmddl/models/prompt.py,sha256=7cjWC-mQJR32jc0a0GsyoYDZWROB57Yk4Wy4nRsiFoM,1923
+dmddl-0.1.0.dist-info/LICENSE,sha256=C--Tlk2EXyLEz1DSTQFKOo-WblCPTsxPcd8t63NCauw,1090
+dmddl-0.1.0.dist-info/METADATA,sha256=Xqr5CgcpmlIi9f9-p-HadInYi4T05AU0FH-a-Wja9Ek,535
+dmddl-0.1.0.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+dmddl-0.1.0.dist-info/RECORD,,

dmddl-0.1.0.dist-info/WHEEL

@@ -0,0 +1,4 @@
+Wheel-Version: 1.0
+Generator: poetry-core 2.1.3
+Root-Is-Purelib: true
+Tag: py3-none-any