Передача аргументов с подстановочными знаками в скрипт Python
Я хочу сделать что-то вроде этого:
c:\data\> python myscript.py *.csv
и передать все файлы .csv в каталоге моему скрипту python (например, sys.argv содержит ["file1.csv", "file2.csv"] и т. д.)
Но sys.argv просто получает ["*.csv"], указывающий, что подстановочный знак не был расширен, поэтому это не работает.
Я чувствую, что есть простой способ сделать это, но не могу найти его в Google. Любые идеи?
Ответы 2
Вы можете использовать модуль glob, таким образом вы не будете зависеть от поведения конкретной оболочки (ну, вы по-прежнему зависите от оболочки, не расширяющей аргументы, но, по крайней мере, вы можете заставить это происходить в Unix, избегая подстановочных знаков :-)).
from glob import glob
filelist = glob('*.csv') #You can pass the sys.argv argument
Ну, вы также можете использовать os.walk, так что это не обязательно: P
@ Винко Врсалович: правда. os.walk кажется более громоздким, чем glob. Glob не требуется, но он идеально подходит для решения этой проблемы.
Спасибо, именно то, что мне было нужно.
Именно то, что я искал :)
Кто-нибудь знает, почему оболочки Windows не справляются с этим за вас?
@Ryan: Потому что утилиты DOS были созданы людьми, которые не знали UNIX, и решили позволить каждой утилите обрабатывать расширение вместо того, чтобы делать оболочку умнее. Ужасный дизайн, но это лишь один из многих :-)
В Unix оболочка расширяет символы подстановки, поэтому программы получают расширенный список имен файлов. Windows этого не делает: оболочка передает подстановочные знаки непосредственно программе, которая сама должна их расширить.
Винко прав: модуль glob выполняет свою работу:
import glob, sys
for arg in glob.glob(sys.argv[1]):
print "Arg:", arg
+1. Обратите внимание, что это также отлично работает, если sys.argv[1] - это полностью указанное имя файла, а не подстановочный знак. В этом случае glob.glob просто возвращает список, содержащий это самое имя файла.
также работает, если вы перенаправляете параметр в свой Python script script.py c:\path\*\subdir glob, генерирующий список, который я могу использовать позже. Отличный модуль!
Это не «может»; для окон это "обязательно".