python ssh

wind1211

1. import paramiko
   
2. import time
   
3. import sys
   
4. #Collecting Huawe Storage
   
5. #Created by wind1211 2013/07
   
6. def ssh2(ip,username,passwd,cmd):
   
7.     #print ip+","+username+","+passwd+","+cmd.rstrip()
   
8.     try:
   
9.         ssh = paramiko.SSHClient()
   
10.         ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
    
11.         ssh.connect(ip,22,username,passwd,timeout=8)
    
12.         stdin,stdout,stderr = ssh.exec_command(cmd)
    
13.         cmd_output = stdout.read()
    
14.         ssh.close()
    
15.     except:
    
16.         print "connect to "+ip+"failed"
    
17.     return cmd_output
    
18. 
    
19. def _wr_csv(buff,_file):
    
20.     fb = open(_file,"w")
    
21.     #argument buff must be list
    
22.     for item in(buff):
    
23.         try:
    
24.             fb.write(item)
    
25.         except:
    
26.             print("write"+_file+" info failed")
    
27.     fb.close()
    
28. 
    
29. #define _ocenStoreSys
    
30. def _ocenStoreSys():
    
31.     ost_instinfo = "name,sn,model,hba\n"
    
32.     _ocenStore_cmd1 = 'license show sys'
    
33.     ocenStore_output1 = ssh2(ip,username,passwd,_ocenStore_cmd1)
    
34.     #print(ocenStore_output1)
    
35.     ocenStore_output = ocenStore_output1.split("\n")
    
36.     for ost_item in (ocenStore_output):
    
37.         ost_inst = ost_item.rstrip().split()
    
38.         ost_inst_len = len(ost_inst)
    
39.         if( ost_inst_len == 5 and ost_inst[1] == "Serial"):
    
40.             ocenstor_sn = ost_inst[3]
    
41.         if(ost_inst_len == 4 and ost_inst[1] == "Model:"):
    
42.             ocenstor_model = ost_inst[2]
    
43.     #print(ocenstor_sn)
    
44.     #print(ocenstor_model)
    
45.     ######OceanStore HBA and array name####
    
46.     store_file=csv_path+ip+"_array.csv"
    
47.     _ocenStore_cmd2_hba = 'storage hba'
    
48.     _ocenStore_output2 =ssh2(ip,username,passwd,_ocenStore_cmd2_hba)
    
49.     _ocenStore_output = _ocenStore_output2.split("\n")
    
50.     for ost_hba_item in (_ocenStore_output):
    
51.         ost_hba_inst = ost_hba_item.rstrip().split()
    
52.         if(len(ost_hba_inst)>=2 and ost_hba_inst[0] != "Node" and ost_hba_inst[0] != "----"):
    
53.             #print(ost_hba_inst)
    
54.             ost_inst_hbainfo = ''.join(ost_hba_inst[1:])
    
55.             ost_instinfo=ost_instinfo+ip+","+ost_hba_inst[0]+","+ocenstor_sn+","+ocenstor_model+","+ost_inst_hbainfo.replace(",","_")+"\n"
    
56.     #print(ost_instinfo)
    
57.     _wr_csv(ost_instinfo,store_file)
    
58. 
    
59. 
    
60. def _ocenStoreDiskDetail():
    
61.     disk_info="name,pool,size,sn\n"
    
62.     disk_file=csv_path+ip+"_disk.csv"
    
63.     disk_cmd = 'storage disk list detail'
    
64.     disk_output1 = ssh2(ip,username,passwd,disk_cmd)
    
65.     disk_output = disk_output1.split("\n")
    
66.     for disk_item in (disk_output):
    
67.         disk_inst = disk_item.rstrip().split()
    
68.         if(len(disk_inst)>1 and disk_inst[0] != "Disk" and disk_inst[0] != "===="):
    
69.             disk_info = disk_info +disk_inst[0]+","+disk_inst[1]+","+disk_inst[3]+","+disk_inst[6]+"\n"
    
70.     #print(disk_info)
    
71.     _wr_csv(disk_info,disk_file)
    
72. 
    
73. def _ocenStoreLun():
    
74.     lun_info="name,size,raid,poolname\n"
    
75.     lun_file=csv_path+ip+"_lun.csv"
    
76.     lun_cmd = 'storage fs list'
    
77.     lun_output1 = ssh2(ip,username,passwd,lun_cmd)
    
78.     lun_output = lun_output1.split("\n")
    
79.     for lun_item in (lun_output):
    
80.         lun_inst = lun_item.rstrip().split()
    
81.         if(len(lun_inst)==11 and lun_inst[0] != "FS" and lun_inst[0] != "===="):
    
82.             lun_info = lun_info+lun_inst[0]+","+lun_inst[2]+","+lun_inst[3]+","+lun_inst[10]+"\n"
    
83.     #print(lun_info)
    
84.     _wr_csv(lun_info,lun_file)
    
85.    
    
86. if __name__=='__main__':
    
87.     ip = '1.2.3.4'
    
88.     username = 'admin'
    
89.     passwd = '1234'
    
90.     csv_path="C:\\test\\test3\\remote\\OceanStore\\"
    
91.     _ocenStoreSys()
    
92. 
    
93.     #Get disk info and reletive pool name
    
94.     _ocenStoreDiskDetail()
    
95. 
    
96.     #get lun info
    
97.     _ocenStoreLun()

复制代码

rdcwayx

能给个代码备注吗？