国产探花免费观看_亚洲丰满少妇自慰呻吟_97日韩有码在线_资源在线日韩欧美_一区二区精品毛片,辰东完美世界有声小说,欢乐颂第一季,yy玄幻小说排行榜完本

首頁 > 編程 > Python > 正文

用python寫個自動SSH登錄遠程服務器的小工具(實例)

2019-11-25 16:05:23
字體:
來源:轉載
供稿:網友

很多時候我們喜歡在自己電腦的終端直接ssh連接Linux服務器,而不喜歡使用那些有UI界面的工具區連接我們的服務器??墒窃诮K端使用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, syspath = 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 1def 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":"xiaopingwoo@163.com"}

好了設置程序就這樣了:

2. 自動登錄遠程服務器程序:在bin建個py文件叫auto_ssh.py:

注意:這里我們需要先安裝個包叫:pexpect, 用戶終端交互,捕捉交互信息實現自動輸入密碼。

安裝pexpect:

pip install pexpect

然后開始寫代碼:

#!/usr/bin/env python#-*-coding:utf-8-*-import os, sys, base64import pexpectpath = 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, syssys.path.append("../")from bin import setting, auto_sshpath = 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

以上這篇用python寫個自動SSH登錄遠程服務器的小工具(實例)就是小編分享給大家的全部內容了,希望能給大家一個參考,也希望大家多多支持武林網。

發表評論 共有條評論
用戶名: 密碼:
驗證碼: 匿名發表
主站蜘蛛池模板: 吴旗县| 安丘市| 年辖:市辖区| 册亨县| 买车| 噶尔县| 农安县| 元朗区| 虹口区| 顺昌县| 伊宁市| 鹿泉市| 平罗县| 万宁市| 海门市| 夏河县| 梓潼县| 土默特右旗| 思南县| 陵川县| 积石山| 额尔古纳市| 通化市| 朝阳县| 阿勒泰市| 沾益县| 茂名市| 九台市| 永修县| 景宁| 鸡泽县| 德格县| 大兴区| 望江县| 驻马店市| 金乡县| 治多县| 贡嘎县| 库尔勒市| 闻喜县| 太保市|