legitdata_project/legitdata/__init__.py

"""
LegitData - Realistic synthetic data generation for analytics warehouses.

Usage:
    from legitdata import LegitGenerator
    
    gen = LegitGenerator(
        url="https://example.com",
        use_case="Retail Analytics",
        connection_string="snowflake://user:pass@account/db/schema"
    )
    gen.load_ddl("schema.sql")
    gen.generate(size="medium")
"""

from .generator import LegitGenerator
from .config import SIZE_PRESETS, USE_CASES, GenerationConfig
from .ddl import parse_ddl, parse_ddl_file, Schema, Table, Column, ColumnClassification
from .analyzer import CompanyContext
from .writers import SnowflakeWriter, DryRunWriter
from .storyspec import StorySpec, build_story_spec, build_story_bundle

__version__ = "0.1.0"

__all__ = [
    'LegitGenerator',
    'SIZE_PRESETS',
    'USE_CASES',
    'GenerationConfig',
    'parse_ddl',
    'parse_ddl_file',
    'Schema',
    'Table',
    'Column',
    'ColumnClassification',
    'CompanyContext',
    'SnowflakeWriter',
    'DryRunWriter',
    'StorySpec',
    'build_story_spec',
    'build_story_bundle',
]