From: Dan White Date: Fri, 24 Feb 2012 21:18:50 +0000 (-0600) Subject: fpfixup.py: more robust detection X-Git-Url: http://git.whiteaudio.com/gitweb/?a=commitdiff_plain;h=ff2f88437d222c8473c041fd5ac28e9d080ae491;p=waeda-fp.git fpfixup.py: more robust detection --- diff --git a/fpfixup.py b/fpfixup.py index c77184a..b20ca65 100755 --- a/fpfixup.py +++ b/fpfixup.py @@ -14,14 +14,18 @@ for fpath in sys.argv[1:]: fp = open(fpath) lines = fp.readlines() fp.close() - if lines[0][0:7] == 'Element': - fields = lines[0].split() + first = lines[0] + if first[0:7] == 'Element': + istart = first.find('[') + iend = first.find(']') + fields = first[istart:iend+1].split() if fields[1] == fname or fields[1] == '"'+fname+'"': print 'Not touching:', fpath else: fields[1] = '"%s"' % fname #put back the split() trailing newline - lines[0] = ' '.join(fields) + '\n' + newfirst = 'Element' + ' '.join(fields) + ']\n' + lines[0] = newfirst print 'Rewriting:', fpath open(fpath, 'w').write(''.join(lines))