2. 程式人生 > 實用技巧 >用Python寫個自動ssh登入遠端伺服器的小工具

用Python寫個自動ssh登入遠端伺服器的小工具

阿新 • • 發佈：2020-10-08

國慶假期已經結束了，放假幾天出行，這些天沒分享Python。今天來給大家分享一下~

很多時候我們喜歡在自己電腦的終端直接ssh連線Linux伺服器，而不喜歡使用那些有UI介面的工具區連線我們的伺服器。可是在終端使用ssh我們每次都需要輸入賬號和密碼，這也是一個煩惱，所以我們可以簡單的打造一個在Linux／Mac os執行的自動ssh登入遠端伺服器的小工具.
來個GIF動畫示例下先：

概述###

我們先理一下我們需要些什麼功能：

1. 新增/刪除連線伺服器需要的IP，埠，密碼
2. 自動輸入密碼登入遠端伺服器

對，我們就做這麼簡單的功能

開始寫程式碼
程式碼比較長，所以我也放在在Github和碼雲，地址在文章最底部:
1.我們建個模組目錄osnssh(Open source noob ssh)，然後在下面再建兩個目錄，一個用來放主程式取名叫bin吧，一個用來儲存登入資料(IP, 埠，密碼)叫data吧。

-osnssh
    -bin
    -data

1.設定程式：新增/刪除IP，埠，密碼. 建立py檔案bin/setting.py:

#!/usr/bin/env python
#-*-coding:utf-8-*-
import re, base64, os, sys
path = os.path.dirname(os.path.abspath(sys.argv[0]))
'''
選項配置管理
__author__ = 'allen woo'
'''
def add_host_main():
    while 1:
        if add_host():
            break
        print("\n\nAgain:")

def add_host():
    '''
    新增主機資訊
    :return: 
    '''
    print("================Add=====================")
    print("[Help]Input '#q' exit")
    # 輸入IP
    host_ip = str_format("Host IP:", "^(25[0-5]|2[0-4][0-9]|1[0-9][0-9]|[1-9]?[0-9])\.(25[0-5]|2[0-4][0-9]|1[0-9][0-9]|[1-9]?[0-9])\.(25[0-5]|2[0-4][0-9]|1[0-9][0-9]|[1-9]?[0-9])\.(25[0-5]|2[0-4][0-9]|1[0-9][0-9]|[1-9]?[0-9])$")
    if host_ip == "#q":
        return 1
    # 輸入埠
    host_port = str_format("Host port(Default 22):", "[0-9]+")
    if host_port == "#q":
        return 1
    # 輸入密碼
    password = str_format("Password:", ".*")
    if password == "#q":
        return 1
    # 密碼加密
    password = base64.encodestring(password)
    # 輸入使用者名稱
    name = str_format("User Name:", "^[^ ]+$")
    if name == "#q":
        return 1
    elif not name:
        os.system("clear")
        print("[Warning]:User name cannot be emptyg")
        return 0
    
    # The alias
    # 輸入別名
    alias = str_format("Local Alias:", "^[^ ]+$")
    if alias == "#q":
        return 1
    elif not alias:
        os.system("clear")
        print("[Warning]:Alias cannot be emptyg")
        return 0
    # 開啟資料儲存檔案
    of = open("{}/data/information.d".format(path))
    hosts = of.readlines()
    # 遍歷檔案資料，查詢是否有存在的Ip，埠，還有別名
    for l in hosts:
        l = l.strip("\n")
        if not l:
            continue
        l_list = l.split(" ")
        if host_ip == l_list[1] and host_port == l_list[2]:
            os.system("clear")
            print("[Warning]{}:{} existing".format(host_ip, host_port))
            return 0
        if alias == l_list[4]:
            os.system("clear")
            print("[Warning]Alias '{}' existing".format(alias))
            return 0
    of.close()
    # save
    # 儲存資料到資料檔案
    of = open("{}/data/information.d".format(path), "a")
    of.write("\n{} {} {} {} {}".format(name.strip("\n"), host_ip.strip("\n"), host_port, password.strip("\n"), alias.strip("\n")))
    of.close()
    print("Add the success:{} {}@{}:{}".format(alias.strip("\n"), name.strip("\n"), host_ip.strip("\n"), host_port, password.strip("\n")))
    return 1
   
def remove_host():
    '''
    刪除主機資訊
    :return: 
    '''
    while 1:
        # 開啟資料檔案
        of = open("{}/data/information.d".format(path))
        hosts = of.readlines()
        of.close
        l = len(hosts)
        if l <= 0:
            os.system("clear")
            print("[Warning]There is no host")
            return
            
        print("================Remove================")
        print("+{}+".format("-"*40))
        print("|     Alias   UserName@IP:PORT")
        hosts_temp = []
        n = 0
        # 遍歷輸出所以資訊(除了密碼)供選擇
        for i in range(0, l):
            if not hosts[i].strip():
                continue
            v_list = hosts[i].strip().split(" ")
            print("+{}+".format("-"*40))
            print("| {} | {}   {}@{}:{}".format(n+1, v_list[4], v_list[0], v_list[1], v_list[2]))
            n += 1
            hosts_temp.append(hosts[i])
        hosts = hosts_temp[:]
        print("+{}+".format("-"*40))
        c = raw_input("[Remove]Choose the Number or Alias('#q' to exit):")
        is_alias = False
        is_y = False
        try:
            c = int(c)
            if c > l or c < 1:
                os.system("clear")
                print("[Warning]:There is no")
                continue
            del hosts[c-1]
            is_y = True
            
        except:
            is_alias = True
        if is_alias:
            if c.strip() == "#q":
                os.system(
 
"clear")
                break  
            n = 0
            for l in hosts:
                if c.strip() == l.split(" ")[4].strip():
                    del hosts[n]
                    is_y = True 
                n += 1
        if not is_y:
            os.system("clear")
            print("[Warning]:There is no"
 
)
            continue
        else: 
            # save
            # 再次確認是否刪除
            c = raw_input("Remove?[y/n]:")
            if c.strip().upper() == "Y":
                of = open("{}/data/information.d".format(path), "w")
                for l in hosts:
                    of.write(l)
                print("Remove the success！")
                of.close()

def str_format(lable, rule):
    '''
    用於驗證輸入的資料格式
    :param lable: 
    :param rule: 
    :return: 
    '''
    while 1:
        print("{} ('#q' exit)".format(lable))
        temp = raw_input().strip()
        m = re.match(r"{}".format(rule), temp)
        if m:
            break
        elif "port" in lable:
            temp = 22
            break
        elif temp.strip() == "#q":
            os.system("clear")
            break
        os.system("clear")
        print("[Warning]:Invalid format")
    
    return temp

2. 我們再新增一個函式在setting.py用於輸出我們的資訊，也就是about me。

def about():
    '''
    輸出關於這個程式的資訊
    :return: 
    '''
    of = open("{}/bin/about.dat".format(path))
    rf = of.read()
    try:
        info = eval(rf)
        os.system("clear")
        print("================About osnssh================")
        for k,v in info.items():
            print("{}: {}".format(k, v))
    except:
        print("For failure.")
    return

然後在bin目錄下面建立個檔案about.dat寫入我們的一些資訊，比如：

{
    "auther":"Allen Woo",
    "Introduction":"In Linux or MAC using SSH, do not need to enter the IP and password for many times",
    "Home page":"",
    "Download address":"https://github.com/osnoob/osnssh",
    "version":"1.1.0",
    "email":"[email protected]"
}

好了設定程式就這樣了：

2. 自動登入遠端伺服器程式：在bin建個py檔案叫auto_ssh.py：
注意：這裡我們需要先安裝個包叫：pexpect， 使用者終端互動，捕捉互動資訊實現自動輸入密碼。
安裝pexpect：

pip install pexpect

然後開始寫程式碼：

#!/usr/bin/env python
#-*-coding:utf-8-*-
import os, sys, base64
import pexpect
path = os.path.dirname(os.path.abspath(sys.argv[0]))

def choose():
    # 開啟我們的資料檔案
    of = open("{}/data/information.d".format(path))
    hosts = of.readlines()
    hosts_temp = []
    for h in hosts:
        if h.strip():
            hosts_temp.append(h)
    hosts = hosts_temp[:]
    l = len(hosts)
    if l <= 0:
        os.system("clear")
        print("[Warning]Please add the host server")
        return
    while 1:
        
        print("=================SSH===================")
        print("+{}+".format("-"*40))
        print("|     Alias   UserName@IP:PORT")
        for i in range(0, l):
            v_list = hosts[i].strip().split(" ")
            print("+{}+".format("-"*40))
            print("| {} | {}   {}@{}:{}".format(i+1, v_list[4], v_list[0], v_list[1], v_list[2]))
        print("+{}+".format("-"*40))
        c = raw_input("[SSH]Choose the number or alias('#q' exit):")
        is_alias = False
        is_y = False
        try:
            c = int(c)
            if c > l or c < 1:
                os.system("clear")
                print("[Warning]:There is no")
                continue
            l_list = hosts[c-1].split(" ")
            name = l_list[0]
            host = l_list[1]
            port = l_list[2]
            password = l_list[3]
            is_y = True
            
        except:
            is_alias = True
        if is_alias:
            if c.strip() == "#q":
                os.system("clear")
                return
            for h in hosts:
                if c.strip() == h.split(" ")[4].strip():
                    l_list = h.split(" ")
                    name = l_list[0]
                    host = l_list[1]
                    port = l_list[2]
                    password = l_list[3]
                    is_y = True
        if not is_y:
            continue
        # ssh
        # 將加密儲存的密碼解密
        password = base64.decodestring(password)
        print("In the connection...")
        # 準備遠端連線，拼接ip：port
        print("{}@{}".format(name, host))
        if port == "22":
            connection("ssh {}@{}".format(name, host), password)
           
        else:
            connection("ssh {}@{}:{}".format(name, host, port), password)
  
def connection(cmd, pwd):
    '''
    連線遠端伺服器
    :param cmd: 
    :param pwd: 
    :return: 
    '''
    child = pexpect.spawn(cmd)
    i = child.expect([".*password.*", ".*continue.*?", pexpect.EOF, pexpect.TIMEOUT])
    if( i == 0 ):
        # 如果互動中出現.*password.*，就是叫我們輸入密碼
        # 我們就把密碼自動填入下去
        child.sendline("{}\n".format(pwd))
        child.interact()
    elif( i == 1):
        # 如果互動提示是否繼續，一般第一次連線時會出現
        # 這個時候我們傳送"yes"，然後再自動輸入密碼
        child.sendline("yes\n")
        child.sendline("{}\n".format(pwd))
        
        #child.interact()    
    else:
        # 連線失敗
        print("[Error]The connection fails")

好了，現在我們只需要啟動檔案了，也就是開啟程式後的第一個選單
3.再osnssh目錄下建個osnssh.py 檔案：

#!/usr/bin/env python
#-*-coding:utf-8-*-
import os, sys
sys.path.append("../")
from bin import setting, auto_ssh
path = os.path.dirname(os.path.abspath(sys.argv[0]))
'''
方便在LINUX終端使用ssh,儲存使用的IP:PORT , PASSWORD
自動登入
__author__ = 'allen woo'
'''
def main():
    while 1:
        
        print("==============OSNSSH [Menu]=============")
        print("1.Connection between a host\n2.Add host\n3.Remove host\n4.About\n[Help]: q:quit   clear:clear screen")
        print("="*40)
        c = raw_input("Please select a:")
        if c == 1 or c == "1":
            auto_ssh.choose()
        if c == 2 or c == "2":
            setting.add_host_main()
        if c == 3 or c == "3":
            setting.remove_host()
        if c == 4 or c == "4":
            setting.about()
        elif c == "clear":
            os.system("clear")
        elif c == "q" or c == "Q" or c == "quit":
            print("Bye")
            sys.exit()
        else:
            print("\n")
       
if __name__ == '__main__':
    try:
        of = open("{}/data/information.d".format(path))
    except:
        of = open("{}/data/information.d".format(path), "w")
    of.close()
    main()

終於寫完了,我們可以試一試了:

$python osnssh.py

具體的演示，就是我在文章開頭放了張GIF動畫圖片，謝謝關注！

相關推薦

用Python寫個自動ssh登入遠端伺服器的小工具

國慶假期已經結束了，放假幾天出行，這些天沒分享Python。今天來給大家分享一下~

用Python寫一個自動木馬程式

電腦作為大家日常辦公的工具，最怕的一件事情之一就是被偷，當我們的電腦被盜的時候，不僅僅是電腦本身，更重要的是電腦儲存的資料都會丟失。如何儘快的找回電腦需要我們想點辦法，今天就教大家一個好的技巧，雖說不

微信群總是有人發廣告？我用Python寫個機器人消滅他！

序幕 微信群牛皮癬，指的是在微信群裡毫無下限的群發小廣告的使用者，是微信群主最痛恨的一波人。如果熟悉早起的讀者可以知道我有一個技術交流群，但是自從建群以來就飽受小廣告的困擾。他們偽裝成正常人混進群然後不

用Python寫個開心消消樂小遊戲

本文的文字及圖片來源於網路,僅供學習、交流使用,不具有任何商業用途,如有問題請及時聯絡我們以作處理

用python寫個計算器

tkinter圖形介面 Tkinter 是 Python 的標準 GUI 庫。Python 使用 Tkinter 可以快速的建立 GUI 應用程式。由於 Tkinter 是內建到 python 的安裝包中、只要安裝好 Python 之後就能 import Tkinter 庫、而且 IDL

詳解使用Python寫一個向資料庫填充資料的小工具(推薦)

一. 背景 公司又要做一個新專案，是一個合作型專案，我們公司出web展示服務，合作伙伴線下提供展示資料。

「懶惰的美德」我用 python 寫了個自動生成給文件生成索引的指令碼

我用 python 寫了一個自動生成索引的指令碼 簡介：為了刷演算法題，建了一個 GitHub倉庫：PiperLiu / ACMOI_Journey，記錄自己的刷題軌跡，並總結一下方法、心得。想到一個需求：能不能在我每新增一條題目的筆記後，

表弟打把王者的時間，我就用python寫了個自動玩貪吃蛇的程式

文章目錄 表弟最近放假了天天打王者，作業也不做，氣得我差點想給他買三年高考五年模擬了…

我用 Python 寫了個基金漲跌通知助手！投了五百翻了四倍

2020 年，世界風雲變化，基金也是。 我們都知道股票或者債券，它們都屬於直接投資，而基金是間接投資，我們把錢投在基金裡，相當於把這些錢交給專業基金公司的專業經理人，這些經理人再去投資股票或者債券。理論上比

偷偷用 Python 寫了個摸魚監控程序

前言 上班就想摸魚，一直上班那是不可能的，總要放鬆一下，摸摸魚啥的吧。繼打遊戲、看視訊等摸魚行為被監控後，現在打工人離

使用python寫一個自動瀏覽文章的指令碼例項

建立python檔案fling.py,程式碼如下： #!/usr/bin/env monkeyrunner import time from com.android.monkeyrunner import MonkeyRunner,MonkeyDevice

500行程式碼使用python寫個微信小遊戲飛機大戰遊戲

這幾天在重溫微信小遊戲的飛機大戰，玩著玩著就在思考人生了，這飛機大戰怎麼就可以做的那麼好，操作簡單，簡單上手。

用python寫測試資料檔案過程解析

這篇文章主要介紹了用python寫測試資料檔案過程解析,文中通過示例程式碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值,需要的朋友可以參考下

為什麼黑客都用python(123個黑客必備的Python工具)

文章開始前首先讓我們來了解一下什麼是Python。 Python 是一個高層次的結合瞭解釋性、編譯性、互動性和麵向物件的指令碼語言。

用Python寫一個“離線語音提示器”來提醒我們別忘記了時間

前言 本文的文字及圖片來源於網路,僅供學習、交流使用,不具有任何商業用途,版權歸原作者所有,如有問題請及時聯絡我們以作處理。

用python寫爬蟲簡單嗎

所謂網路爬蟲，通俗的講，就是通過向我們需要的URL發出http請求，獲取該URL對應的http報文主體內容，之後提取該報文主體中我們所需要的資訊。

用django 寫個使用者管理頁面

1. 新建一個app01 python manage.py startapp app01 1 # 在 project 在外層 urls 的配置： from django.contrib import admin

用Python寫一個隨機數字生成程式碼，5行程式碼超簡單

本文的文字及圖片來源於網路,僅供學習、交流使用,不具有任何商業用途,版權歸原作者所有,如有問題請及時聯絡我們以作處理。

—用python寫截圖小工具

前言 一、技術實現 1. 相關庫2. 定義視窗3. 繪製工具條4. 定義事件函式 二、後續改進

剛上班的朋友老是想著還有多久下班，用程式碼寫個計算下班時間指令碼

朋友剛畢業，每天從出門就開始等著下班，每天16:30準點下班，上班的時候老是跟我聊還有多久下班，我想著我是做Python的。