Python中获取文件名来实现选项菜单

Zss 发表于:

在写shell脚本时需要写获取到远程服务器上的某个目录下的所有软件版本,所以想着在python中其实也是很好实现的

思路是通过  os.listdir(‘  path ‘)  将获取到的所有的文件名,可以按照一定的规则先过滤,然后依次循环打印,在每个元素打印前

加上一个index,每循环一次则加一,也就生成了一个可以给用户选择的列表了,这种界面的交互是很好的

 

#coding:utf-8
import os

def Option():
    file_list = os.listdir('C:\Users\Z-one\Desktop\install_OnPremise')
    print('----------------------------------')
    print('列表选择:')
    index = 1
    for i in file_list:
        if 'install_OnPremise' in i:
            print('(%s).%s '%(index,i))
            index += 1
    print('----------------------------------')
    user_choice = raw_input('输入您的选择:')
    print(user_choice)
    
Option()
----------------------------------
列表选择:
(1).install_OnPremise_1.1.sh 
(2).install_OnPremise_1.3.tar.gz 
(3).install_OnPremise_1.4.tar.gz 
(4).install_OnPremise_1.5 
(5).install_OnPremise_1.5.tar.gz 
----------------------------------
输入您的选择:5
5