Commit 46eb9884 authored by Julien Muchembled's avatar Julien Muchembled

BT: make genbt5list safer for reuse in another process

- stop using chdir
- make logging to stdout/stderr optional
- generateInformation: return a StringIO instead of writing to a temporary file
parent a82324ca
......@@ -34,9 +34,8 @@
import tarfile
import os
import sys
import tempfile
import shutil
import cgi
from cStringIO import StringIO
property_list = '''
title
......@@ -47,20 +46,10 @@ license
dependency_list
provision_list
copyright_list
'''.strip().splitlines()
'''.split()
bt_title_path = os.path.join('bt', 'title')
def info(message):
"""Print a message to stdout.
"""
sys.stdout.write(message)
def err(message):
"""Print a message to stderr.
"""
sys.stderr.write(message)
def readProperty(property_dict, property_name, property_file):
try:
text = property_file.read()
......@@ -96,64 +85,56 @@ def readBusinessTemplateDirectory(dir):
return property_dict
def generateInformation(fd):
os.write(fd, '<?xml version="1.0"?>\n')
os.write(fd, '<repository>\n')
for file in sorted(os.listdir(os.getcwd())):
if file.endswith('.bt5'):
info('Reading %s... ' % (file,))
def generateInformation(dir, info=id, err=None):
xml = StringIO()
xml.write('<?xml version="1.0"?>\n<repository>\n')
for name in sorted(os.listdir(dir)):
path = os.path.join(dir, name)
if name.endswith('.bt5'):
info('Reading %s... ' % name)
try:
tar = tarfile.open(file, 'r:gz')
tar = tarfile.open(path, 'r:gz')
except tarfile.TarError:
err('An error happened in %s; skipping\n' % (file,))
if err:
err('An error happened in %s; skipping\n' % name)
continue
raise
try:
property_dict = readBusinessTemplate(tar)
finally:
tar.close()
elif os.path.isfile(os.path.join(file, bt_title_path)):
info('Reading Directory %s... ' % (file,))
property_dict = readBusinessTemplateDirectory(file)
elif os.path.isfile(os.path.join(path, bt_title_path)):
info('Reading Directory %s... ' % name)
property_dict = readBusinessTemplateDirectory(path)
else:
continue
os.write(fd, ' <template id="%s">\n' % (file,))
for property_id, property_value in sorted(property_dict.items()):
if type(property_value) is str:
os.write(fd, ' <%s>%s</%s>\n' % (
property_id, cgi.escape(property_value), property_id))
else:
for value in property_value:
os.write(fd, ' <%s>%s</%s>\n' % (
property_id, cgi.escape(value), property_id))
os.write(fd, ' </template>\n')
xml.write(' <template id="%s">\n' % name)
for k, v in sorted(property_dict.iteritems()):
for v in (v,) if type(v) is str else v:
xml.write(' <%s>%s</%s>\n' % (k, cgi.escape(v), k))
xml.write(' </template>\n')
info('done\n')
os.write(fd, '</repository>\n')
xml.write('</repository>\n')
return xml
def main():
if len(sys.argv) < 2:
dir_list = ['.']
else:
dir_list = sys.argv[1:]
def main(dir_list=None, **kw):
if dir_list is None:
kw.setdefault('info', sys.stdout.write)
kw.setdefault('err', sys.stderr.write)
dir_list = sys.argv[1:] or '.'
cur_umask = os.umask(0666)
os.umask(cur_umask)
cwd = os.getcwd()
for d in dir_list:
os.chdir(d)
try:
fd, path = tempfile.mkstemp(dir='.')
bt5list = generateInformation(d, **kw).getvalue()
d = os.path.join(d, 'bt5list.new')
try:
generateInformation(fd)
os.fchmod(fd, 0666 & ~cur_umask)
os.rename(path, 'bt5list')
except:
os.remove(path)
raise
finally:
os.close(fd)
with open(d, 'wb') as f:
f.write(bt5list)
os.rename(d, d[:-4])
finally:
os.chdir(cwd)
try:
os.remove(d)
except OSError:
pass
if __name__ == "__main__":
main()
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment