атрибуты данных python в модели sqlalchemy - PullRequest
1 голос
/ 06 августа 2010

У меня возникают проблемы с атрибутами класса Python по сравнению с атрибутами данных в моей модели sqlalchemy. Вот небольшой пример, чтобы продемонстрировать, что происходит:

# -*- coding: utf-8 -*-
import cherrypy
import sqlalchemy
from sqlalchemy import create_engine
from sqlalchemy import Table, Column, Integer, String, MetaData, ForeignKey
from sqlalchemy.orm import sessionmaker
from sqlalchemy.ext.declarative import declarative_base

engine = create_engine('mysql://USER:PASS@localhost/SCHEMA?charset=utf8', echo=True)
Session = sessionmaker(bind=engine)
session = Session()
Base = declarative_base()

class User(Base):
    __tablename__ = 'user'

    id = Column(Integer, primary_key=True)
    username = Column(String, nullable=False)
    name = Column(String, nullable=False)
    email = Column(String, nullable=False)

    def __init__(self, username, name, email):
        self.username = username
        self.name = name
        self.email = email
        self.data_attribute = "my data attribute"
        print "__init__"

    def __repr__(self):
        print "__repr__"
        return "<User('%s','%s', '%s')>" % (self.username, self.name, self.email)

class Root(object):

    @cherrypy.expose
    def test(self):        
        our_user = session.query(User).one()        
        return our_user.data_attribute #error

if __name__ == '__main__':

    cherrypy.tree.mount(Root())
    cherrypy.server.quickstart()
    cherrypy.engine.start()

Этот код ошибки, потому что когда объекты взяты из БД, __init__ не запускается, поэтому data_attribute не существует. Если я добавлю data_attribute к объектам Column, это станет атрибутом класса (я думаю) и, следовательно, будет иметь одинаковое значение для всех экземпляров User. Это не то, что я хочу.

Что мне нужно сделать, чтобы у моего объекта User был атрибут данных, который не приходит из БД, но также уникален для каждого экземпляра User?

Редактировать: Я, вероятно, должен указать, что data_attribute не всегда будет простой строкой, установленной в __init__ - это просто для демонстрации проблемы.

Ответы [ 2 ]

2 голосов
/ 08 августа 2010

SQLAlchemy предоставляет специальный синтаксис для таких случаев, http://docs.sqlalchemy.org/en/latest/orm/constructors.html?highlight=object%20initialization

1 голос
/ 06 августа 2010

Вы должны использовать super () для вызова конструктора базового класса:

def __init__(self, username, name, email, *args, **kwargs ):
    super( User, self ).__init__( *args, **kwargs ) # Python 2.x or 3.x
    #super().__init__( *args, **kwargs ) # Python 3.x only
    self.username = username
    ...
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...