use abc instead raising error in unqiueIndex class

tortoise · Jun 8, 2024 · 6c02d74 · 6c02d74
1 parent 9c8e553
commit 6c02d74
Showing 1 changed file with 4 additions and 6 deletions.
diff --git a/tortoise/indexes.py b/tortoise/indexes.py
@@ -1,3 +1,4 @@
+import abc
 from typing import TYPE_CHECKING, Optional, Tuple, Type, Generic, TypeVar
 
 
@@ -80,7 +81,7 @@ def __init__(
 T = TypeVar("T")
 
 
-class UniqueIndexABC(Index, Generic[T]):
+class UniqueIndexABC(Index, abc.ABC, Generic[T]):
     """
     T type is used for nulls type arg beyond extends
     """
@@ -98,8 +99,5 @@ def __init__(
         if nulls is not None:
             self.extra += f" {self.nulls(nulls)}"
 
-    def nulls(self, distinct: T):
-        raise NullsNotHandledInUniqueIndex
-
-
-class NullsNotHandledInUniqueIndex(NotImplementedError): ...
+    @abc.abstractmethod
+    def nulls(cls, distinct: T): ...