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

1047521767

为了安全起见，很多公司服务器数据库的访问多半是要做限制的，由专门的DBA管理，而且都是做的集群，数据库只能内网访问，所以就有一个直接的问题是，往往多数时候，在别的机器上（比如自己本地），是不能访问数据库的，给日常开发调试造成了很大不便。最近几天在做Python相关项目，有个需求 ，是希望在任何机器上都可以ssh到某台在数据库白名单的机器上，然后访问数据库，不然的话就要去服务器安装Python环境，运行程序，比较麻烦，翻阅多篇博客文章，决定自己去实现。
; K; o2 z% r  e" T# }, ]涉及库：pymssql、sshtunnel
9 j1 }1 v+ _/ v% ?# r# r/ M涉及数据库：SQLSERVER
' O  @& f/ \7 e; k$ {4 M
: f6 a1 U. X6 K6 ~2 p" J: P" W7 o场景如下：
, e5 p" u2 W& L' d跳板机核心代码   def __get_ssh_connector(self):        # 远程连接
( Z/ q' O) J0 j/ ~- O        # 跳板机地址 端口，服务器账号，密码配置
* |* ~! O" p0 x+ r1 _9 P        server = SSHTunnelForwarder(
) S7 Z. y/ u- _3 J& w/ p  g            (
                self.connect_config.get("ssh_host", ‘白名单服务器地址’),
                self.connect_config.get("ssh_port", 22)
6 x$ ^: H* x2 x& H+ D            ),
- p1 u" q& ?" \" H: ?4 k            ssh_username=self.connect_config.get("ssh_username", ‘白名单服务器账号 ’),
            ssh_password=self.connect_config.get("ssh_password", ‘白名单服务器密码 ’),
            # 内网数据库地址和端口
            remote_bind_address=(self.connect_config.get("dbserver", '目标数据库地址'), self.connect_config.get("ssh_mssql_port", ‘数据库端口’))
        )
        server.start()
        return server  # 远程主机上的mssql通过ssh连接映射到本地的端口
& {1 u* \+ z7 ^; Rpymssql 连接数据库核心代码
def get_mssql_connector(self):
% o4 x- `) Y7 o       return get_mssql_connector({
* A. A$ ?" d; q, b6 {# x0 j! q' l$ p           "port": self.server.local_bind_port, #非常重要
' k5 D6 W& ]' ?: c           "username": ‘目标数据库账号’,
           "password": ‘目标数据库密码’,
( ^' X0 W; A4 z( u" {, r6 }           "db": self.connect_config.get('db', ‘目标数据库’)
       })

def __get_mssql_connector(database_config):
    host = database_config.get('host', '127.0.0.1')
    port = database_config.get('port', ‘数据库端口’)
    user = database_config.get('username', ‘目标数据库账号’)
! x5 P  h, H( h) e- v0 t2 ?  G    passwd = database_config.get('password', ‘目标数据库密码’)
, |* `8 s; _2 ]+ V0 R5 v    db = database_config.get('db', ‘目标数据库’)
    return  pymssql.connect(host=host, port=port, user=user, password=passwd, database=db, charset="UTF-8")
' s# ]' ?0 \4 l( f6 \
$ p, i* w7 i" l* e' k( d' \2 VMySql 和SQLSERVER方法一样，不一样的地方pymssql.connect改成pymysql.connect
* e) p$ Z7 g3 }! z
; `5 N. d" _0 h' F