Skip to content

Commit

Permalink
prepare empty classes for change comparisons
Browse files Browse the repository at this point in the history
  • Loading branch information
sdebruyn committed Apr 1, 2024
1 parent a698846 commit b00dd94
Show file tree
Hide file tree
Showing 5 changed files with 54 additions and 1 deletion.
2 changes: 2 additions & 0 deletions dbt/adapters/timescaledb/timescaledb_adapter.py
Original file line number Diff line number Diff line change
Expand Up @@ -7,10 +7,12 @@
TimescaleDBConnectionManager,
)
from dbt.adapters.timescaledb.timescaledb_index_config import TimescaleDBIndexConfig
from dbt.adapters.timescaledb.timescaledb_relation import TimescaleDBRelation


class TimescaleDBAdapter(PostgresAdapter):
ConnectionManager = TimescaleDBConnectionManager
Relation = TimescaleDBRelation

@available
def parse_index(self, raw_index: Any) -> Optional[TimescaleDBIndexConfig]:
Expand Down
Original file line number Diff line number Diff line change
@@ -0,0 +1,10 @@
from dataclasses import dataclass

from dbt.adapters.postgres.relation_configs.materialized_view import (
PostgresMaterializedViewConfigChangeCollection,
)


@dataclass
class TimescaleDBContinuousAggregateConfigChangeCollection(PostgresMaterializedViewConfigChangeCollection):
pass
Original file line number Diff line number Diff line change
@@ -0,0 +1,10 @@
from dataclasses import dataclass

from dbt.adapters.postgres.relation_configs.materialized_view import (
PostgresMaterializedViewConfigChangeCollection,
)


@dataclass
class TimescaleDBHypertableConfigChangeCollection(PostgresMaterializedViewConfigChangeCollection):
pass
31 changes: 31 additions & 0 deletions dbt/adapters/timescaledb/timescaledb_relation.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,31 @@
from typing import Optional

from dbt.adapters.contracts.relation import RelationConfig
from dbt.adapters.postgres.relation import PostgresRelation
from dbt.adapters.relation_configs import (
RelationResults,
)
from dbt.adapters.timescaledb.timescaledb_continuous_aggregate_config_change_collection import (
TimescaleDBContinuousAggregateConfigChangeCollection,
)
from dbt.adapters.timescaledb.timescaledb_hypertable_config_change_collection import (
TimescaleDBHypertableConfigChangeCollection,
)


class TimescaleDBRelation(PostgresRelation):
pass

def get_hypertable_config_change_collection(
self, relation_results: RelationResults, relation_config: RelationConfig
) -> Optional[TimescaleDBHypertableConfigChangeCollection]:
return self.get_materialized_view_config_change_collection(
relation_results=relation_results, relation_config=relation_config
)

def get_continuous_aggregate_config_change_collection(
self, relation_results: RelationResults, relation_config: RelationConfig
) -> Optional[TimescaleDBContinuousAggregateConfigChangeCollection]:
return self.get_materialized_view_config_change_collection(
relation_results=relation_results, relation_config=relation_config
)
2 changes: 1 addition & 1 deletion pyproject.toml
Original file line number Diff line number Diff line change
Expand Up @@ -68,7 +68,7 @@ docs = [
]

[tool.ruff]
line-length = 110
line-length = 120
target-version = "py311"

[tool.ruff.lint]
Expand Down

0 comments on commit b00dd94

Please sign in to comment.