SQLAlchemy - отображение одного класса в две таблицы

Question

У меня есть две реализации очереди базы данных (они используют разные таблицы) и хочу, чтобы они использовали объекты одного и того же класса. Итак, они оба выглядят очень похожими:

class AbstractDBQueue(object):
    def __init__(self, tablename):
        self.tablename = tablename
        self.metadata = MetaData()
        self.engine = create_engine('mysql+mysqldb://%s:%s@%s:%d/%s' % (
            settings.DATABASE.get('USER'),
            settings.DATABASE.get('PASSWORD'),
            settings.DATABASE.get('HOST') or '127.0.0.1',
            settings.DATABASE.get('PORT') or 3306,
            settings.DATABASE.get('NAME')
        ), encoding='cp1251', echo=True, pool_recycle=7200)
        self.metadata.bind = self.engine
        self.session = sessionmaker(bind=self.engine)()

    def setup_table(self, table, entity_name):
        self.table = table
        newcls = type(entity_name, (SMSMessage, ), {})
        mapper(newcls, table)
        return newcls

    def put(self, message=None, many_messages=[]):
        if message:
            self.session.add(message)
        else:
            for m in many_messages:
                self.session.add(m)
        self.session.commit()

    def get(self, limit=None):
        if limit:
            q = self.session.query(self.SMSClass).limit(limit)
        else:
            q = self.session.query(self.SMSClass)
        smslist = []
        for sms in q:
            smslist.append(sms)
        self.session.expunge_all()
        return smslist

class DBQueue(AbstractDBQueue):
    """
    MySQL database driver with queue interface
    """
    def __init__(self):
        self.tablename = settings.DATABASE.get('QUEUE_TABLE')
        super(DBQueue, self).__init__(self.tablename)
        self.logger = logging.getLogger('DBQueue')
        self.SMSClass = self.setup_table(Table(self.tablename, self.metadata, autoload=True), "SMSQueue")

class DBWorkerQueue(AbstractDBQueue):
    """
    MySQL database driver with queue interface for separate workers queue
    """

    def __init__(self):
        self.tablename = settings.DATABASE.get('WORKER_TABLE')
        super(DBWorkerQueue, self).__init__(self.tablename)
        self.logger = logging.getLogger('DBQueue')
        self.SMSClass = self.setup_table(Table(self.tablename, self.metadata, autoload=True), "SMSWorkerQueue")

    def _install(self):
        self.metadata.create_all(self.engine)

SMSMessage - это имя класса, который я хочу использовать. Функция map_class_to_table () - это хак, который я нашел в документации по SQLAlchemy: http://www.sqlalchemy.org/trac/wiki/UsageRecipes/EntityName

Но, похоже, это не помогает - когда первый экземпляр очереди отображает SMSMessage на свою таблицу, тогда все объекты, которые я передаю в put () второй очереди, неявно преобразуются в сопоставленный класс первой очереди, а вторая база данных все еще пуста после session.commit () .

Мне нужно использовать обе очереди одновременно, может быть, даже используя потоки (я думаю, подключение к пулу будет полезно), но я просто не могу заставить это работать. Не могли бы вы помочь, пожалуйста?

Answer 1

Я думаю, что ваша проблема связана с переменной tablename.Это переменная класса , которая определяется при создании класса, а не изменяется .Таким образом, он будет одинаковым для обоих ваших экземпляров, когда они получат к нему доступ с помощью self.tablename.Чтобы это исправить, переместите его внутрь функции init и сделайте его self.tablename.Это будет инициализировать его каждый раз, когда вы создаете новый объект.