sshfind通过主机别名查找主机信息Python实现

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
#根据主机别名查找openSSH配置文件~/.ssh/config对应主机配置项
#自用sshfind命令python实现版本
#原始sshfind为shell脚本实现，参看：/v/bin/sshfind
import sys
import os
import re

sshConfigFile="~/.ssh/config"
sshConfigFile=os.path.expanduser(sshConfigFile)

if len(sys.argv)<2 or (len(sys.argv)==2 and (str(sys.argv[1]).lower()=="--help" or str(sys.argv[1]).lower()=="-h")):
    print("""
    /v/bin/sshfind.py
    使用主机别名或别名部分关键字查找主机配置项信息，查找 ~/.ssh/config 文件内容.
    注：脚本优先做主机别名全字匹配，没有结果再尝试模糊匹配.

Usage:
        sshfind.py [keyword]
Example:
        sshfind.py racknerd
    
    """)
    sys.exit(0)
else:
    hoststr=str(sys.argv[1]).strip()

foundCount=0
wrapMatch=False #全字匹配标志符

with open(sshConfigFile,"r") as read_f:
    Tag=False
    for num,line in enumerate(read_f.readlines(),1):
        #if re.match(r'Host .*[ ]?'+hoststr+'[ $]?',line,re.I):
        if re.match(r'Host .*[^-\.]?'+hoststr+'[ $]?',line,re.I):
            Tag=True
            foundCount+=1
            if re.match(r'.* '+hoststr+'( |$)',line,re.I):
                #标识全字匹配
                wrapMatch=True
            #找到两个及以上主机时，打印空行隔开
            if foundCount>=2:
                print("\n")
            #找到主机匹配则先打印开始行的行号，以供sshedit命令使用
            print(num)
            print(line,end='')
            continue
        if Tag==True and re.match(r'[ ]*Host ',line,re.I):
            Tag=False
        if Tag==True:
            print(line,end='')

if wrapMatch==True and foundCount==1:
    print("\n\t以上为全字匹配结果")
    
print("\nsshfind执行完毕；Python版本:%s"%sys.version)