From 281f1c49a8f13349084107389128ad2714e8b089 Mon Sep 17 00:00:00 2001 From: Jonathan Golder Date: Thu, 9 Mar 2017 00:00:17 +0100 Subject: [PATCH] mysqlred: Set family via pywikibot Get family/language part of table names from PyWikiBot Site --- lib/mysqlred.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/lib/mysqlred.py b/lib/mysqlred.py index 4f6101e..1760fda 100644 --- a/lib/mysqlred.py +++ b/lib/mysqlred.py @@ -58,7 +58,7 @@ engine = create_engine(url, echo=True) Session = sessionmaker(bind=engine) session = Session() -family = "dewpbeta" +family = pywikibot.Site().family.dbName(pywikibot.Site().code) class Mysql(object): @@ -222,7 +222,7 @@ class MysqlRedFam( Mysql, Base ): heading = Column( Text, nullable=False ) redpageid = Column( - Integer, ForeignKey( "dewpbeta_redpages.pageid" ), nullable=False ) + Integer, ForeignKey( family + "_redpages.pageid" ), nullable=False ) beginning = Column( DateTime, nullable=False ) ending = Column( DateTime, nullable=True ) _status = Column( 'status', MutableSet.as_mutable(Status(255)),