datasecops-cli 0.1.0__py3-none-any.whl

datasecops_cli/services/snowflake_service.py

@@ -0,0 +1,62 @@
+import json
+from typing import Any, Optional
+import snowflake.connector
+
+from datasecops_cli.models.project_config import DatasecopsConfig
+
+
+class SnowflakeService:
+    """Connects to Snowflake and calls native app API procedures."""
+    
+    def __init__(self, config: DatasecopsConfig):
+        self.config = config
+        self._conn: Optional[snowflake.connector.SnowflakeConnection] = None
+    
+    def connect(self) -> None:
+        self._conn = snowflake.connector.connect(
+            connection_name=self.config.connection_name,
+        )
+    
+    def close(self) -> None:
+        if self._conn:
+            self._conn.close()
+            self._conn = None
+    
+    @property
+    def conn(self) -> snowflake.connector.SnowflakeConnection:
+        if self._conn is None:
+            self.connect()
+        return self._conn
+    
+    def execute_query(self, sql: str, params: list = None) -> list[dict]:
+        cur = self.conn.cursor(snowflake.connector.DictCursor)
+        try:
+            cur.execute(sql, params or [])
+            return cur.fetchall()
+        finally:
+            cur.close()
+    
+    def call_procedure(self, proc_name: str, *args) -> Any:
+        db = self.config.app_database
+        arg_placeholders = ", ".join(["%s"] * len(args))
+        sql = f"CALL {db}.api.{proc_name}({arg_placeholders})"
+        rows = self.execute_query(sql, list(args))
+        if rows:
+            first_val = list(rows[0].values())[0]
+            if isinstance(first_val, str):
+                try:
+                    return json.loads(first_val)
+                except (json.JSONDecodeError, TypeError):
+                    return first_val
+            return first_val
+        return None
+    
+    def get_framework_config(self, config_code: str) -> Optional[dict]:
+        return self.call_procedure("get_framework_config", config_code)
+    
+    def get_project_profiles(self) -> Optional[list[dict]]:
+        return self.call_procedure("get_project_profiles")
+    
+    def get_account_info(self) -> dict:
+        rows = self.execute_query("SELECT CURRENT_ACCOUNT_NAME() as account, CURRENT_USER() as user_name")
+        return rows[0] if rows else {}

datasecops_cli/utilities/__init__.py

@@ -0,0 +1 @@
+

datasecops_cli/utilities/display.py

@@ -0,0 +1,122 @@
+import os
+from colorama import init, Fore, Style
+
+init(autoreset=True)
+
+
+def clear():
+    os.system('cls' if os.name == 'nt' else 'clear')
+
+
+def print_detail(message: str, color: str = Fore.WHITE) -> None:
+    print(Style.NORMAL + color + message)
+
+
+def print_long_line(color: str = Fore.GREEN) -> None:
+    print_detail("=" * 76, color)
+
+
+def section_header(message: str, active_profile: str = None, current_branch: str = None) -> None:
+    print_long_line(Fore.GREEN)
+    print_detail(message, Fore.GREEN)
+    print_long_line(Fore.GREEN)
+    if active_profile:
+        print_detail(f"  Profile: {active_profile}", Fore.GREEN)
+    if current_branch:
+        print_detail(f"  Branch:  {current_branch}", Fore.GREEN)
+    if active_profile or current_branch:
+        print_long_line(Fore.GREEN)
+
+
+def display_action_header(message: str) -> None:
+    clear()
+    print_long_line(Fore.GREEN)
+    print_detail(f"  {message}", Fore.GREEN)
+    print_long_line(Fore.GREEN)
+
+
+def menu_option(number: int, text: str) -> None:
+    print_detail(f"  [{number}] {text}", Fore.WHITE)
+
+
+def info_line(message: str) -> None:
+    print_detail(message, Fore.BLUE)
+
+
+def warning_line(message: str) -> None:
+    print_detail(f"WARNING: {message}", Fore.YELLOW)
+
+
+def error_line(message: str) -> None:
+    print_detail(f"ERROR: {message}", Fore.RED)
+
+
+def success_line(message: str) -> None:
+    print_detail(message, Fore.GREEN)
+
+
+def complete_action(message: str = None) -> None:
+    print_long_line()
+    input(message or "Press Enter to continue...")
+
+
+def get_input_string(message: str, allow_empty: bool = False) -> str:
+    while True:
+        result = input(message)
+        if result.strip() or allow_empty:
+            return result.strip()
+
+
+def get_input_number(message: str) -> int:
+    while True:
+        result = input(message).strip()
+        if result.isdigit() or (result.startswith('-') and result[1:].isdigit()):
+            return int(result)
+        if not result:
+            return 0
+
+
+def get_input_true_false(message: str, default: str = "n") -> bool:
+    while True:
+        result = input(f"{message} (y/n, default: {default}): ").strip().lower()
+        if not result:
+            result = default
+        if result in ("y", "n"):
+            return result == "y"
+
+
+def get_input_string_or_default(message: str, default_value: str) -> str:
+    result = input(f"{message} (default: '{default_value}'): ").strip()
+    return result if result else default_value
+
+
+def select_from_dict(options: dict[str, str], select_name: str, add_back: bool = True) -> str:
+    """Numbered selection from a dict. Returns the selected value. '0' returns 'back'."""
+    items = list(options.items())
+    if add_back and "back" not in options:
+        items.append(("back", "back"))
+
+    for idx, (key, _) in enumerate(items):
+        if key == "back":
+            print_detail(f"  [0] {key}", Fore.WHITE)
+        else:
+            print_detail(f"  [{idx + 1}] {key}", Fore.WHITE)
+
+    while True:
+        raw = get_input_string(f"Select {select_name}: ")
+        if raw == "0":
+            return "back"
+        try:
+            num = int(raw) - 1
+            if 0 <= num < len(items) and items[num][0] != "back":
+                info_line(f"Selected {select_name}: {items[num][0]}")
+                return items[num][1]
+        except ValueError:
+            pass
+        warning_line(f"Please select a valid {select_name} number")
+
+
+def select_from_list(options: list[str], select_name: str, add_back: bool = True) -> str:
+    """Numbered selection from a list. Returns the selected value."""
+    d = {opt: opt for opt in options}
+    return select_from_dict(d, select_name, add_back)

datasecops_cli/utilities/file_utils.py

@@ -0,0 +1,33 @@
+from pathlib import Path
+
+
+def ensure_dir(path: Path) -> Path:
+    path.mkdir(parents=True, exist_ok=True)
+    return path
+
+
+def read_file(path: Path) -> str:
+    if path.exists():
+        return path.read_text(encoding="utf-8")
+    return ""
+
+
+def write_file(path: Path, content: str) -> None:
+    ensure_dir(path.parent)
+    path.write_text(content, encoding="utf-8")
+
+
+def file_exists(path: Path) -> bool:
+    return path.exists() and path.is_file()
+
+
+def find_files(root: Path, pattern: str) -> list[Path]:
+    return sorted(root.rglob(pattern))
+
+
+def get_dbt_profiles_dir() -> Path:
+    return Path.home() / ".dbt"
+
+
+def get_cortex_skills_dir() -> Path:
+    return Path.home() / ".cortex" / "skills"

datasecops_cli/utilities/yaml_utils.py

@@ -0,0 +1,39 @@
+from pathlib import Path
+from typing import Any, Optional
+import yaml
+
+
+def read_yaml(path: Path) -> Optional[dict]:
+    if not path.exists():
+        return None
+    with open(path, "r", encoding="utf-8") as f:
+        return yaml.safe_load(f)
+
+
+def write_yaml(path: Path, data: Any, sort_keys: bool = False) -> None:
+    path.parent.mkdir(parents=True, exist_ok=True)
+    with open(path, "w", encoding="utf-8") as f:
+        yaml.dump(data, f, default_flow_style=False, sort_keys=sort_keys)
+
+
+def read_datasecops_config(project_dir: Path) -> Optional[dict]:
+    return read_yaml(project_dir / ".datasecops.yml")
+
+
+def write_datasecops_config(project_dir: Path, config: dict) -> None:
+    write_yaml(project_dir / ".datasecops.yml", config)
+
+
+def read_dbt_project(project_dir: Path) -> Optional[dict]:
+    for name in ["dbt_project.yml", "dbt/dbt_project.yml"]:
+        p = project_dir / name
+        if p.exists():
+            return read_yaml(p)
+    return None
+
+
+def get_profile_name(project_dir: Path) -> str:
+    data = read_dbt_project(project_dir)
+    if data:
+        return data.get("profile", "")
+    return ""

datasecops_cli-0.1.0.dist-info/METADATA

@@ -0,0 +1,16 @@
+Metadata-Version: 2.4
+Name: datasecops-cli
+Version: 0.1.0
+Summary: DataSecOps Framework CLI for Snowflake Native App
+License-Expression: MIT
+License-File: LICENSE
+Requires-Python: >=3.10
+Requires-Dist: colorama>=0.4
+Requires-Dist: gitpython>=3.1
+Requires-Dist: pydantic>=2.0
+Requires-Dist: pyyaml>=6.0
+Requires-Dist: snowflake-connector-python>=3.0
+Requires-Dist: sqlfluff>=3.0
+Provides-Extra: test
+Requires-Dist: pytest-cov>=4.0; extra == 'test'
+Requires-Dist: pytest>=7.0; extra == 'test'

datasecops_cli-0.1.0.dist-info/RECORD

@@ -0,0 +1,26 @@
+datasecops_cli/__init__.py,sha256=kUR5RAFc7HCeiqdlX36dZOHkUI5wI6V_43RpEcD8b-0,22
+datasecops_cli/config.py,sha256=2e0gdNvbuv-9EZgTrF9IG9KId-jFmPL3tgCa2X5442w,4222
+datasecops_cli/main.py,sha256=MrI3rr4iMLuvZXPghoajFWtmchU6gfYkPDMKF7hm1kM,4430
+datasecops_cli/menus/__init__.py,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
+datasecops_cli/menus/development.py,sha256=kYAklLdpJrA8rIbbE3dfmfGejsL4kF00uz4sbhIZ8nM,7032
+datasecops_cli/menus/downloads.py,sha256=VufM5Sl2wkIYGbEKU1NRPdmjAUbqykNo9vL34Hd7FuY,3474
+datasecops_cli/menus/git_operations.py,sha256=nhFnmHK3JE-2rbbd6hzwQm_R9eekr5tBLrFMWuSPe_k,8470
+datasecops_cli/models/__init__.py,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
+datasecops_cli/models/git_helpers.py,sha256=Mi-5dnlbvNjBTZGEOG6ChA5_FupiW86ckMH8ANjjy-c,714
+datasecops_cli/models/project_config.py,sha256=87YaVP9P_pF0FMk_-XZGNTKrrB2pv_2Y61qwHj1yEMU,2860
+datasecops_cli/services/__init__.py,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
+datasecops_cli/services/dbt_runner.py,sha256=xfOvslXPPYtRb5vQPhqH655mQXNL9Xt0m9wgnnSl5rw,5123
+datasecops_cli/services/download_service.py,sha256=roMPgpBWWBRICa4f8HrqxuIG9bk4ifF69Z5QvhjTUTQ,4122
+datasecops_cli/services/git_service.py,sha256=wiVSxD-jyp-Rt4UA9RgRCluruWhZdSSeM_LNyDMWYxY,7158
+datasecops_cli/services/linting_service.py,sha256=4r5PdnjIZ2zAtGDJpQE19xu0iJKH8IoRhuWewPJyIe4,1743
+datasecops_cli/services/skill_service.py,sha256=eOTKay7pKxZCZq4mCQDPc0d_zRINZ-KxGcYWj1wb-Bs,3427
+datasecops_cli/services/snowflake_service.py,sha256=FXZ_zoTy1wGjq9JB0BvIh5XW-d-Gk2PrXHnEfFeawZ8,2167
+datasecops_cli/utilities/__init__.py,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
+datasecops_cli/utilities/display.py,sha256=f07CfRJXRPxoVj6byyt6o1Foo7wS2JeZFzi7bS8U7E0,3675
+datasecops_cli/utilities/file_utils.py,sha256=BeYghjXFe2n9JiVchZaV8KAFr7V_q9MkuCGuZCo3qYs,705
+datasecops_cli/utilities/yaml_utils.py,sha256=dty6t4zbn5PQWNqFSRjXsJtLF2qGxJzGUQfzzduJRGU,1122
+datasecops_cli-0.1.0.dist-info/METADATA,sha256=inaMVWUw9etoRi7CIHIkEYbhjgRNOaxWTEl33dCAFIM,491
+datasecops_cli-0.1.0.dist-info/WHEEL,sha256=QccIxa26bgl1E6uMy58deGWi-0aeIkkangHcxk2kWfw,87
+datasecops_cli-0.1.0.dist-info/entry_points.txt,sha256=wv98kE6CKKApNQkW5-wU9JpMa0i2Llsl3Gmp6tn_Hvs,56
+datasecops_cli-0.1.0.dist-info/licenses/LICENSE,sha256=uBqkXEuQ68MBM2A-Ea6UVcxVM0LUt1rGtkDOp69XQ1k,1071
+datasecops_cli-0.1.0.dist-info/RECORD,,

datasecops_cli-0.1.0.dist-info/WHEEL

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

datasecops_cli-0.1.0.dist-info/entry_points.txt

@@ -0,0 +1,2 @@
+[console_scripts]
+datasecops = datasecops_cli.main:main

datasecops_cli-0.1.0.dist-info/licenses/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2026 Data Engineers
+
+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.