26 lines
663 B
Python
26 lines
663 B
Python
from psycopg.sql import SQL, Identifier
|
|
from based.columns import (
|
|
IntegerColumnDefinition,
|
|
)
|
|
|
|
|
|
class UserRefColumnDefinition(IntegerColumnDefinition):
|
|
def __init__(self, name: str):
|
|
super().__init__(name)
|
|
|
|
def sql(self):
|
|
return SQL("{} INTEGER NOT NULL").format(Identifier(self.name))
|
|
|
|
def serialize(self):
|
|
return f"{self.name}:int-user"
|
|
|
|
|
|
class AssetRefColumnDefinition(IntegerColumnDefinition):
|
|
def __init__(self, name: str):
|
|
super().__init__(name)
|
|
|
|
def sql(self):
|
|
return SQL("{} INTEGER NOT NULL").format(Identifier(self.name))
|
|
|
|
def serialize(self):
|
|
return f"{self.name}:int-asset"
|