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

1047521767

为了安全起见，很多公司服务器数据库的访问多半是要做限制的，由专门的DBA管理，而且都是做的集群，数据库只能内网访问，所以就有一个直接的问题是，往往多数时候，在别的机器上（比如自己本地），是不能访问数据库的，给日常开发调试造成了很大不便。最近几天在做Python相关项目，有个需求 ，是希望在任何机器上都可以ssh到某台在数据库白名单的机器上，然后访问数据库，不然的话就要去服务器安装Python环境，运行程序，比较麻烦，翻阅多篇博客文章，决定自己去实现。
涉及库：pymssql、sshtunnel
涉及数据库：SQLSERVER
( C5 G0 x/ Z$ p
场景如下：
* }0 y0 N( o/ k0 K跳板机核心代码   def __get_ssh_connector(self):        # 远程连接
$ C" I: Q4 a5 m4 `; |0 r% _        # 跳板机地址 端口，服务器账号，密码配置
        server = SSHTunnelForwarder(
            (
  u. m" ~9 f/ r$ N* t! n. V                self.connect_config.get("ssh_host", ‘白名单服务器地址’),
) U* m4 Y9 Q* @4 j: Q- M4 S                self.connect_config.get("ssh_port", 22)
0 ?9 l( F  Y. \4 Z* j$ W            ),
7 B" @: t/ i# j- D$ S7 Z; c            ssh_username=self.connect_config.get("ssh_username", ‘白名单服务器账号 ’),
6 c) m: T+ i' u/ r            ssh_password=self.connect_config.get("ssh_password", ‘白名单服务器密码 ’),
5 _) t/ s/ s* V) P1 g; o  W            # 内网数据库地址和端口
. b3 {7 z6 H. l. X8 G' Z* P            remote_bind_address=(self.connect_config.get("dbserver", '目标数据库地址'), self.connect_config.get("ssh_mssql_port", ‘数据库端口’))
/ `: ?* `2 U6 V' \        )
        server.start()
2 b! O$ z3 p/ {3 v* |& x        return server  # 远程主机上的mssql通过ssh连接映射到本地的端口
pymssql 连接数据库核心代码
def get_mssql_connector(self):
       return get_mssql_connector({
           "port": self.server.local_bind_port, #非常重要
           "username": ‘目标数据库账号’,
           "password": ‘目标数据库密码’,
2 _+ R" ^: z* l' y- F. s           "db": self.connect_config.get('db', ‘目标数据库’)
       })

def __get_mssql_connector(database_config):
% D3 C) v* ~7 p7 C9 D5 c$ J; `    host = database_config.get('host', '127.0.0.1')
0 c) R2 v; H1 r* k3 R) V    port = database_config.get('port', ‘数据库端口’)
    user = database_config.get('username', ‘目标数据库账号’)
4 b# F+ H2 t7 i6 F/ ]    passwd = database_config.get('password', ‘目标数据库密码’)
# E1 L, g% h- U5 q, S% J6 }, Q. c/ ~    db = database_config.get('db', ‘目标数据库’)
    return  pymssql.connect(host=host, port=port, user=user, password=passwd, database=db, charset="UTF-8")
+ Q: S1 f0 i' K- c, J  U" h5 {
MySql 和SQLSERVER方法一样，不一样的地方pymssql.connect改成pymysql.connect
: k+ f: B/ K' e0 h  k: m