python ssh链接Mysql数据库

from sshtunnel import SSHTunnelForwarder
import time, datetime, pymysql, json, requests

def ssh_mysql(sql, method):
# SSH 信息
ssh_ip = ‘’
ssh_port = # type:int
ssh_username = ‘root’
ssh_password = ‘’
# DB信息
db_user = ‘root’
db_password = ‘’
db_host = ‘127.0.0.1’
database = ‘bigadmin’
with SSHTunnelForwarder(
(ssh_ip, ssh_port),
ssh_username=ssh_username,
ssh_password=ssh_password,
remote_bind_address=(‘127.0.0.1’, 3306)) as server:
db = pymysql.connect(host=db_host,
port=server.local_bind_port,
user=db_user,
passwd=db_password,
database=database,
charset=‘utf8’)
cursor = db.cursor()
if method == ‘get’:
cursor.execute(sql)
data = cursor.fetchall()
db.close()
return data
if method == ‘add’:
cursor.execute(sql)
insert_id = cursor.lastrowid
db.commit()
db.close()
return ‘添加的id是:{}’.format(insert_id)
if method == ‘edit’:
cursor.execute(sql)
edit_id = cursor.lastrowid
db.commit()
db.close()
return ‘更新的id是:{}’.format(edit_id)
if method == ‘del’:
cursor.execute(sql)
del_id = cursor.lastrowid
db.commit()
db.close()
return ‘删除的id是{]’.format(del_id)