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)),