Python аргумент командной строки, который принимает в каталоге и выводит список файлов

Question

Я хочу:

Как говорит название,

запустить что-то такое, как:

python3 program.py [~/главная/мне/Документы/*. TXT]

и я хотел бы получить

[ 'text1.txt', 'text2.txt', ..., 'textn.txt']

Solutions Я нашел в Интернете:

Много решений я нашел в Интернете с помощью включенных fileinput.input() и sys.argv. Проблема в том, что я неправильно реализую эти решения (я полный noob на Python).

Вот наиболее перспективное решение:

Source

import fileinput 
for line in fileinput.input(): 
    process(line) 

, но он продолжает говорить мне, что «процесс не определен» - - Я до сих пор найти решение.

Другим решением является sys.argv [0]. но он продолжает говорить мне, что индекс списка выходит за пределы диапазона.

EDIT: This Решение не переходит к аргументу командной строки для python. Это то, с чем я в основном испытываю проблемы.

Answer 1

Это работает для меня:

import os 
import sys 

def getfilenames(path): 
    ftype = path.split(".")[-1] 
    folder = path.split("*")[0] 
    return [i for i in os.listdir(folder) if ftype in i.split(".")[-1]] 

print(getfilenames(sys.argv[1])) 

указать путь в виде строки:

python example.py "/path/to/file/*.type" 

Answer 2

вы можете комбинировать ответы, представленные в how to list all files of a directory и command line arguments in python , чтобы создать что-то вроде этого:

import sys 
from os import walk 

path = sys.argv[1] 
files = [] 
for (dirpath, dirnames, filenames) in walk(path): 
    files.extend(filenames) 
print (files) 

и r ип это с

python3 listdir.py $PWD