From 77d1de44731b1aa51649a3e0a4a0550488dea63c Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?GOLDERWEB=20=E2=80=93=20Jonathan=20Golder?=
 <jonathan@golderweb.de>
Date: Wed, 24 Aug 2016 23:53:10 +0200
Subject: [PATCH] Add a tablename prefix depending on Site

To be able to run the bot on different wikis the db tables should be
named pywikibot.Site dependend and changed automatically

Related Task: [https://fs.golderweb.de/index.php?do=details&task_id=79 FS#79]
---
 lib/mysqlred.py | 39 +++++++++++++++++++++++++--------------
 1 file changed, 25 insertions(+), 14 deletions(-)

diff --git a/lib/mysqlred.py b/lib/mysqlred.py
index 77eae35..9eb7f4b 100644
--- a/lib/mysqlred.py
+++ b/lib/mysqlred.py
@@ -33,6 +33,7 @@ except ImportError:
 
 import atexit
 
+import pywikibot
 from pywikibot import config
 
 import jogobot
@@ -53,6 +54,7 @@ class MysqlRed:
     db_username = config.db_username
     db_password = config.db_password
     db_name = config.db_username + jogobot.config['db_suffix']
+    db_table_prefix = pywikibot.Site().family.dbName(pywikibot.Site().code)
 
     # Class variables for storing cached querys
     _cached_update_data = []
@@ -136,12 +138,14 @@ class MysqlRedPage( MysqlRed ):
 
     # Class variables for storing cached querys
     _cached_update_data = []
-    _update_query = 'UPDATE `red_pages` \
-SET `page_title` = ?, `rev_id` = ?, `status`= ? WHERE `page_id` = ?;'
+    _update_query = 'UPDATE `{pre}_red_pages` \
+SET `page_title` = ?, `rev_id` = ?, `status`= ? WHERE `page_id` = ?;'.format(
+        pre=MysqlRed.db_table_prefix)
 
     _cached_insert_data = {}
-    _insert_query = 'INSERT INTO `red_pages` \
-( page_id, page_title, rev_id, status ) VALUES ( ?, ?, ?, ? );'
+    _insert_query = 'INSERT INTO `{pre}_red_pages` \
+( page_id, page_title, rev_id, status ) VALUES ( ?, ?, ?, ? );'.format(
+        pre=MysqlRed.db_table_prefix)
 
     def __init__( self, page_id ):
         """
@@ -169,8 +173,10 @@ SET `page_title` = ?, `rev_id` = ?, `status`= ? WHERE `page_id` = ?;'
 
         cursor = type( self ).connection.cursor(mysqldb.DictCursor)
 
-        cursor.execute( 'SELECT * FROM `red_pages` WHERE `page_id` = ?;',
-                        ( self.__page_id, ) )
+        cursor.execute(
+            'SELECT * FROM `{pre}_red_pages` WHERE `page_id` = ?;'.format(
+                pre=MysqlRed.db_table_prefix), ( self.__page_id, ) )
+
         res = cursor.fetchone()
 
         if res:
@@ -221,15 +227,17 @@ class MysqlRedFam( MysqlRed ):
 
     # Class variables for storing cached querys
     _cached_update_data = []
-    _update_query = 'UPDATE `red_families` \
+    _update_query = 'UPDATE `{pre}_red_families` \
 SET `red_page_id` = ?, `heading` = ?, `beginning` = ?, `ending` = ?, \
-`status`= ? WHERE `fam_hash` = ?;'
+`status`= ? WHERE `fam_hash` = ?;'.format(
+        pre=MysqlRed.db_table_prefix)
 
     _cached_insert_data = {}
-    _insert_query = 'INSERT INTO `red_families` \
+    _insert_query = 'INSERT INTO `{pre}_red_families` \
 ( fam_hash, red_page_id, beginning, ending, status, heading, \
 article0, article1, article2, article3, article4, article5, article6, \
-article7 ) VALUES ( ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ? );'
+article7 ) VALUES ( ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ? );'.format(
+        pre=MysqlRed.db_table_prefix)
 
     def __init__( self ):
         """
@@ -252,8 +260,10 @@ article7 ) VALUES ( ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ? );'
 
         cursor = type( self ).connection.cursor( mysqldb.DictCursor )
 
-        cursor.execute( 'SELECT * FROM `red_families` WHERE `fam_hash` = ?;',
-                        ( fam_hash, ) )
+        cursor.execute(
+            'SELECT * FROM `{pre}_red_families` WHERE `fam_hash` = ?;'.format(
+                pre=MysqlRed.db_table_prefix), ( fam_hash, ) )
+
         self.data = cursor.fetchone()
 
     def add_fam( self, articlesList, heading, red_page_id,
@@ -301,8 +311,9 @@ article7 ) VALUES ( ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ? );'
 
         cursor = type( self ).connection.cursor( mysqldb.DictCursor )
 
-        cursor.execute( 'SELECT * FROM `red_families` WHERE `status` = ?;',
-                        ( status, ) )
+        cursor.execute(
+            'SELECT * FROM `{pre}_red_families` WHERE `status` = ?;'.format(
+                pre=type( self ).db_table_prefix), ( status, ) )
 
         while True:
             res = cursor.fetchmany( 1000 )