栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 软件开发 > 后端开发 > Python

Python:使用SSH隧道链接数据库

Python 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

Python:使用SSH隧道链接数据库

目录

 

一、SSH隧道

二、使用 sshtunnel 包建立隧道

2.1)使用账户、密码登录建立隧道

2.2)使用密钥文件代替密码

三、通过SSH隧道链接数据库


一、SSH隧道

SSh隧道,就是当本地机器因为网络限制无法直接连接目标机器时,可以先在本地机器上,用ssh协议登录一台跳板机,然后本地机器可以通过跳板机链接目标机器。

本地机器和跳板机之间的这个链接,就是SSH隧道。

二、使用 sshtunnel 包建立隧道

2.1)使用账户、密码登录建立隧道

在Python 中可以使用 sshtunnel 包创建 SSH 隧道,演示代码如下:

from sshtunnel import SSHTunnelForwarder

def get_ssh_tunnel():
    server = SSHTunnelForwarder(
            ssh_address_or_host=('192.168.1.2', 22),
            ssh_username='username',
            ssh_password='ssh-password',
            remote_bind_address=('10.16.3.3', 3306),
            local_bind_address=('0.0.0.0', 3306)
            )   
    server.start()

    return server

参数含义:

  • ssh_address_or_host : 跳板机的 IP 和 Port
  • ssh_username : 登录跳板机时使用的账户名称
  • ssh_password : 登录跳板机时使用的密码
  • remote_bind_address : 本地机器要访问的目标机器的IP(如果目标机器就是跳板机自己,则是 127.0.0.1)和 Port
  • local_bind_address : SSH 隧道监听的本地机器的 IP(可以是 0.0.0.0 监听所有IP,也可以选择指定的网口) 和 Port

2.2)使用密钥文件代替密码

上面是使用了账户和密码登录跳板机建立的隧道,但是使用密码明文有密码泄露的风险,其实 sshtunnel 还可以使用密钥进行登录,只需要把 ssh_password 参数去掉,换成 ssh_pkey 即可,代码如下:

from sshtunnel import SSHTunnelForwarder

def get_ssh_tunnel():
    server = SSHTunnelForwarder(
            ssh_address_or_host=('192.168.1.2', 22),
            ssh_username='username',
            ssh_pkey="/var/ssh/id_rsa",  # 密钥文件的路径
            remote_bind_address=('10.16.3.3', 3306),
            local_bind_address=('0.0.0.0', 3306)
            )   
    server.start()

    return server

其中 /var/ssh/id_rsa 就是密钥文件的路径。

想要使用密钥的方法创建 SSH 隧道,需要先把公钥 id_rsa.pub 的内容添加到跳板机的 ~/.ssh/authorized_keys 中才可以。

三、通过SSH隧道链接数据库

创建完了隧道之后,就可以连接目标机器了。这里使用的例子是连接目标机器上的 mysql 数据库。

  • 跳板机IP : 192.168.1.2
  • mysql数据库IP : 10.16.3.3
from sshtunnel import SSHTunnelForwarder

def get_ssh_tunnel():
    server = SSHTunnelForwarder(
            ssh_address_or_host=('192.168.1.2', 22),
            ssh_username='username',
            ssh_password='ssh-password',
            remote_bind_address=('10.16.3.3', 3306),
            local_bind_address=('0.0.0.0', 3306)
            )
    server.start()

    return server

import MySQLdb

def get_db_conn(sshserver):
    dbip = sshserver.local_bind_host
    dbport = sshserver.local_bind_port
    db = MySQLdb.connect(
            host=dbip,
            port=dbport,
            user='db-username',
            passwd='db-password',
            db='db-name',
            charset='utf8')

    return db

if "__main__" == __name__:
    sshserver = get_ssh_tunnel()
    db = get_db_conn(sshserver)
    cursor = db.cursor()
    cursor.execute("SELECT VERSION()")
    data = cursor.fetchone()
    print("DB version : {0}".format(data))
    db.close()

 

转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/444481.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号