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

1047521767

为了安全起见，很多公司服务器数据库的访问多半是要做限制的，由专门的DBA管理，而且都是做的集群，数据库只能内网访问，所以就有一个直接的问题是，往往多数时候，在别的机器上（比如自己本地），是不能访问数据库的，给日常开发调试造成了很大不便。最近几天在做Python相关项目，有个需求 ，是希望在任何机器上都可以ssh到某台在数据库白名单的机器上，然后访问数据库，不然的话就要去服务器安装Python环境，运行程序，比较麻烦，翻阅多篇博客文章，决定自己去实现。
" z& v$ W# |: i$ B涉及库：pymssql、sshtunnel
涉及数据库：SQLSERVER
" J. L2 P- v* j5 T' t+ R& K$ ^
6 r8 E' d. c# g6 Q/ [/ m' S0 x场景如下：
. e  b8 L3 B4 j# U5 Z* i' P9 O$ ^0 ~跳板机核心代码   def __get_ssh_connector(self):        # 远程连接
% s4 @0 d  P* b% t0 G* u4 z! E3 m' G        # 跳板机地址 端口，服务器账号，密码配置
8 p9 j6 O8 n6 M1 _6 G5 N3 o, l: p        server = SSHTunnelForwarder(
            (
                self.connect_config.get("ssh_host", ‘白名单服务器地址’),
! S- i1 j$ A5 R) ?' u3 m) u                self.connect_config.get("ssh_port", 22)
            ),
( U# O0 j7 H  z, }5 R            ssh_username=self.connect_config.get("ssh_username", ‘白名单服务器账号 ’),
% _0 |- a4 O8 f            ssh_password=self.connect_config.get("ssh_password", ‘白名单服务器密码 ’),
            # 内网数据库地址和端口
+ k+ i5 }( g$ I+ W            remote_bind_address=(self.connect_config.get("dbserver", '目标数据库地址'), self.connect_config.get("ssh_mssql_port", ‘数据库端口’))
        )
        server.start()
        return server  # 远程主机上的mssql通过ssh连接映射到本地的端口
6 a8 z# d: U. {* u. W/ h; Wpymssql 连接数据库核心代码
6 `0 k5 I5 f/ S8 k9 t& C( U  x9 S def get_mssql_connector(self):
       return get_mssql_connector({
           "port": self.server.local_bind_port, #非常重要
* ^4 h7 I, z* Z2 a8 F, b           "username": ‘目标数据库账号’,
           "password": ‘目标数据库密码’,
           "db": self.connect_config.get('db', ‘目标数据库’)
       })

% n' u* O$ S1 j6 ^1 ?  c& Q9 n- E4 Ydef __get_mssql_connector(database_config):
    host = database_config.get('host', '127.0.0.1')
    port = database_config.get('port', ‘数据库端口’)
    user = database_config.get('username', ‘目标数据库账号’)
% a4 d2 D6 J4 U- t$ q% P$ ?    passwd = database_config.get('password', ‘目标数据库密码’)
5 k( P1 N( S1 C4 r4 Z    db = database_config.get('db', ‘目标数据库’)
" n! z, w; ]- L3 K    return  pymssql.connect(host=host, port=port, user=user, password=passwd, database=db, charset="UTF-8")
! K+ M/ S- q1 z
; I( k2 k" V1 E2 ?/ WMySql 和SQLSERVER方法一样，不一样的地方pymssql.connect改成pymysql.connect
% w* v" x1 x& ^/ L) z
; d1 G/ s, B9 y