def get_data_source()

in jetstream/config.py [0:0]

• 16 lines of code
• 3 McCabe index (conditional complexity)

    def get_data_source(self, data_source_slug: str, app_name: str) -> DataSource | None:
        """Return the data source matching the specified slug."""
        data_source_definition = self.configs.get_data_source_definition(data_source_slug, app_name)
        if data_source_definition is None:
            raise Exception(f"Could not find definition for data source {data_source_slug}")

        return DataSource(
            name=data_source_definition.name,
            from_expression=data_source_definition.from_expression,
            client_id_column=data_source_definition.client_id_column,
            submission_date_column=data_source_definition.submission_date_column,
            experiments_column_type=(
                None
                if data_source_definition.experiments_column_type == "none"
                else data_source_definition.experiments_column_type
            ),
            default_dataset=data_source_definition.default_dataset,
        )