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

1047521767

为了安全起见，很多公司服务器数据库的访问多半是要做限制的，由专门的DBA管理，而且都是做的集群，数据库只能内网访问，所以就有一个直接的问题是，往往多数时候，在别的机器上（比如自己本地），是不能访问数据库的，给日常开发调试造成了很大不便。最近几天在做Python相关项目，有个需求 ，是希望在任何机器上都可以ssh到某台在数据库白名单的机器上，然后访问数据库，不然的话就要去服务器安装Python环境，运行程序，比较麻烦，翻阅多篇博客文章，决定自己去实现。
4 C, ]8 b, |  P! ~9 \0 U& J, v8 s. A涉及库：pymssql、sshtunnel
4 }2 l, ^4 \$ B涉及数据库：SQLSERVER

场景如下：
) d1 b, n" r1 X4 x( L* P跳板机核心代码   def __get_ssh_connector(self):        # 远程连接
1 V! r. t. I# T* ^& Q( V! M        # 跳板机地址 端口，服务器账号，密码配置
1 `5 b% C7 P0 W. `- c1 K        server = SSHTunnelForwarder(
6 g( E0 z) ~3 J- T6 ~3 T3 D            (
                self.connect_config.get("ssh_host", ‘白名单服务器地址’),
. M7 n# z% B" U3 d                self.connect_config.get("ssh_port", 22)
! ?1 [! g& S- _1 }6 L5 ~* A            ),
            ssh_username=self.connect_config.get("ssh_username", ‘白名单服务器账号 ’),
            ssh_password=self.connect_config.get("ssh_password", ‘白名单服务器密码 ’),
            # 内网数据库地址和端口
, ]9 o$ Q! Q5 g( C            remote_bind_address=(self.connect_config.get("dbserver", '目标数据库地址'), self.connect_config.get("ssh_mssql_port", ‘数据库端口’))
        )
* i/ |2 H+ G1 g7 ?        server.start()
        return server  # 远程主机上的mssql通过ssh连接映射到本地的端口
pymssql 连接数据库核心代码
4 B6 l2 E4 `5 w def get_mssql_connector(self):
       return get_mssql_connector({
           "port": self.server.local_bind_port, #非常重要
" }! S& a. ^" }0 `" h; g+ h           "username": ‘目标数据库账号’,
3 s* T4 P' ^; M5 a* ]  S' [4 T' \           "password": ‘目标数据库密码’,
           "db": self.connect_config.get('db', ‘目标数据库’)
) b& a& r) F& |& N       })

7 r: r* |7 m. P2 Adef __get_mssql_connector(database_config):
& n. t& u! A1 w5 E$ f    host = database_config.get('host', '127.0.0.1')
& x) S+ @1 ]( [3 y) i    port = database_config.get('port', ‘数据库端口’)
    user = database_config.get('username', ‘目标数据库账号’)
# u1 g, k4 ~( o) m  Q    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")

MySql 和SQLSERVER方法一样，不一样的地方pymssql.connect改成pymysql.connect
; @) q$ F( F7 V  x# s
5 y' o' F8 q2 }- N% \