Class DbtCodeGenerator

sdk/python/feast/dbt/codegen.py:178–379

Generates Python code for Feast objects from dbt models. This class creates complete, importable Python files containing Entity, DataSource, and FeatureView definitions. Example:: generator = DbtCodeGenerator( data_source_type="bigquery", timestamp

class DbtCodeGenerator:
    """
    Generates Python code for Feast objects from dbt models.

    This class creates complete, importable Python files containing
    Entity, DataSource, and FeatureView definitions.

    Example::

        generator = DbtCodeGenerator(
            data_source_type="bigquery",
            timestamp_field="event_timestamp",
            ttl_days=7
        )
        code = generator.generate(
            models=models,
            entity_column="user_id",
            manifest_path="target/manifest.json",
            project_name="my_project"
        )
        with open("features.py", "w") as f:
            f.write(code)
    """

    def __init__(
        self,
        data_source_type: str = "bigquery",
        timestamp_field: str = "event_timestamp",
        ttl_days: int = 1,
    ):
        self.data_source_type = data_source_type.lower()
        self.timestamp_field = timestamp_field
        self.ttl_days = ttl_days

        # Set up Jinja2 environment
        self.env = Environment(
            loader=BaseLoader(),
            trim_blocks=True,
            lstrip_blocks=True,
        )
        self.template = self.env.from_string(FEAST_FILE_TEMPLATE)

    def generate(
        self,
        models: List[DbtModel],
        entity_columns: Union[str, List[str]],
        manifest_path: str = "",
        project_name: str = "",
        exclude_columns: Optional[List[str]] = None,
        online: bool = True,
    ) -> str:
        """
        Generate Python code for Feast objects from dbt models.

        Args:
            models: List of DbtModel objects to generate code for
            entity_columns: Entity column name(s) - single string or list of strings
            manifest_path: Path to the dbt manifest (for documentation)