python使用sshtunnel+pymssql连接远程数据库详解

1047521767

为了安全起见，很多公司服务器数据库的访问多半是要做限制的，由专门的DBA管理，而且都是做的集群，数据库只能内网访问，所以就有一个直接的问题是，往往多数时候，在别的机器上（比如自己本地），是不能访问数据库的，给日常开发调试造成了很大不便。最近几天在做Python相关项目，有个需求 ，是希望在任何机器上都可以ssh到某台在数据库白名单的机器上，然后访问数据库，不然的话就要去服务器安装Python环境，运行程序，比较麻烦，翻阅多篇博客文章，决定自己去实现。
涉及库：pymssql、sshtunnel
1 Y5 e$ ?; V8 a* W  Q/ L+ j6 A涉及数据库：SQLSERVER
# o1 I2 z! z# C  j$ J: v4 v. l
! L& K3 B6 ?; N场景如下：
跳板机核心代码   def __get_ssh_connector(self):        # 远程连接
0 L! J- N5 O' M$ |; |) s" T9 S# R        # 跳板机地址 端口，服务器账号，密码配置
& X. e4 h  ~5 S8 `9 p( }        server = SSHTunnelForwarder(
% g" |$ b4 a* p9 u) Z0 W            (
                self.connect_config.get("ssh_host", ‘白名单服务器地址’),
                self.connect_config.get("ssh_port", 22)
            ),
- G& F5 Y" y3 t# @            ssh_username=self.connect_config.get("ssh_username", ‘白名单服务器账号 ’),
            ssh_password=self.connect_config.get("ssh_password", ‘白名单服务器密码 ’),
4 q4 x# r* P% |( b; N3 m            # 内网数据库地址和端口
            remote_bind_address=(self.connect_config.get("dbserver", '目标数据库地址'), self.connect_config.get("ssh_mssql_port", ‘数据库端口’))
8 ?8 I& ]" }9 H( K, J$ F9 }: ^        )
7 ?1 K5 o! e# `8 r3 t, f        server.start()
        return server  # 远程主机上的mssql通过ssh连接映射到本地的端口
pymssql 连接数据库核心代码
2 r) J# Q+ {( |5 O) a/ M! b def get_mssql_connector(self):
) F7 e* W* \0 ?- l* q- M2 L0 D       return get_mssql_connector({
7 j# P. e8 K. b, ]) `, h           "port": self.server.local_bind_port, #非常重要
5 f- E) P8 `* y# A/ i2 A           "username": ‘目标数据库账号’,
           "password": ‘目标数据库密码’,
           "db": self.connect_config.get('db', ‘目标数据库’)
2 d. g2 W: z: Q$ k6 E       })
! \9 m/ t+ }; ^' L
def __get_mssql_connector(database_config):
    host = database_config.get('host', '127.0.0.1')
' R/ ?) _, k( p    port = database_config.get('port', ‘数据库端口’)
    user = database_config.get('username', ‘目标数据库账号’)
0 @. ?- F* k3 p7 h, x2 h    passwd = database_config.get('password', ‘目标数据库密码’)
    db = database_config.get('db', ‘目标数据库’)
    return  pymssql.connect(host=host, port=port, user=user, password=passwd, database=db, charset="UTF-8")

( h- b2 z3 S6 a. n3 p' h1 RMySql 和SQLSERVER方法一样，不一样的地方pymssql.connect改成pymysql.connect
$ d- [" P3 x/ {
' H7 e) [, h0 |4 f7 y