from pydantic import BaseModel, Field
import sqlite3
import pandas as pd
from typing import Optional
from .metadata import Metadata

class Run(BaseModel):
    # === FETCHERS ===
    onet_conn: Optional[sqlite3.Connection] = None
    onet_version: Optional[str] = None

    oesm_df: Optional[pd.DataFrame] = None
    oesm_version: Optional[str] = None

    epoch_df: Optional[pd.DataFrame] = None
    epoch_version: Optional[str] = None

    # === ENRICHMENTS ===
    task_estimateability_df: Optional[pd.DataFrame] = None
    task_estimates_df: Optional[pd.DataFrame] = None

    meta: Metadata = Field(default_factory=Metadata)