ya2/tools/set_diffuse.py

   1 from os import listdir
   2
   3
   4 def __set_diffuse(fpath):
   5     out_lines = []
   6     with open(fpath) as fin:
   7         for line in fin.readlines():
   8             out_lines = __process_line(line, out_lines)
   9     with open(fpath, 'w') as fin:
  10         list(map(lambda outl: fin.write(outl + '\n'), out_lines))
  11
  12
  13 def __process_line(line, out_lines):
  14     if not line.strip().startswith('<Scalar> diff'):
  15         new_lines = [line.rstrip()]
  16     else: new_lines = [line.split(' { ')[0] + ' { 1.000000 }']
  17     return out_lines + new_lines
  18
  19
  20 list(map(__set_diffuse,
  21          [fname for fname in listdir('.') if fname.endswith('.egg')]))