以前在windows系统下面使用xshell来管理阿里云上的服务器，winscp来上传网站的资料，由于winscp使用的私钥必须是.ppk格式，而xshell使用普通的私钥，每次都要用putty工具来转换，很麻烦。后来使用xftp来替代winscp，xftp和xshell使用同样的私钥，这套组合在windows平台上面管理服务器非常方便，推荐使用。
然而Mac OS上面，没有这个套工具，由于Terminal可以直接使用sftp和ssh命令，用来管理阿里云服务器也很不错。但是，管理多台服务器非常麻烦，最好脚本统一管理。因此我设计的脚本具有如下功能：
- 同一个命令可以用于ssh连接或者sftp连接
- 一个脚本最好能管理所有的服务器
- 脚本能够展示所有可用的服务器，方便选择
- 要有良好的帮助信息
最终的命令格式如下：
sshlink -l展示所有可以用的服务器名
sshlink -h或者sshlink --help显示帮助信息
sshlink -s server_name表示使用ssh协议连接服务器
sshlink -f server_name表示使用sftp协议连接服务器
代码如下，注意：代码中隐藏了关键的IP信息，请自行修改:
#! /usr/bin/env python
# -*- coding: utf-8 -*-
"""
作用：'sshlink -[lsfh] severName'命令对服务器进行访问
arg参数如下：
-l: 显示可用的服务器名字
-s: 使用ssh命令进行连接服务器
-f: 使用sftp命令进行连接服务器
-h: 显示帮助信息
－-help: 显示帮助信息
"""
import sys
import os
argv = sys.argv
BASE_URL = '/Users/HomePath/.ssh/' # 存放私钥的路径
serverName = {
'server1': ('ip1', 'server1.key'),
'server2': ('ip2', 'server2.key'),
}
def sshlink():
if len(argv) == 1: # 没有参数的时候显示帮助信息
print __doc__
elif argv[1] == '-l': # 显示所有的可用服务器名
print 'Servers can be used: ', ' '.join(serverName.keys())
elif argv[1] == '-s': # 使用ssh进行连接
if len(argv) == 3 and argv[2] and argv[2] in serverName:
sshcmd = list()
sshcmd.append('ssh root@' + serverName[argv[2]][0])
sshcmd.append('-i')
sshcmd.append(os.path.join(BASE_URL, serverName[argv[2]][1]))
os.system(' '.join(sshcmd))
else:
print "Need Server Name: ", ' '.join(serverName.keys())
elif argv[1] == '-f': #使用sftp进行连接
if len(argv) == 3 and argv[2] and argv[2] in serverName:
sftpcmd = list()
sftpcmd.append('sftp -i')
sftpcmd.append(os.path.join(BASE_URL, serverName[argv[2]][1]))
sftpcmd.append('root@' + serverName[argv[2]][0])
os.system(' '.join(sftpcmd))
else:
print "Need Server Name: ", ' '.join(serverName.keys())
elif argv[1] == '--help' or argv[1] == '-h':
print __doc__
print "Servers can be used: ", ' '.join(serverName.keys())
else:
print "Only support -l, -s, -f, --help"
if __name__ == '__main__':
sshlink()
使用sudo ln -s在系统路径中建立一个软连接，并且赋予执行权限sudo chmod 700 sshlink，然后就可以正常使用命令了。如果需要增加服务器，那么就修改该代码中的ServerName字典，然后就能正常使用。
