很多時候我們喜歡在自己電腦的終端直接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, 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
新聞熱點
疑難解答