有关Python获取用户输入选项

admin

还是上代码吧：

1. import sys
   
2. #====================
   
3. def main(argv):
   
4.     #debug:
   
5.     print('Len(argv)=' + str(len(argv)) + '; argv[0]=' + argv[0])
   
6.     if len(argv) < 3:
   
7.         if len(argv) == 1 :
   
8.             print("Lack arguments. Not run. Notice: Place the data-file in /home/cuibe/work/replace/, and invoke it explicitly.")
   
9.             return -1
   
10.         elif len(argv) == 2 and (argv[1] == '--help' or argv[1] == '-h'):
    
11.             print("\nReplacer.py [-t|--exec|-h|--help] datafile \n-h --help   get the help notice.\n--exec   do real copy\n-t   do not real copy,test and verbos it only. ")
    
12.             return -1
    
13.         elif len(argv) == 2:
    
14.             print("Bad or lack option. Please reference help to give correct one.")
    
15.             return -1
    
16.     else:
    
17.         if argv[1] == '--exec':
    
18.             notTest = True
    
19.         elif argv[1] == '-t':
    
20.             notTest = False
    
21.         else:
    
22.             print("You provides wrong option.Try again.")
    
23.             return -1
    
24.     try:
    
25.         file = str(argv[2])
    
26.         filePath = "./" + file
    
27. 
    
28.         define_outfile = False
    
29.         if len(argv) > 3 :
    
30.             outfile = './' + str(argv[3])  # argv[3] must have if len > 3.
    
31.             define_outfile = True
    
32.         #debug:
    
33.         print(filePath, outfile)  # may raise exception : reference before assignment
    
34.         print(argv)
    
35.     except Exception, err:
    
36.         print('Make path error: ' + str(Exception) + str(err))
    
37. 
    
38.     result = -1
    
39.     try:
    
40.         if define_outfile == True:
    
41.             result = dosomething(, notTest, outfile)
    
42.         else:
    
43.             result = printAttr(entities, notTest)
    
44.     except Exception, err:
    
45.         print('Changing error: ' + str(Exception) + str(err))
    
46.         traceback.print_exc()
    
47.    
    
48.     print("------------------------------")
    
49.     if result == 0:
    
50.         print("Done.")
    
51.         return 0
    
52.     else:
    
53.         print("Proceed with error, check logs.")
    
54.         return -1
    
55. 
    
56. 
    
57. if __name__ == '__main__' : main(sys.argv)
    

复制代码

另一种代码：

1. #====================
   
2. def main(argv):
   
3.         #debug:
   
4.         #print('Len(argv)=' + str(len(argv)) + '; argv[0]=' + argv[0], end='')
   
5.         #end
   
6.         arg_dict = dict()
   
7.         arg_dict["k_words"] = []
   
8.         length = len(argv)
   
9. 
   
10.         if len(argv) >= 2 and argv[1][0] != '-':
    
11.                 try:
    
12.                         arg_dict['k_num'] = int(argv[1])
    
13.                         length -= 1
    
14.                         #debug:
    
15.                         #print("; argv[1]=" + argv[1])
    
16.                         #print('length=', length, end='')
    
17.                         #end
    
18.                         
    
19.                 except Exception as err:
    
20.                         # asset  Todo:  Write to log file.
    
21.                         #print("\nConver argv[1] to int error:", str(Exception), str(err))
    
22.                         # end
    
23.                         pass
    
24.         for arg in argv[len(argv)-length+1:]:  # minus first num arg if has. Plus arg[0].
    
25.         # Switch option
    
26.                
    
27.                 if (arg == '-H' or arg == '--Hash256'):
    
28.                         # arg = '-H'  # Flush --Hash256
    
29.                         
    
30.                         if (len(arg_dict["k_words"]) > 0) and not in_all_words("-H", "--Hash256", arg_dict):
    
31.                                 print("\n-H can not occur after words. Quit.\n")
    
32.                                 return -1
    
33.                         if "k_H" in arg_dict:  # And arg_dict['k_H']==Ture;
    
34.                                 arg_dict["k_words"].append(arg)
    
35.                         else:
    
36.                                 arg_dict["k_H"] = True
    
37.                         length -= 1
    
38.                         continue
    
39.                 elif (arg == '-R'):
    
40.                         if len(arg_dict["k_words"]) > 0 and not all_words("-R",arg_dict):
    
41.                                 print("\n-R can not occur after words. Quit.\n")
    
42.                                 return -1
    
43.                         if "k_R" in arg_dict:  # And arg_dict['k_R']==Ture;
    
44.                                 arg_dict["k_words"].append(arg)
    
45.                         else:
    
46.                                 arg_dict['k_R'] = True
    
47.                         length -= 1
    
48.                         continue
    
49.                 elif (arg == '--help' or arg == '-h'):
    
50.                         arg_dict["k_help"] = True
    
51.                         def validate_help():
    
52.                                 if len(argv) > 2:  # Notice here is not length.
    
53.                                         print("\nMiss using the help: --help (-h) do not occur with other arguments.\n")
    
54.                                         return -1
    
55.                                 else:
    
56.                                         return 0
    
57.                         if validate_help() == 0:
    
58.                                 break  # Jump out of loop.
    
59.                         else:
    
60.                                 return -1
    
61.                 else:
    
62.                         # "(-M) or (R)"
    
63.                         arg_dict['k_words'].append(arg)
    
64.                         length -= 1
    
65. 
    
66. 
    
67. 
    
68.         if length > 1:  # argv[0] is programme itself. So length of left is larger than 1 (not 0).
    
69.                 for word in argv[-(length-1):]:  # Reason -1 ref up.
    
70.                         arg_dict['k_words'].append(word)  # validate len(arg_dict['k_words']) to make sure it has value.
    
71.         if "k_help" in arg_dict:
    
72.                 print("\nTo generate a more complex passwd. \
    
73.                                 \ngenpasswd.py ([num] [-R] [-H|--Hash256] [words])|[-h|--help] \
    
74.                                 \nnum  passwd length \
    
75.                                 \nwords  words that you want to include in, seperate by blank space. \
    
76.                                 \n-R shuffle the charactors. \
    
77.                                 \n-H --Hash256 progress hash layer. \
    
78.                                 \n-h --help   get the help notice. ")
    
79.                 return 0
    
80. 
    
81.         if len(argv) == 1 :  # Run main prog without arguments.
    
82.                 # print("Using default arguments. Notice: You can input length of passwd or other arg, see --help.")
    
83.                 genPasswd(20)
    
84.                 return 0
    
85.         #debug
    
86.         print("\narg_dict:", arg_dict, "len(argv):", len(argv), "length:", length)
    
87. 
    
88.         if 'k_words' in arg_dict:
    
89.                 addWords(arg_dict['k_words'])
    
90.         if 'k_help' in arg_dict:
    
91.                 print("arg_dict['k_help']", arg_dict['k_help'])  # Has return , not proceed this line.
    
92.         if 'k_R' in arg_dict:
    
93.                 print("arg_dict['k_R']", arg_dict['k_R'])
    
94.         if 'k_H' in arg_dict:
    
95.                 print("arg_dict['k_H']", arg_dict['k_H'])
    
96.         #end
    
97.         if 'k_num' in arg_dict:
    
98.                 print("argv[1]", argv[1])
    
99.                 #print("arg_dict['k_num']", arg_dict['k_num'])
    
100.                 genPasswd(arg_dict['k_num'])
     
101.         else:
     
102.                 genPasswd()
     
103.         return 0
     
104.         # try:
     
105.         #         file = str(argv[2])
     
106.         #         filePath = "/" + file
     
107.                
     
108.         #         #debug:
     
109.         #         print(filePath)
     
110.         #         print(argv)
     
111.         # except Exception, err:
     
112.         #         print('Make path error: ' + str(Exception) + str(err))
     
113.                
     
114.         # entities = []
     
115.         # try:
     
116.         #         entities = readEntries(filePath)
     
117.         # except Exception, err:
     
118.         #         print('Reading file error: ' + str(Exception) + str(err))
     
119.         # result = -1
     
120.         # if len(entities) > 0:
     
121.         #         try:
     
122.         #                 result = replaceThem(entities, notTest)
     
123.         #         except Exception, err:
     
124.         #                 print('Replacing error: ' + str(Exception) + str(err))
     
125.         #                 traceback.print_exc()
     
126.         # else:
     
127.         #         print("No file to copy, check your source-data file.")
     
128.         # print("------------------------------")
     
129.         # if result == 0:
     
130.         #         print("Done.")
     
131.         #         return 0
     
132.         # else:
     
133.         #         print("Proceed with error, check logs.")
     
134.         #         return -1
     
135. 
     
136. if __name__ == '__main__' : main(sys.argv)
     

复制代码