SQLAlchemy: evitare la ripetizione nella definizione della classe di stile dichiarativa

Question

Sto utilizzando SQLAlchemy e molte classi nel mio modello a oggetti hanno gli stessi due attributi: id e (numero intero & chiave primaria) e nome (una stringa). Sto cercando di evitare dichiarandoli in ogni classe in questo modo:

class C1(declarative_base()): 
    id = Column(Integer, primary_key = True) 
    name = Column(String) 
    #... 

class C2(declarative_base()): 
    id = Column(Integer, primary_key = True) 
    name = Column(String) 
    #... 

Che cosa è un buon modo per farlo? Ho provato ad usare le metaclassi ma non ha funzionato ancora.

Answer 1

Si potrebbe scomporre i tuoi attributi comuni in una mixin class, e moltiplicare ereditare fianco declarative_base():

from sqlalchemy import Column, Integer, String 
from sqlalchemy.ext.declarative import declarative_base 

class IdNameMixin(object): 
    id = Column(Integer, primary_key=True) 
    name = Column(String) 

class C1(declarative_base(), IdNameMixin): 
    __tablename__ = 'C1' 

class C2(declarative_base(), IdNameMixin): 
    __tablename__ = 'C2' 

print C1.__dict__['id'] is C2.__dict__['id'] 
print C1.__dict__['name'] is C2.__dict__['name'] 

EDIT: Si potrebbe pensare che questo si tradurrebbe in C1 e C2 condividendo gli stessi Column oggetti, ma come indicato nella SQLAlchemy docs, oggetti colonna sono copiati quando proveniente da una classe intermedia. Ho aggiornato il codice di esempio per dimostrare questo comportamento.

Answer 2

Penso che ho avuto di lavorare.

ho creato un metaclasse che deriva da DeclarativeMeta, e fatto si che la metaclasse di C1 e C2. In quel nuovo metaclass, ho semplicemente detto

def __new__(mcs, name, base, attr): 
    attr['__tablename__'] = name.lower() 
    attr['id'] = Column(Integer, primary_key = True) 
    attr['name'] = Column(String) 
    return super().__new__(mcs, name, base, attr) 

E sembra funzionare bene.

Answer 3

Potrebbe anche utilizzare il metodo copia della colonna? In questo modo, i campi possono essere definiti in modo indipendente di tavoli, e quei campi che vengono riutilizzati sono solo field.copy() - ed.

id = Column(Integer, primary_key = True) 
name = Column(String) 

class C1(declarative_base()): 
    id = id.copy() 
    name = name.copy() 
    #... 

class C2(declarative_base()): 
    id = id.copy() 
    name = name.copy() 
    #...