graphlagoon 0.8.4__tar.gz

graphlagoon-0.8.4/.env.databricks

@@ -0,0 +1,17 @@
+# Databricks mode WITHOUT local PostgreSQL
+# Copy to .env and fill in your values
+
+# Disable local sql-warehouse - connect directly to Databricks
+GRAPH_LAGOON_DATABRICKS_MODE=true
+GRAPH_LAGOON_DATABRICKS_HOST=adb-XXXXX.X.azuredatabricks.net
+GRAPH_LAGOON_DATABRICKS_TOKEN=dapi-XXXXXXXXXXXXX
+GRAPH_LAGOON_DATABRICKS_WAREHOUSE_ID=XXXXXXXXXXXXXXXX
+GRAPH_LAGOON_DATABRICKS_CATALOG=main
+GRAPH_LAGOON_DATABRICKS_SCHEMA=default
+
+# Disable PostgreSQL (contexts/explorations stored in browser localStorage)
+GRAPH_LAGOON_DATABASE_ENABLED=false
+
+# General settings
+GRAPH_LAGOON_DEV_MODE=false
+GRAPH_LAGOON_PORT=8000

graphlagoon-0.8.4/.env.databricks-localdb

@@ -0,0 +1,18 @@
+# Databricks mode WITH local PostgreSQL
+# Copy to .env and fill in your values
+
+# Connect directly to Databricks SQL Warehouse
+GRAPH_LAGOON_DATABRICKS_MODE=true
+GRAPH_LAGOON_DATABRICKS_HOST=adb-XXXXX.X.azuredatabricks.net
+GRAPH_LAGOON_DATABRICKS_TOKEN=dapi-XXXXXXXXXXXXX
+GRAPH_LAGOON_DATABRICKS_WAREHOUSE_ID=XXXXXXXXXXXXXXXX
+GRAPH_LAGOON_DATABRICKS_CATALOG=main
+GRAPH_LAGOON_DATABRICKS_SCHEMA=default
+
+# Use local PostgreSQL for storing contexts/explorations
+GRAPH_LAGOON_DATABASE_ENABLED=true
+GRAPH_LAGOON_DATABASE_URL=postgresql+asyncpg://sgraph:sgraph@localhost:5432/sgraph
+
+# General settings
+GRAPH_LAGOON_DEV_MODE=false
+GRAPH_LAGOON_PORT=8000

graphlagoon-0.8.4/.env.example

@@ -0,0 +1,24 @@
+# Graph Lagoon Studio Configuration
+# All variables use GRAPH_LAGOON_ prefix to avoid conflicts with other applications
+
+# Local sql-warehouse (dev mode)
+GRAPH_LAGOON_SQL_WAREHOUSE_URL=http://localhost:8001
+
+# Database (optional - disabled by default, uses localStorage on frontend)
+# Set GRAPH_LAGOON_DATABASE_ENABLED=true to enable PostgreSQL persistence
+GRAPH_LAGOON_DATABASE_URL=postgresql+asyncpg://sgraph:sgraph@localhost:5432/sgraph
+GRAPH_LAGOON_DATABASE_ENABLED=false
+
+# General settings
+GRAPH_LAGOON_DEV_MODE=true
+GRAPH_LAGOON_PORT=8000
+
+# Databricks connection (production mode)
+# Set GRAPH_LAGOON_DATABRICKS_MODE=true to connect directly to Databricks
+# When GRAPH_LAGOON_DATABRICKS_MODE=true, CATALOG and SCHEMA are required
+GRAPH_LAGOON_DATABRICKS_MODE=false
+GRAPH_LAGOON_DATABRICKS_HOST=
+GRAPH_LAGOON_DATABRICKS_TOKEN=
+GRAPH_LAGOON_DATABRICKS_WAREHOUSE_ID=
+GRAPH_LAGOON_DATABRICKS_CATALOG=
+GRAPH_LAGOON_DATABRICKS_SCHEMA=

graphlagoon-0.8.4/.gitignore

@@ -0,0 +1,105 @@
+scripts/ieee_cis_fraud/data/
+frontend/playwright-report-integration/trace/
+.vitepres
+frontend/ext-3d-force_raw
+frontend/playwright-report-integration/data
+frontend/test-results-integration/.last-run.json
+frontend/test-results-integration
+docs/dew
+
+# =============================================================================
+# Project-specific
+# =============================================================================
+frontend/ext-3d-force/exp
+.logs
+.pids
+uv.toml
+metastore_db/
+derby.log
+spark-warehouse/
+alembic/versions/
+
+# IDE / Editor
+.vscode
+.idea
+*.swp
+
+# Claude Code
+.claude
+
+# =============================================================================
+# Python
+# =============================================================================
+__pycache__/
+*.py[codz]
+*$py.class
+*.so
+.Python
+build/
+dist/
+*.egg-info/
+*.egg
+.installed.cfg
+MANIFEST
+.venv
+venv/
+env/
+
+# Testing
+htmlcov/
+.tox/
+.nox/
+.coverage
+.coverage.*
+.cache
+.pytest_cache/
+coverage/
+nosetests.xml
+coverage.xml
+
+# Type checkers
+.mypy_cache/
+.pytype/
+.ruff_cache/
+
+# =============================================================================
+# Node.js
+# =============================================================================
+node_modules/
+
+# =============================================================================
+# Environment
+# =============================================================================
+.env
+.envrc
+*.log
+
+# =============================================================================
+# OS
+# =============================================================================
+**/.DS_Store
+
+# =============================================================================
+# Playwright
+# =============================================================================
+playwright-report/
+test-results/
+frontend/e2e/screenshots/
+
+# =============================================================================
+# Performance reports
+# =============================================================================
+frontend/perf-report.json
+
+# =============================================================================
+# Build artifacts (generated by make build)
+# =============================================================================
+api/graphlagoon/static/assets/
+api/graphlagoon/static/.vite/
+api/graphlagoon/static/index.html
+CHANGELOG-release.md
+
+# =============================================================================
+# Data (warehouse)
+# =============================================================================
+warehouse/data

graphlagoon-0.8.4/Makefile

@@ -0,0 +1,22 @@
+.PHONY: install run dev test clean migrate migrate-create
+
+install:
+	uv sync
+
+run:
+	uv run uvicorn src.main:app --host 0.0.0.0 --port 8000 --reload
+
+dev: run
+
+test:
+	uv run pytest -v
+
+migrate:
+	uv run alembic upgrade head
+
+migrate-create:
+	uv run alembic revision --autogenerate -m "$(name)"
+
+clean:
+	rm -rf .venv __pycache__ .pytest_cache
+	find . -type d -name "__pycache__" -exec rm -rf {} +

graphlagoon-0.8.4/PKG-INFO

@@ -0,0 +1,92 @@
+Metadata-Version: 2.4
+Name: graphlagoon
+Version: 0.8.4
+Summary: Graph visualization and exploration tool with FastAPI backend
+Project-URL: Homepage, https://github.com/graphlagoon/graphlagoon
+Project-URL: Documentation, https://github.com/graphlagoon/graphlagoon#readme
+Project-URL: Repository, https://github.com/graphlagoon/graphlagoon
+Author-email: Bruno Messias <devmessias@gmail.com>
+License: AGPL-3.0
+Classifier: Development Status :: 4 - Beta
+Classifier: Framework :: FastAPI
+Classifier: Intended Audience :: Developers
+Classifier: Intended Audience :: Science/Research
+Classifier: License :: OSI Approved :: GNU Affero General Public License v3
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Topic :: Scientific/Engineering :: Visualization
+Requires-Python: >=3.11
+Requires-Dist: asyncpg>=0.31.0
+Requires-Dist: fastapi>=0.109.0
+Requires-Dist: gsql2rsql
+Requires-Dist: httpx>=0.26.0
+Requires-Dist: jinja2>=3.1.0
+Requires-Dist: pydantic-settings>=2.1.0
+Requires-Dist: pydantic>=2.5.0
+Requires-Dist: python-dotenv>=1.0.0
+Requires-Dist: sqlalchemy[asyncio]>=2.0.0
+Requires-Dist: sqlglot>=26.0.0
+Requires-Dist: uvicorn[standard]>=0.27.0
+Provides-Extra: all
+Requires-Dist: alembic>=1.13.0; extra == 'all'
+Requires-Dist: asyncpg>=0.29.0; extra == 'all'
+Requires-Dist: databricks-sdk>=0.38.0; extra == 'all'
+Provides-Extra: dev
+Requires-Dist: debugpy>=1.8.0; extra == 'dev'
+Requires-Dist: pytest-asyncio>=0.23.0; extra == 'dev'
+Requires-Dist: pytest>=7.4.0; extra == 'dev'
+Requires-Dist: ruff>=0.4.0; extra == 'dev'
+Provides-Extra: lakebase
+Requires-Dist: databricks-sdk>=0.38.0; extra == 'lakebase'
+Provides-Extra: postgres
+Requires-Dist: alembic>=1.13.0; extra == 'postgres'
+Requires-Dist: asyncpg>=0.29.0; extra == 'postgres'
+Description-Content-Type: text/markdown
+
+# Graph Lagoon Studio
+
+Graph visualization and exploration tool with FastAPI backend.
+
+## Installation
+
+```bash
+pip install graphlagoon
+```
+
+## Quick Start
+
+```bash
+# Start the server
+uvicorn graphlagoon.app:app --host 0.0.0.0 --port 8000
+```
+
+## Usage
+
+### Standalone Mode
+
+```python
+from graphlagoon import create_app
+
+app = create_app()
+```
+
+### Embedded Mode
+
+```python
+from fastapi import FastAPI
+from graphlagoon import create_api_router
+
+app = FastAPI()
+app.include_router(create_api_router(), prefix="/graphlagoon")
+```
+
+## Configuration
+
+Environment variables:
+
+- `DATABASE_ENABLED` - Enable/disable PostgreSQL (default: true)
+- `DATABASE_URL` - PostgreSQL connection string
+- `DATABRICKS_MODE` - Connect to Databricks directly (default: false)
+- `DATABRICKS_HOST` - Databricks workspace host
+- `DATABRICKS_TOKEN` - Databricks access token
+- `DATABRICKS_WAREHOUSE_ID` - SQL warehouse ID

graphlagoon-0.8.4/README.md

@@ -0,0 +1,47 @@
+# Graph Lagoon Studio
+
+Graph visualization and exploration tool with FastAPI backend.
+
+## Installation
+
+```bash
+pip install graphlagoon
+```
+
+## Quick Start
+
+```bash
+# Start the server
+uvicorn graphlagoon.app:app --host 0.0.0.0 --port 8000
+```
+
+## Usage
+
+### Standalone Mode
+
+```python
+from graphlagoon import create_app
+
+app = create_app()
+```
+
+### Embedded Mode
+
+```python
+from fastapi import FastAPI
+from graphlagoon import create_api_router
+
+app = FastAPI()
+app.include_router(create_api_router(), prefix="/graphlagoon")
+```
+
+## Configuration
+
+Environment variables:
+
+- `DATABASE_ENABLED` - Enable/disable PostgreSQL (default: true)
+- `DATABASE_URL` - PostgreSQL connection string
+- `DATABRICKS_MODE` - Connect to Databricks directly (default: false)
+- `DATABRICKS_HOST` - Databricks workspace host
+- `DATABRICKS_TOKEN` - Databricks access token
+- `DATABRICKS_WAREHOUSE_ID` - SQL warehouse ID

graphlagoon-0.8.4/alembic/env.py

@@ -0,0 +1,87 @@
+import asyncio
+from logging.config import fileConfig
+
+from sqlalchemy import pool
+from sqlalchemy.engine import Connection
+from sqlalchemy.ext.asyncio import async_engine_from_config
+
+from alembic import context
+
+from graphlagoon.config import get_settings
+from graphlagoon.db.database import Base
+
+config = context.config
+
+if config.config_file_name is not None:
+    fileConfig(config.config_file_name)
+
+target_metadata = Base.metadata
+
+settings = get_settings()
+
+if settings.lakebase_enabled:
+    from graphlagoon.db.lakebase import build_lakebase_url
+
+    config.set_main_option("sqlalchemy.url", build_lakebase_url(settings))
+else:
+    config.set_main_option("sqlalchemy.url", settings.database_url)
+
+
+def run_migrations_offline() -> None:
+    """Run migrations in 'offline' mode."""
+    url = config.get_main_option("sqlalchemy.url")
+    context.configure(
+        url=url,
+        target_metadata=target_metadata,
+        literal_binds=True,
+        dialect_opts={"paramstyle": "named"},
+    )
+
+    with context.begin_transaction():
+        context.run_migrations()
+
+
+def do_run_migrations(connection: Connection) -> None:
+    context.configure(connection=connection, target_metadata=target_metadata)
+
+    with context.begin_transaction():
+        context.run_migrations()
+
+
+async def run_async_migrations() -> None:
+    """Run migrations in 'online' mode with async engine."""
+    engine_kwargs = {}
+    if settings.lakebase_enabled:
+        server_settings = {"application_name": "graphlagoon-alembic"}
+        if settings.default_postgres_schema:
+            server_settings["search_path"] = settings.default_postgres_schema
+        engine_kwargs["connect_args"] = {
+            "ssl": "require",
+            "server_settings": server_settings,
+        }
+
+    connectable = async_engine_from_config(
+        config.get_section(config.config_ini_section, {}),
+        prefix="sqlalchemy.",
+        poolclass=pool.NullPool,
+        **engine_kwargs,
+    )
+
+    async with connectable.connect() as connection:
+        await connection.run_sync(do_run_migrations)
+
+    await connectable.dispose()
+
+
+def run_migrations_online() -> None:
+    """Run migrations in 'online' mode."""
+    asyncio.run(run_async_migrations())
+
+
+if "connection" in config.attributes:
+    # Called programmatically from create_tables() — connection already exists
+    do_run_migrations(config.attributes["connection"])
+elif context.is_offline_mode():
+    run_migrations_offline()
+else:
+    run_migrations_online()

graphlagoon-0.8.4/alembic/script.py.mako

@@ -0,0 +1,26 @@
+"""${message}
+
+Revision ID: ${up_revision}
+Revises: ${down_revision | comma,n}
+Create Date: ${create_date}
+
+"""
+from typing import Sequence, Union
+
+from alembic import op
+import sqlalchemy as sa
+${imports if imports else ""}
+
+# revision identifiers, used by Alembic.
+revision: str = ${repr(up_revision)}
+down_revision: Union[str, None] = ${repr(down_revision)}
+branch_labels: Union[str, Sequence[str], None] = ${repr(branch_labels)}
+depends_on: Union[str, Sequence[str], None] = ${repr(depends_on)}
+
+
+def upgrade() -> None:
+    ${upgrades if upgrades else "pass"}
+
+
+def downgrade() -> None:
+    ${downgrades if downgrades else "pass"}

graphlagoon-0.8.4/alembic.ini

@@ -0,0 +1,40 @@
+[alembic]
+script_location = alembic
+prepend_sys_path = .
+version_path_separator = os
+
+[post_write_hooks]
+
+[loggers]
+keys = root,sqlalchemy,alembic
+
+[handlers]
+keys = console
+
+[formatters]
+keys = generic
+
+[logger_root]
+level = WARN
+handlers = console
+qualname =
+
+[logger_sqlalchemy]
+level = WARN
+handlers =
+qualname = sqlalchemy.engine
+
+[logger_alembic]
+level = INFO
+handlers =
+qualname = alembic
+
+[handler_console]
+class = StreamHandler
+args = (sys.stderr,)
+level = NOTSET
+formatter = generic
+
+[formatter_generic]
+format = %(levelname)-5.5s [%(name)s] %(message)s
+datefmt = %H:%M:%S

graphlagoon-0.8.4/graphlagoon/__init__.py

@@ -0,0 +1,114 @@
+"""Graph Lagoon Studio - Graph visualization and exploration tool for Spark.
+
+Usage as standalone app:
+    uvicorn graphlagoon.app:app --host 0.0.0.0 --port 8000
+
+Usage as importable module (API only):
+    from fastapi import FastAPI
+    from graphlagoon import create_api_router
+
+    app = FastAPI()
+    app.include_router(create_api_router(), prefix="/api")
+
+Usage as mountable sub-app (API + Frontend):
+    from fastapi import FastAPI
+    from graphlagoon import create_mountable_app, add_mount_redirect
+
+    app = FastAPI()
+    sgraph_app = create_mountable_app(mount_prefix="/graphlagoon")
+
+    # Add redirect for /graphlagoon -> /graphlagoon/
+    add_mount_redirect(app, "/graphlagoon")
+
+    app.mount("/graphlagoon", sgraph_app)
+    # Access at: http://localhost:8000/graphlagoon/
+
+Usage with Databricks (specifying catalog and schema):
+    from fastapi import FastAPI
+    from graphlagoon import create_mountable_app
+
+    app = FastAPI()
+    app.mount("/graphlagoon", create_mountable_app(
+        databricks_catalog="my_catalog",
+        databricks_schema="my_schema",
+    ))
+
+Usage with multiple catalog.schema pairs:
+    from fastapi import FastAPI
+    from graphlagoon import create_mountable_app
+
+    app = FastAPI()
+    app.mount("/graphlagoon", create_mountable_app(
+        catalog_schemas=[
+            ("catalog_a", "schema_1"),
+            ("catalog_b", "schema_2"),
+        ],
+    ))
+
+Usage with dynamic header provider (token refresh):
+    from fastapi import FastAPI
+    from graphlagoon import create_mountable_app, HeaderProvider
+
+    # Provider retorna o token (str), não o dict de headers
+    async def get_fresh_token() -> str:
+        return await my_token_service.get_token()
+
+    app = FastAPI()
+    app.mount("/graphlagoon", create_mountable_app(
+        header_provider=get_fresh_token,  # Token é convertido para Authorization: Bearer
+        databricks_catalog="my_catalog",
+        databricks_schema="my_schema",
+    ))
+
+Usage with custom user provider (integrate with parent app's auth):
+    from fastapi import FastAPI, Request
+    from graphlagoon import create_mountable_app, UserProvider
+
+    def get_user_from_parent(request: Request) -> str:
+        # Parent app's middleware sets request.state.current_user
+        return request.state.current_user.email
+
+    app = FastAPI()
+    app.mount("/graphlagoon", create_mountable_app(
+        user_provider=get_user_from_parent,
+        databricks_catalog="my_catalog",
+        databricks_schema="my_schema",
+    ))
+"""
+
+from graphlagoon.app import (
+    create_app,
+    create_api_router,
+    create_frontend_router,
+    create_mountable_app,
+    add_mount_redirect,
+)
+from graphlagoon.config import Settings, get_settings
+from graphlagoon.services.warehouse import HeaderProvider
+from graphlagoon.middleware.auth import UserProvider, configure_auth
+
+__all__ = [
+    "create_app",
+    "create_api_router",
+    "create_frontend_router",
+    "create_mountable_app",
+    "add_mount_redirect",
+    "Settings",
+    "get_settings",
+    "HeaderProvider",
+    "UserProvider",
+    "configure_auth",
+]
+
+
+def _get_version() -> str:
+    """Read version from pyproject.toml (single source of truth)."""
+    from importlib.metadata import version, PackageNotFoundError
+
+    try:
+        return version("graphlagoon")
+    except PackageNotFoundError:
+        return "0.0.0-dev"
+
+
+__version__ = _get_version()