I currently have a script, which uses file globbing via the sys.argv variable like this:
if len(sys.argv) > 1:
for filename in sys.argv[1:]:
This works great for processing a bunch of files; however, I would like to use this with the argparse module as well. So, I would like my program to be able to handle something like the following:
foo@bar:~$ myScript.py --filter=xyz *.avi
Has anyone tried to do this, or have some pointers on how to proceed?
If I got you correctly, your question is about passing a list of files together with a few flag or optional parameters to the command. If I got you right, then you just must leverage the argument settings in argparse:
File p.py
import argparse
parser = argparse.ArgumentParser(description='SO test.')
parser.add_argument('--doh', action='store_true')
parser.add_argument('files', nargs='*') # This is it!!
args = parser.parse_args()
print(args.doh)
print(args.files)
The commented line above inform the parser to expect an undefined number >= 0 (nargs ='*') of positional arguments.
Running the script from the command line gives these outputs:
$ ./p.py --doh *.py
True
['p2.py', 'p.py']
$ ./p.py *.py
False
['p2.py', 'p.py']
$ ./p.py p.py
False
['p.py']
$ ./p.py
False
[]
Observe how the files will be in a list regardless of them being several or just one.
HTH!
parser.add_argument('-b', '--blacklist', nargs='*' and I'd like to pass *.mp3 into the program from the shell. Also, blacklist is used in os.walk, so the actual mp3 files won't be in working directory.
*] wildcard but without letting the shell making the expansion for you? If that's the case, you have to quote the argument: ./p.py --doh '*.py'. From within the p.py script you will have to use fnmatch to do the expansion for you. Or did I misunderstand you?for filename in [f for f in filenames if f not in blacklist]: within an os.walk I'd like to be able to pass *.mp3 into -b and have all mp3 files be ignored in that for loop.os.walk loop then (or am I misunderstanding you again?). From what I got from your comment above I would simply use the builtin function filter on the list returned by os.walk. Something like filter(lambda fn: fn[-3:] != 'mp3', list_of_files) (untested). HTH, but in case it doesn't, you're better opening your own question! :)Alternatively you may use both in the following way:
import sys
import argparse
if __name__ == '__main__':
parser = argparse.ArgumentParser()
parser.add_argument("-v", "--verbose", help="increase verbosity", action="store_true")
args, unknown = parser.parse_known_args()
for file in sys.argv:
if not file.startswith("-"):
print(file)
However this will work only for standalone parameters, otherwise the argument values would be treated as file arguments (unless you'll not separate them with space or you'll improve the code further more).
len(sys.argv)stuff and just letargparseinteract withsys.argv, the functionality is in there.