xenfra-sdk 0.1.1__py3-none-any.whl → 0.1.3__py3-none-any.whl

xenfra_sdk/db/session.py

@@ -1,30 +1,30 @@
-# src/xenfra/db/session.py
-
-import os
-from pathlib import Path
-
-import click
-from sqlmodel import Session, SQLModel, create_engine
-
-# Get the app directory for the current user
-app_dir = Path(click.get_app_dir("xenfra"))
-app_dir.mkdir(parents=True, exist_ok=True)
-db_path = app_dir / "xenfra.db"
-
-# For now, we will use a simple SQLite database for ease of setup.
-# In production, this should be a PostgreSQL database URL from environment variables.
-DATABASE_URL = os.getenv("DATABASE_URL", f"sqlite:///{db_path}")
-
-# Only echo SQL in development (set SQL_ECHO=1 to enable)
-SQL_ECHO = os.getenv("SQL_ECHO", "0") == "1"
-
-engine = create_engine(DATABASE_URL, echo=SQL_ECHO)
-
-
-def create_db_and_tables():
-    SQLModel.metadata.create_all(engine)
-
-
-def get_session():
-    with Session(engine) as session:
-        yield session
+# src/xenfra/db/session.py
+
+import os
+from pathlib import Path
+
+import click
+from sqlmodel import Session, SQLModel, create_engine
+
+# Get the app directory for the current user
+app_dir = Path(click.get_app_dir("xenfra"))
+app_dir.mkdir(parents=True, exist_ok=True)
+db_path = app_dir / "xenfra.db"
+
+# For now, we will use a simple SQLite database for ease of setup.
+# In production, this should be a PostgreSQL database URL from environment variables.
+DATABASE_URL = os.getenv("DATABASE_URL", f"sqlite:///{db_path}")
+
+# Only echo SQL in development (set SQL_ECHO=1 to enable)
+SQL_ECHO = os.getenv("SQL_ECHO", "0") == "1"
+
+engine = create_engine(DATABASE_URL, echo=SQL_ECHO)
+
+
+def create_db_and_tables():
+    SQLModel.metadata.create_all(engine)
+
+
+def get_session():
+    with Session(engine) as session:
+        yield session

xenfra_sdk/dependencies.py

@@ -1,38 +1,39 @@
-# src/xenfra/dependencies.py
-
-from fastapi import Depends, HTTPException, status
-from fastapi.security import OAuth2PasswordBearer
-from sqlmodel import Session, select
-from xenfra.db.models import User
-from xenfra.db.session import get_session
-from xenfra.security import decode_token
-
-oauth2_scheme = OAuth2PasswordBearer(tokenUrl="/auth/token")
-
-
-def get_current_user(
-    token: str = Depends(oauth2_scheme), session: Session = Depends(get_session)
-) -> User:
-    credentials_exception = HTTPException(
-        status_code=status.HTTP_401_UNAUTHORIZED,
-        detail="Could not validate credentials",
-        headers={"WWW-Authenticate": "Bearer"},
-    )
-    payload = decode_token(token)
-    if payload is None:
-        raise credentials_exception
-
-    email: str = payload.get("sub")
-    if email is None:
-        raise credentials_exception
-
-    user = session.exec(select(User).where(User.email == email)).first()
-    if user is None:
-        raise credentials_exception
-    return user
-
-
-def get_current_active_user(current_user: User = Depends(get_current_user)) -> User:
-    if not current_user.is_active:
-        raise HTTPException(status_code=400, detail="Inactive user")
-    return current_user
+# src/xenfra/dependencies.py
+
+from fastapi import Depends, HTTPException, status
+from fastapi.security import OAuth2PasswordBearer
+from sqlmodel import Session, select
+
+from xenfra.db.models import User
+from xenfra.db.session import get_session
+from xenfra.security import decode_token
+
+oauth2_scheme = OAuth2PasswordBearer(tokenUrl="/auth/token")
+
+
+def get_current_user(
+    token: str = Depends(oauth2_scheme), session: Session = Depends(get_session)
+) -> User:
+    credentials_exception = HTTPException(
+        status_code=status.HTTP_401_UNAUTHORIZED,
+        detail="Could not validate credentials",
+        headers={"WWW-Authenticate": "Bearer"},
+    )
+    payload = decode_token(token)
+    if payload is None:
+        raise credentials_exception
+
+    email: str = payload.get("sub")
+    if email is None:
+        raise credentials_exception
+
+    user = session.exec(select(User).where(User.email == email)).first()
+    if user is None:
+        raise credentials_exception
+    return user
+
+
+def get_current_active_user(current_user: User = Depends(get_current_user)) -> User:
+    if not current_user.is_active:
+        raise HTTPException(status_code=400, detail="Inactive user")
+    return current_user

xenfra_sdk/dockerizer.py

@@ -1,87 +1,87 @@
-import os
-from pathlib import Path
-
-from jinja2 import Environment, FileSystemLoader
-
-
-def detect_framework(path="."):
-    """
-    Scans common Python project structures to guess the framework and entrypoint.
-    Returns: (framework_name, default_port, start_command) or (None, None, None)
-    """
-    project_root = Path(path).resolve()
-
-    # Check for Django first (common pattern: manage.py in root)
-    if (project_root / "manage.py").is_file():
-        project_name = project_root.name
-        return "django", 8000, f"gunicorn {project_name}.wsgi:application --bind 0.0.0.0:8000"
-
-    candidate_files = []
-
-    # Check directly in project root
-    for name in ["main.py", "app.py"]:
-        if (project_root / name).is_file():
-            candidate_files.append(project_root / name)
-
-    # Check in src/*/ (standard package layout)
-    for src_dir in project_root.glob("src/*"):
-        if src_dir.is_dir():
-            for name in ["main.py", "app.py"]:
-                if (src_dir / name).is_file():
-                    candidate_files.append(src_dir / name)
-
-    for file_path in candidate_files:
-        with open(file_path, "r") as f:
-            content = f.read()
-
-            module_name = str(file_path.relative_to(project_root)).replace(os.sep, ".")[:-3]
-            if module_name.startswith("src."):
-                module_name = module_name[4:]
-
-            if "FastAPI" in content:
-                return "fastapi", 8000, f"uvicorn {module_name}:app --host 0.0.0.0 --port 8000"
-
-            if "Flask" in content:
-                return "flask", 5000, f"gunicorn {module_name}:app -b 0.0.0.0:5000"
-
-    return None, None, None
-
-
-def generate_templated_assets(context: dict):
-    """
-    Generates deployment assets (Dockerfile, docker-compose.yml) using Jinja2 templates.
-
-    Args:
-        context: A dictionary containing information for rendering templates,
-                 e.g., {'database': 'postgres', 'python_version': 'python:3.11-slim'}
-    """
-    # Path to the templates directory
-    template_dir = Path(__file__).parent / "templates"
-    env = Environment(loader=FileSystemLoader(template_dir))
-
-    # Detect framework specifics
-    framework, port, command = detect_framework()
-    if not framework:
-        print("Warning: No recognizable web framework detected.")
-        return []
-
-    # Merge detected context with provided context
-    render_context = {"port": port, "command": command, **context}
-
-    generated_files = []
-
-    # --- 1. Dockerfile ---
-    dockerfile_template = env.get_template("Dockerfile.j2")
-    dockerfile_content = dockerfile_template.render(render_context)
-    with open("Dockerfile", "w") as f:
-        f.write(dockerfile_content)
-    generated_files.append("Dockerfile")
-
-    # --- 2. docker-compose.yml ---
-    compose_template = env.get_template("docker-compose.yml.j2")
-    compose_content = compose_template.render(render_context)
-    with open("docker-compose.yml", "w") as f:
-        f.write(compose_content)
-    generated_files.append("docker-compose.yml")
-
-    return generated_files
+import os
+from pathlib import Path
+
+from jinja2 import Environment, FileSystemLoader
+
+
+def detect_framework(path="."):
+    """
+    Scans common Python project structures to guess the framework and entrypoint.
+    Returns: (framework_name, default_port, start_command) or (None, None, None)
+    """
+    project_root = Path(path).resolve()
+
+    # Check for Django first (common pattern: manage.py in root)
+    if (project_root / "manage.py").is_file():
+        project_name = project_root.name
+        return "django", 8000, f"gunicorn {project_name}.wsgi:application --bind 0.0.0.0:8000"
+
+    candidate_files = []
+
+    # Check directly in project root
+    for name in ["main.py", "app.py"]:
+        if (project_root / name).is_file():
+            candidate_files.append(project_root / name)
+
+    # Check in src/*/ (standard package layout)
+    for src_dir in project_root.glob("src/*"):
+        if src_dir.is_dir():
+            for name in ["main.py", "app.py"]:
+                if (src_dir / name).is_file():
+                    candidate_files.append(src_dir / name)
+
+    for file_path in candidate_files:
+        with open(file_path, "r") as f:
+            content = f.read()
+
+            module_name = str(file_path.relative_to(project_root)).replace(os.sep, ".")[:-3]
+            if module_name.startswith("src."):
+                module_name = module_name[4:]
+
+            if "FastAPI" in content:
+                return "fastapi", 8000, f"uvicorn {module_name}:app --host 0.0.0.0 --port 8000"
+
+            if "Flask" in content:
+                return "flask", 5000, f"gunicorn {module_name}:app -b 0.0.0.0:5000"
+
+    return None, None, None
+
+
+def generate_templated_assets(context: dict):
+    """
+    Generates deployment assets (Dockerfile, docker-compose.yml) using Jinja2 templates.
+
+    Args:
+        context: A dictionary containing information for rendering templates,
+                 e.g., {'database': 'postgres', 'python_version': 'python:3.11-slim'}
+    """
+    # Path to the templates directory
+    template_dir = Path(__file__).parent / "templates"
+    env = Environment(loader=FileSystemLoader(template_dir))
+
+    # Detect framework specifics
+    framework, port, command = detect_framework()
+    if not framework:
+        print("Warning: No recognizable web framework detected.")
+        return []
+
+    # Merge detected context with provided context
+    render_context = {"port": port, "command": command, **context}
+
+    generated_files = []
+
+    # --- 1. Dockerfile ---
+    dockerfile_template = env.get_template("Dockerfile.j2")
+    dockerfile_content = dockerfile_template.render(render_context)
+    with open("Dockerfile", "w") as f:
+        f.write(dockerfile_content)
+    generated_files.append("Dockerfile")
+
+    # --- 2. docker-compose.yml ---
+    compose_template = env.get_template("docker-compose.yml.j2")
+    compose_content = compose_template.render(render_context)
+    with open("docker-compose.yml", "w") as f:
+        f.write(compose_content)
+    generated_files.append("docker-compose.yml")
+
+    return generated_files