Tuuli/based
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Better quantificators in column def.

Browse source
This commit is contained in:
Andrew 2023-03-12 10:36:23 +07:00
parent 9e1ca9dd43
commit 03797f4c27
1 changed files with 3 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

7
src/based/columns.py
View file

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