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映射到它的表时，然后我传递给第二个队列的所有对象都被隐式转换为第一个队列的映射类，而第二个数据库在session.commit（）后仍然为空。

我需要同时使用两个队列，甚至可能使用线程（我认为，池连接将很有用），但我就是不能让它工作。你能帮忙吗？

- Enchantner

---

如果您先使用第二个队列，会不会颠倒顺序呢？ - Spencer Rathbun

我刚刚使用继承重写了代码，现在它只会引发一个错误：“类'sms.message.SMSMessage'未映射”。我应该将输入对象强制转换为映射类吗？这似乎很奇怪。 - Enchantner

Answer 1

我认为你的问题与tablename变量有关。它是一个class variable，在创建类时定义，然后不会改变。因此，当两个实例使用self.tablename访问它时，它们将是相同的。要修复这个问题，将它移动到init函数内，并将其设为self.tablename。这样每次创建新对象时都会初始化它。

- Spencer Rathbun

---

我已经更新了主题中的代码 - 重写了使用继承来传递表名作为构造函数参数的队列类。但它仍然不起作用。 - Enchantner