1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
|
#!/usr/bin/env python
# -*- coding: utf-8 -*-
from os import path
from xml.etree import ElementTree as etree
import argparse
def update_text(tree, name, value=None, func=None):
e = tree.find(name)
if value is not None:
e.text = value
elif func is not None:
e.text = func(e.text)
def update(fname, args):
tree = etree.parse(fname)
update_text(tree, 'InputFilename', func=path.basename)
update_text(tree, 'OutputFilename', func=path.basename)
if args.saturation is not None:
update_text(tree, 'Saturation', value=args.saturation)
tree.write(fname, encoding='utf-8', xml_declaration=True)
def setup_parser():
parser = argparse.ArgumentParser(description='Remove path from input and output file names.')
parser.add_argument('--saturation', metavar='SAT',
help='Adjust the color saturation. Range 0.00 to 8.00. Default 1.0, use 0 for black & white output.')
parser.add_argument('files', metavar='FILES', nargs=argparse.REMAINDER,
help='Input UFRaw ID-files.')
return parser
def main():
parser = setup_parser()
args = parser.parse_args()
if len(args.files) == 0:
parser.print_help()
for f in filter(path.isfile, args.files):
update(f, args)
if __name__ == '__main__':
main()
|
