Jak reprezentować niestandardową domenę PostgreSQL w SQLAlchemy?

Question

Zaczynam wprowadzać Alembic do mojego projektu, który już używa definicji tabel SQLAlchemy. Obecnie mój schemat bazy danych jest zarządzany na zewnątrz mojej aplikacji i chcę wprowadzić cały schemat do pliku definicji tabel.

W PostgreSQL używam domeny niestandardowej do przechowywania adresów e-mail. PostgreSQL DDL jest:

CREATE DOMAIN email_address TEXT CHECK (value ~ '.+@.+') 

Jak mogę reprezentować tworzenia tej domenie i użycie go jako typ danych kolumny w SQLAlchemy?

Answer 1

Prawdopodobnie daleki od sprawnego rozwiązania, ale myślę, że najlepszym sposobem na zrobienie tego byłaby podklasa sqlalchemy.schema._CreateDropBase.

from sqlalchemy.schema import _CreateDropBase 

class CreateDomain(_CreateDropBase): 
    '''Represent a CREATE DOMAIN statement.''' 

    __visit_name__ = 'create_domain' 

    def __init__(self, element, bind=None, **kw): 
     super(CreateDomain, self).__init__(element, bind=bind, **kw) 

class DropDomain(_CreateDropBase): 
    '''Represent a DROP BASE statement.''' 

    __visit_name__ = 'drop_domain' 

    def __init__(self, element, bind=None, **kw): 
     super(DropDomain, self).__init__(element, bind=bind, **kw) 

@compiles(CreateDomain, 'postgresql') 
def visit_create_domain(element, compiler, **kw): 
    text = '\nCREATE DOMAIN %s AS %s' % (
     compiler.prepare.format_column(element.name), 
     compiler.preparer.format_column(element.type_)) # doesn't account for arrays and such I don't think 

    default = compiler.get_column_default_string(column) 
    if default is not None: 
     text += " DEFAULT %s" % default 

    return text 

Oczywiście, jest to niekompletne, ale powinno dać dobry punkt wyjścia, jeśli chcesz tego wystarczająco mocno. :)