Python用MySQLdb， pymssql 模塊通過sshtunnel連接遠程數據庫

安全起見，數據庫的訪問多半是要做限制的，所以就有一個直接的問題是，往往多數時候，在別的機器上（比如自己本地），是不能訪問數據庫的，給日常使用造成了很大不便。所以前幾天做了個需求，是希望在任何機器上都可以ssh到某台在數據庫白名單的機器上，然后訪問數據庫。

經人推薦，查詢了一個工具叫sshtunnel ，需要在你想要登錄數據庫的機器上安裝。

sshtunnel的基本介紹 ： http://sshtunnel.readthedocs.io/en/latest/?badge=latest

這里邊的圖2，正好是我們描述的場景，數據庫在遠程某台服務器上，只能同在一起的另一台服務器才能訪問，但是我們本地就需要ssh到這台服務器上

 

但是這個里邊涉及到的代碼是服務器相關，遠程連接的是服務器，我們需要的是遠程服務器的數據庫，這個也一樣，只需要變更代碼中后半部分就可以了。

原代碼是下邊這樣：

import paramiko
from sshtunnel import SSHTunnelForwarder

with SSHTunnelForwarder(
    (REMOTE_SERVER_IP, 443),
    ssh_username="",
    ssh_pkey="/var/ssh/rsa_key",
    ssh_private_key_password="secret",
    remote_bind_address=(PRIVATE_SERVER_IP, 22),
    local_bind_address=('0.0.0.0', 10022)
) as tunnel:
    client = paramiko.SSHClient()
    client.load_system_host_keys()
    client.set_missing_host_key_policy(paramko.AutoAddPolicy())
    client.connect('127.0.0.1', 10022)
    # do some operations with client session
    client.close()

print('FINISH!')

連接mysql數據庫

連接mysql數據庫的時候，看到網上有一位程序員已經實現了： python 使用mysqldb模塊通過ssh隧道連接mysql

代碼如下：

import MySQLdb
from sshtunnel import SSHTunnelForwarder

with SSHTunnelForwarder(
         ('sshhost.domain.com', 22),    #B機器的配置
         ssh_password="sshpasswd",
         ssh_username="sshusername",
         remote_bind_address=('mysqlhost.domain.com', mysql.port)) as server:  #A機器的配置

    conn = MySQLdb.connect(host='127.0.0.1',              #此處必須是是127.0.0.1
                           port=server.local_bind_port,
                           user='user',
                           passwd='password',
                           db='dbname')

 然后接下來的查詢什么的，直接寫在with那里邊，與conn對齊就可以了。

對於我這邊來說有一個問題是，因為我們對於數據庫連接這一部分，往往是在一個單獨的函數里，與其他數據庫的查詢插入刪除更新操作往往不在一起，這樣的話，with as 有個特點就是，離開這塊作用域，對象就被銷毀掉了，別的函數里是沒法用的，也就會出現一種情況是，連接上了，但是對象又給銷毀掉了，結果查詢的時候直接顯示這個錯誤：OperationalError: (2006, 'MySQL server has gone away')， 而網上查詢這個錯誤，多半說的是因為你查詢的 sql操作的時間過長，或者是傳送的數據太大 ，但是我這個地方實際上就是因為出了with as 的作用域，導致連接又給關閉掉了，所以出現這樣的結果。

關於 with as ，有篇文章寫得很詳盡。理解Python中的with…as…語法

所以我把上邊那個ssh代碼改掉了，像是sshtunnel文檔里邊圖一所對應的代碼，一樣，將SSHTunnelForwarder出來的對象賦值給server，然后啟動server，然后進行一系列操作之后，再stop掉。

本來數據庫連接我們寫成了一個單獨的函數，改了之后，直接也還放在這個函數里就好了，替代原來的connect語句。

def connect(self):
    '''
    self.client = MySQLdb.connect(host=self.server, port=self.port, user=self.user,
                                  passwd=self.password, db=self.database,
                                  charset=self.charset)
    # log.info('Connect to MySQL Server: ' + self.server)
    '''

    server = SSHTunnelForwarder(
            ('sshhost.domain.com', 22),  # B機器的配置
            ssh_password='ssh_password',
            ssh_username='ssh_username',
            remote_bind_address=('mysqlhost.domain.com', mysql.port)
    )
    server.start()

    self.client = MySQLdb.connect(host='127.0.0.1',  # 此處必須是是127.0.0.1
                                  port=server.local_bind_port,
                                  user='username',
                                  passwd='password',
                                  db='dbname')

然后在進行查詢更新刪除等操作的時候，先連接一下數據庫就好了，用self.client.

連接sqlserver數據庫

跟mysql的一致，但是db那里要注意，SQLServer的是database, 然后是pymssql.connect就可以了，但是這個地方還要說我踩過的一個坑，我寫完sqlserver之后怎么連接都連不上數據庫，后來才發現是版本的問題，我把本地SQLServer更新了之后就可以了。感覺版本是個大坑