Wie kann ich eine benutzerdefinierte PostgreSQL-Domäne in SQLAlchemy darstellen?

Question

Ich fange an, Alembic in mein Projekt zu integrieren, das bereits SQLAlchemy Tabellendefinitionen verwendet. Derzeit wird mein DB-Schema außerhalb meiner Anwendung verwaltet, und ich möchte das gesamte Schema in meine Tabellendefinitionsdatei bringen.

In PostgreSQL verwende ich eine benutzerdefinierte Domain zum Speichern von E-Mail-Adressen. Die PostgreSQL DDL ist:

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

Wie vertrete ich die Schaffung dieser Domäne und die Verwendung davon als Spaltendatentyp, in SQLAlchemy?

Answer 1

Dies ist wahrscheinlich weit von einer funktionierenden Lösung, aber ich denke, der beste Weg, dies zu tun wäre Subklasse 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 

Offensichtlich ist dies unvollständig, aber es sollte Ihnen einen guten Ausgangspunkt geben, wenn Sie dies stark genug wollen. :)