Use PyMySQL for MySQL commands

Author: mdmintz

seleniumbase/core/mysql.py

@@ -15,15 +15,15 @@ def __init__(self, database_env='test', conf_creds=None):
         """
         Gets database information from mysql_conf.py and creates a connection.
         """
-        import MySQLdb
+        import pymysql
         db_server, db_user, db_pass, db_schema = \
             conf.APP_CREDS[conf.Apps.TESTCASE_REPOSITORY][database_env]
         retry_count = 3
         backoff = 1.2  # Time to wait (in seconds) between retries.
         count = 0
         while count < retry_count:
             try:
-                self.conn = MySQLdb.connect(host=db_server,
+                self.conn = pymysql.connect(host=db_server,
                                             user=db_user,
                                             passwd=db_pass,
                                             db=db_schema)