from psycopg.sql import SQL, Identifier, Literal, Composed
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}:asset"


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}:asset"