From 03797f4c2774e7fa6f2a1f168b3abdd1db3870b9 Mon Sep 17 00:00:00 2001 From: Andrew nuark G Date: Sun, 12 Mar 2023 10:36:23 +0700 Subject: [PATCH] Better quantificators in column def. --- src/based/columns.py | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/src/based/columns.py b/src/based/columns.py index 13a5267..da35c87 100644 --- a/src/based/columns.py +++ b/src/based/columns.py @@ -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):