Better quantificators in column def.

This commit is contained in:
Andrew 2023-03-12 10:36:23 +07:00
parent 9e1ca9dd43
commit 03797f4c27

View file

@ -13,7 +13,6 @@ class ColumnDefinition:
raise NotImplementedError
# TODO: Not so good implementation (maybe)
class UniqueColumnDefinition(ColumnDefinition):
def __init__(self, wrapped: ColumnDefinition):
super().__init__(wrapped.name)
@ -23,7 +22,7 @@ class UniqueColumnDefinition(ColumnDefinition):
return SQL("{} UNIQUE").format(self.wrapped.sql())
def serialize(self):
return self.wrapped.serialize()
return f"{self.wrapped.serialize()}:unique"
def make_column_unique(column) -> UniqueColumnDefinition:
@ -38,7 +37,7 @@ class PrimarySerialColumnDefinition(ColumnDefinition):
return SQL("{} SERIAL PRIMARY KEY").format(Identifier(self.name))
def serialize(self):
return f"{self.name}:int"
return f"{self.name}:serial:primary"
class PrimaryUUIDColumnDefinition(ColumnDefinition):
@ -51,7 +50,7 @@ class PrimaryUUIDColumnDefinition(ColumnDefinition):
)
def serialize(self):
return f"{self.name}:uuid"
return f"{self.name}:uuid:primary"
class TextColumnDefinition(ColumnDefinition):