码迷,mamicode.com
首页 > 数据库 > 详细

Python用MySQLdb, pymssql 模块通过sshtunnel连接远程数据库

时间:2018-04-26 16:54:03      阅读:218      评论:0      收藏:0      [点我收藏+]

标签:name   场景   adt   local   client   cli   mysqld   ssh   main   

转载自 https://www.cnblogs.com/luyingfeng/p/6386093.html

安全起见,数据库的访问多半是要做限制的,所以就有一个直接的问题是,往往多数时候,在别的机器上(比如自己本地),是不能访问数据库的,给日常使用造成了很大不便。所以前几天做了个需求,是希望在任何机器上都可以ssh到某台在数据库白名单的机器上,然后访问数据库。

经人推荐,查询了一个工具叫sshtunnel ,需要在你想要登录数据库的机器上安装。

sshtunnel的基本介绍 : http://sshtunnel.readthedocs.io/en/latest/?badge=latest

这里边的图2,正好是我们描述的场景,数据库在远程某台服务器上,只能同在一起的另一台服务器才能访问,但是我们本地就需要ssh到这台服务器上

技术分享图片

 

但是这个里边涉及到的代码是服务器相关,远程连接的是服务器,我们需要的是远程服务器的数据库,这个也一样,只需要变更代码中后半部分就可以了。

原代码是下边这样:

技术分享图片
 1 import paramiko
 2 from sshtunnel import SSHTunnelForwarder
 3 
 4 with SSHTunnelForwarder(
 5     (REMOTE_SERVER_IP, 443),
 6     ssh_username="",
 7     ssh_pkey="/var/ssh/rsa_key",
 8     ssh_private_key_password="secret",
 9     remote_bind_address=(PRIVATE_SERVER_IP, 22),
10     local_bind_address=(‘0.0.0.0‘, 10022)
11 ) as tunnel:
12     client = paramiko.SSHClient()
13     client.load_system_host_keys()
14     client.set_missing_host_key_policy(paramko.AutoAddPolicy())
15     client.connect(‘127.0.0.1‘, 10022)
16     # do some operations with client session
17     client.close()
18 
19 print(‘FINISH!‘)
技术分享图片

连接mysql数据库

连接mysql数据库的时候,看到网上有一位程序员已经实现了: python 使用mysqldb模块通过ssh隧道连接mysql

代码如下:

技术分享图片
 1 import MySQLdb
 2 from sshtunnel import SSHTunnelForwarder
 3 
 4 with SSHTunnelForwarder(
 5          (‘sshhost.domain.com‘, 22),    #B机器的配置
 6          ssh_password="sshpasswd",
 7          ssh_username="sshusername",
 8          remote_bind_address=(‘mysqlhost.domain.com‘, mysql.port)) as server:  #A机器的配置
 9 
10     conn = MySQLdb.connect(host=‘127.0.0.1‘,              #此处必须是是127.0.0.1
11                            port=server.local_bind_port,
12                            user=‘user‘,
13                            passwd=‘password‘,
14                            db=‘dbname‘)
技术分享图片

 然后接下来的查询什么的,直接写在with那里边,与conn对齐就可以了。

对于我这边来说有一个问题是,因为我们对于数据库连接这一部分,往往是在一个单独的函数里,与其他数据库的查询插入删除更新操作往往不在一起,这样的话,with as 有个特点就是,离开这块作用域,对象就被销毁掉了,别的函数里是没法用的,也就会出现一种情况是,连接上了,但是对象又给销毁掉了,结果查询的时候直接显示这个错误:OperationalError: (2006, ‘MySQL server has gone away‘), 而网上查询这个错误,多半说的是因为你查询的 sql操作的时间过长,或者是传送的数据太大 ,但是我这个地方实际上就是因为出了with as 的作用域,导致连接又给关闭掉了,所以出现这样的结果。

关于 with as ,有篇文章写得很详尽。理解Python中的with…as…语法

所以我把上边那个ssh代码改掉了,像是sshtunnel文档里边图一所对应的代码,一样,将SSHTunnelForwarder出来的对象赋值给server,然后启动server,然后进行一系列操作之后,再stop掉。

本来数据库连接我们写成了一个单独的函数,改了之后,直接也还放在这个函数里就好了,替代原来的connect语句。

技术分享图片
 1 def connect(self):
 2     ‘‘‘
 3     self.client = MySQLdb.connect(host=self.server, port=self.port, user=self.user,
 4                                   passwd=self.password, db=self.database,
 5                                   charset=self.charset)
 6     # log.info(‘Connect to MySQL Server: ‘ + self.server)
 7     ‘‘‘
 8 
 9     server = SSHTunnelForwarder(
10             (‘sshhost.domain.com‘, 22),  # B机器的配置
11             ssh_password=‘ssh_password‘,
12             ssh_username=‘ssh_username‘,
13             remote_bind_address=(‘mysqlhost.domain.com‘, mysql.port)
14     )
15     server.start()
16 
17     self.client = MySQLdb.connect(host=‘127.0.0.1‘,  # 此处必须是是127.0.0.1
18                                   port=server.local_bind_port,
19                                   user=‘username‘,
20                                   passwd=‘password‘,
21                                   db=‘dbname‘)
技术分享图片

然后在进行查询更新删除等操作的时候,先连接一下数据库就好了,用self.client.

连接sqlserver数据库

跟mysql的一致,但是db那里要注意,SQLServer的是database, 然后是pymssql.connect就可以了,但是这个地方还要说我踩过的一个坑,我写完sqlserver之后怎么连接都连不上数据库,后来才发现是版本的问题,我把本地SQLServer更新了之后就可以了。感觉版本是个大坑

Python用MySQLdb, pymssql 模块通过sshtunnel连接远程数据库

标签:name   场景   adt   local   client   cli   mysqld   ssh   main   

原文地址:https://www.cnblogs.com/1204guo/p/8952901.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!