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

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

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

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

b7ff9b5dc6520729968d9ce081705825.png

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

原代码是下边这样:

48304ba5e6f9fe08f3fa1abda7d326ab.png

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!')

48304ba5e6f9fe08f3fa1abda7d326ab.png

连接mysql数据库

代码如下:

48304ba5e6f9fe08f3fa1abda7d326ab.png

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')

48304ba5e6f9fe08f3fa1abda7d326ab.png

然后接下来的查询什么的,直接写在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语句。

48304ba5e6f9fe08f3fa1abda7d326ab.png

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')

48304ba5e6f9fe08f3fa1abda7d326ab.png

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

连接sqlserver数据库

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

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值