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

1047521767

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

7 x/ |2 e- |2 \( e场景如下：
跳板机核心代码   def __get_ssh_connector(self):        # 远程连接
. Y, h2 @) T5 h. G        # 跳板机地址 端口，服务器账号，密码配置
        server = SSHTunnelForwarder(
) h! I" A4 }: h9 X            (
                self.connect_config.get("ssh_host", ‘白名单服务器地址’),
                self.connect_config.get("ssh_port", 22)
, }9 M3 T: Y+ {9 w) e1 ]2 ?) f3 U- `            ),
            ssh_username=self.connect_config.get("ssh_username", ‘白名单服务器账号 ’),
            ssh_password=self.connect_config.get("ssh_password", ‘白名单服务器密码 ’),
# E: H* q- d% P( P6 E            # 内网数据库地址和端口
. X2 y. l8 G0 T  X( w/ k1 \            remote_bind_address=(self.connect_config.get("dbserver", '目标数据库地址'), self.connect_config.get("ssh_mssql_port", ‘数据库端口’))
  f) J6 C% H( m' Z        )
        server.start()
. T0 V+ R6 K# {6 u/ e) |0 W/ K        return server  # 远程主机上的mssql通过ssh连接映射到本地的端口
pymssql 连接数据库核心代码
' \0 o# g5 y% Q0 M5 H2 I+ l def get_mssql_connector(self):
       return get_mssql_connector({
           "port": self.server.local_bind_port, #非常重要
           "username": ‘目标数据库账号’,
6 t. R* J/ n0 F) @4 i           "password": ‘目标数据库密码’,
! F: Q: d/ q/ }0 w. e           "db": self.connect_config.get('db', ‘目标数据库’)
       })
6 @3 H& r3 f4 P8 Q% ?: f  u* g4 q
def __get_mssql_connector(database_config):
    host = database_config.get('host', '127.0.0.1')
    port = database_config.get('port', ‘数据库端口’)
9 v1 c& E4 G; k    user = database_config.get('username', ‘目标数据库账号’)
. \1 ]: s1 g- Q9 Q+ J! {    passwd = database_config.get('password', ‘目标数据库密码’)
1 C# Q2 {  F- F# R" T1 x    db = database_config.get('db', ‘目标数据库’)
* ~- S2 |) g  c6 P  U    return  pymssql.connect(host=host, port=port, user=user, password=passwd, database=db, charset="UTF-8")
+ [, N+ u! k' h. f5 b5 G' @, Z
2 c" m7 y, f1 p8 qMySql 和SQLSERVER方法一样，不一样的地方pymssql.connect改成pymysql.connect
, J& O& O$ s# Q. B  [
" n, [- a( ?5 m