Fix #149: PostgresRangePartitioningStrategy should derive from Postgr…

…esPartitioningStrategy
SectorLabs · Nov 4, 2021 · 98b9e32 · 98b9e32
1 parent 1f7701b
commit 98b9e32
Showing 1 changed file with 2 additions and 16 deletions.
diff --git a/psqlextra/partitioning/range_strategy.py b/psqlextra/partitioning/range_strategy.py
@@ -1,24 +1,10 @@
-from abc import abstractmethod
-from typing import Generator
 
-from .range_partition import PostgresRangePartition
+from .strategy import PostgresPartitioningStrategy
 
 
-class PostgresRangePartitioningStrategy:
+class PostgresRangePartitioningStrategy(PostgresPartitioningStrategy):
     """Base class for implementing a partitioning strategy for a range
     partitioned table."""
 
-    @abstractmethod
-    def to_create(
-        self,
-    ) -> Generator[PostgresRangePartition, None, None]:
-        """Generates a list of partitions to be created."""
-
-    @abstractmethod
-    def to_delete(
-        self,
-    ) -> Generator[PostgresRangePartition, None, None]:
-        """Generates a list of partitions to be deleted."""
-
 
 __all__ = ["PostgresRangePartitioningStrategy"]