src/data/schema.py

"""
Class for table schema definition
"""
from dataclasses import dataclass, field
from typing import List, Dict, Optional


@dataclass
class Column:
    name: str
    type: str
    description: Optional[str] = None


@dataclass
class TableSchema:
    table_name: str
    columns: List[Column] = field(default_factory=list)