- 在线时间
- 514 小时
- 最后登录
- 2023-12-1
- 注册时间
- 2018-7-17
- 听众数
- 15
- 收听数
- 0
- 能力
- 0 分
- 体力
- 39375 点
- 威望
- 0 点
- 阅读权限
- 255
- 积分
- 12508
- 相册
- 0
- 日志
- 0
- 记录
- 0
- 帖子
- 1388
- 主题
- 1158
- 精华
- 0
- 分享
- 0
- 好友
- 15
TA的每日心情 | 开心 2023-7-31 10:17 |
---|
签到天数: 198 天 [LV.7]常住居民III
- 自我介绍
- 数学中国浅夏
|
发表于 2021-10-28 22:18
|显示全部楼层
|
为了安全起见,很多公司服务器数据库的访问多半是要做限制的,由专门的DBA管理,而且都是做的集群,数据库只能内网访问,所以就有一个直接的问题是,往往多数时候,在别的机器上(比如自己本地),是不能访问数据库的,给日常开发调试造成了很大不便。最近几天在做Python相关项目,有个需求 ,是希望在任何机器上都可以ssh到某台在数据库白名单的机器上,然后访问数据库,不然的话就要去服务器安装Python环境,运行程序,比较麻烦,翻阅多篇博客文章,决定自己去实现。
1 o0 P- _! H3 ^! F. c' Q' G涉及库:pymssql、sshtunnel& t5 H4 i/ W7 ]/ A! i1 S
涉及数据库:SQLSERVER4 Q( I. I" T6 H8 X$ `# N+ P2 W, N) ^
7 A( i( b" M2 g+ o0 N, K
场景如下:
4 T" [# f% I1 ?5 e! W$ e跳板机核心代码 def __get_ssh_connector(self): # 远程连接
& B7 j0 w- V3 C1 m" S- {& ~ # 跳板机地址 端口,服务器账号,密码配置* ?+ ]; p9 ^+ L3 S; o" t' E1 p
server = SSHTunnelForwarder(" ^1 U% m# t. ]( A
(
* M) Q6 F; H8 v+ x" E# t self.connect_config.get("ssh_host", ‘白名单服务器地址’),
! }7 [4 X( s2 [ self.connect_config.get("ssh_port", 22)" w8 P/ m$ d S* Y! I/ a) Y
),7 d6 J/ H1 p! a; D$ b3 ?
ssh_username=self.connect_config.get("ssh_username", ‘白名单服务器账号 ’),
( Q0 u, L0 D" ~0 n- v9 Z" J ssh_password=self.connect_config.get("ssh_password", ‘白名单服务器密码 ’),6 E% |5 v9 d. A' L- Z6 z
# 内网数据库地址和端口
4 k3 S' {' T: N: y: g remote_bind_address=(self.connect_config.get("dbserver", '目标数据库地址'), self.connect_config.get("ssh_mssql_port", ‘数据库端口’))/ Y! t8 y9 p& {! L
)
5 K' C. u$ j8 A$ e$ z \4 \6 i server.start()+ `1 M7 S* i3 e- K1 ~0 X# ]
return server # 远程主机上的mssql通过ssh连接映射到本地的端口
' s0 r3 ?/ t' O3 _9 w; ^) Npymssql 连接数据库核心代码 6 B* q& c7 P, G
def get_mssql_connector(self):
0 X1 f0 L$ c9 } l2 G return get_mssql_connector({& Z2 j" E0 E! L0 G$ E/ T
"port": self.server.local_bind_port, #非常重要
) E r, H2 S) I* @1 I3 A "username": ‘目标数据库账号’,) k. ^" t9 Z" s, h! c1 N* \
"password": ‘目标数据库密码’,# n$ H1 Q* }) B2 P" l8 _
"db": self.connect_config.get('db', ‘目标数据库’)! a: O/ B0 p, c4 {) o$ p
}): a; T- C9 x4 }3 C* t; D8 |
2 S5 x) N6 k) A% @
def __get_mssql_connector(database_config):
; b' y$ |5 N4 H3 T! R: o host = database_config.get('host', '127.0.0.1')
: |) h6 Y8 l5 H* \& H& M port = database_config.get('port', ‘数据库端口’)
4 J# B5 P. N- s& C4 Q user = database_config.get('username', ‘目标数据库账号’)0 _3 t# K# C& q7 v! V
passwd = database_config.get('password', ‘目标数据库密码’)
9 x% }- t' {' {0 w1 t7 S, l8 t5 r% r db = database_config.get('db', ‘目标数据库’); o" A% [6 g' i" j+ J
return pymssql.connect(host=host, port=port, user=user, password=passwd, database=db, charset="UTF-8")8 ?. l) Q" T) F- C2 z+ S
# D1 _( }& i6 d0 J+ U6 P2 [6 ?
MySql 和SQLSERVER方法一样,不一样的地方pymssql.connect改成pymysql.connect! t1 o0 e4 _) I* x
0 \/ z- ]# ?+ y6 ?) v |
zan
|