封装带SSH跳板机的MYSQL

一、封装带SSH跳板机的MYSQL

二、配置settting

 import pymysql
 from sshtunnel import SSHTunnelForwarder

 class MyDb(object):
     #mysql类方法
     def __del__(self):
         self.cur.close()
         self.coon.close()
         self.server.close()

     def __init__(self,host,ssh_username,ssh_password,ip,db,sql_username,password):
         try:
             server = SSHTunnelForwarder(
                 ssh_address_or_host=(host, 22),
                 ssh_username=ssh_username,
                 ssh_password=ssh_password,
                 remote_bind_address=(ip, 3306)
             )
             self.server = server
             self.server.daemon_forward_servers = True #不加server关闭不了
             self.server.start()
             self.connect = pymysql.connect(
                 host='127.0.0.1',user=sql_username,passwd=password,port=self.server.local_bind_port,charset='utf8',db=db,
                 autocommit=True,
             )

         except Exception as e:
             print('mysql连接失败，错误信息%s'%e)

         else:
             self.cursor = self.connect.cursor(cursor=pymysql.cursors.DictCursor)

     def ex_sql(self,sql,many=True):
         try:
             self.cursor.execute(sql)
         except Exception as e:
             print('sql语句有问题，%s'%sql)
         else:
             if many:
                 result = self.cursor.fetchall()
             else:
                 result = self.cursor.fetchone()
             return result

 #哪里用哪里实例化吧
 # my_sql = MyDb(**setting.MYSQL_INFO)
 #直接在这里实例化的话，用的时候，直接导入就ok了

ssh_mysql

 import os
 import nnlog
 BASE_PATH = os.path.dirname(
     os.path.dirname(os.path.abspath(__file__))
 )

 # MAIL_HOST='smtp.qq.com'
 MAIL_HOST='smtp.exmail.qq.com'
 MAIL_USER='xxxx.com'
 MAIL_PASSWRD='xxxxx'

 TO = ['xxxx.com'
 ]

 LEVEL = 'debug' #日志级别

 LOG_PATH = os.path.join(BASE_PATH,'logs')
 CASE_PATH = os.path.join(BASE_PATH,'cases')
 YAML_PATH = os.path.join(BASE_PATH,'case_data')
 CASE_TEMPLATE = os.path.join(BASE_PATH,'conf','base.txt') #用例模板的路径
 REPORT_PATH = os.path.join(BASE_PATH,'report')
 DATA_PATH = os.path.join(BASE_PATH,'data') #存放测试用到数据的目录
 BASE_URL = 'https://xxx.xxx.xxx.xx' #线上环境
 #BASE_URL = 'http://xxx.xxx.xxx.xx' #开发环境

 LOG_NAME='xxx.log' #日志的文件名

 log  = nnlog.Logger(os.path.join(LOG_PATH,LOG_NAME),'debug') #定义好日志

 #dev环境redis配置文件
 REDIS_dev = {
     'host': 'xxx.xxx.xxx.xxx',
     'ip' : 'xxx.xxx.xxx.xxx',
     'ssh_username' : 'xxx',
     'ssh_password': 'xxx',
     'db':2,
     'password' : 'xxx'
 }

 #线上环境redis配置文件
 REDIS_online = {
     'host': 'xxx.xxx.xxx.xxx',
     'ip' : 'xx.xxx.xxx.xxx',
     'ssh_username' : 'xxx',
     'ssh_password': 'xxxx',
     'db':2,
     'password' : 'xxx'
 }

 #dev环境mysql配置文件
 MYSQL_dev  = {
     'host': 'xxx.xxx.xxx.xxx',
     'ip' : 'xx.xx.xx.xx',
     'ssh_username' : 'xxx',
     'ssh_password': 'xxx',
     'db':'xxx',
     'sql_username':'xx',
     'password' : 'xxx'
     }

 #线上环境mysql配置文件
 MYSQL_online  = {
     'host': 'xxx.xxx.xxx.xxx',
     'ip' : '172.xx.xx.xx',
     'ssh_username' : 'xxx',
     'ssh_password': 'xxxx',
     'db':'xxx',
     'sql_username':'xxx',
     'password' : 'xxx'
     }

setting设置