SQLAlchemy: avoiding repetition in declarative style class definition

后端 未结 3 622
谎友^
谎友^ 2021-02-05 21:48

I\'m using SQLAlchemy, and many classes in my object model have the same two attributes: id and (integer & primary key), and name (a string). I\'m trying to avoid declaring

3条回答
  •  悲哀的现实
    2021-02-05 22:53

    You could factor out your common attributes into a mixin class, and multiply inherit it alongside 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: You might think this would result in C1 and C2 sharing the same Column objects, but as noted in the SQLAlchemy docs, Column objects are copied when originating from a mixin class. I've updated the code sample to demonstrate this behavior.

提交回复
热议问题