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

1047521767

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

场景如下：
: a2 m2 v, h9 b1 Z9 p* k! ?7 u跳板机核心代码   def __get_ssh_connector(self):        # 远程连接
        # 跳板机地址 端口，服务器账号，密码配置
        server = SSHTunnelForwarder(
- M: E8 b7 G3 F            (
/ s% p' q, }! m/ F! @                self.connect_config.get("ssh_host", ‘白名单服务器地址’),
                self.connect_config.get("ssh_port", 22)
            ),
            ssh_username=self.connect_config.get("ssh_username", ‘白名单服务器账号 ’),
6 A+ R- b  S" j$ z. z            ssh_password=self.connect_config.get("ssh_password", ‘白名单服务器密码 ’),
            # 内网数据库地址和端口
            remote_bind_address=(self.connect_config.get("dbserver", '目标数据库地址'), self.connect_config.get("ssh_mssql_port", ‘数据库端口’))
. i* t  q3 g2 H7 k        )
4 m8 b1 e# {/ u: d  \3 w  t        server.start()
9 q5 h- G% M. g3 G        return server  # 远程主机上的mssql通过ssh连接映射到本地的端口
pymssql 连接数据库核心代码
def get_mssql_connector(self):
1 k# f1 g: S8 m1 c) j6 l       return get_mssql_connector({
- K, L9 E" s% ?" V7 M           "port": self.server.local_bind_port, #非常重要
5 ~" P' x6 l! J+ F2 G$ J           "username": ‘目标数据库账号’,
5 Y6 s  k, b/ c8 v' |, _" M- S0 o           "password": ‘目标数据库密码’,
: |/ z6 }0 @1 D  ~( w! r. K& B# O$ I           "db": self.connect_config.get('db', ‘目标数据库’)
3 E- J1 ^7 ^& J$ H5 g9 b' E% {       })
+ m! U  @, ^' q) U) k
def __get_mssql_connector(database_config):
    host = database_config.get('host', '127.0.0.1')
; R6 t  i6 h6 l: o1 t) q  u! \    port = database_config.get('port', ‘数据库端口’)
    user = database_config.get('username', ‘目标数据库账号’)
0 b( B$ b8 x# l# I    passwd = database_config.get('password', ‘目标数据库密码’)
; p3 ?1 J" X( @) W: G  u! d2 d    db = database_config.get('db', ‘目标数据库’)
8 l& K8 I7 P+ Q. _    return  pymssql.connect(host=host, port=port, user=user, password=passwd, database=db, charset="UTF-8")
0 D+ B+ y) r9 ^1 a5 \, z' E
MySql 和SQLSERVER方法一样，不一样的地方pymssql.connect改成pymysql.connect
4 ^$ `$ g- ~7 q5 R