80 lines
2.0 KiB
Python
Executable File
80 lines
2.0 KiB
Python
Executable File
#!/usr/bin/python3
|
|
|
|
# Extract the fields of a problem report into separate files into a new or
|
|
# empty directory.
|
|
#
|
|
# Copyright (c) 2006 Canonical Ltd.
|
|
# Author: Martin Pitt <martin.pitt@ubuntu.com>
|
|
#
|
|
# This program is free software; you can redistribute it and/or modify it
|
|
# under the terms of the GNU General Public License as published by the
|
|
# Free Software Foundation; either version 2 of the License, or (at your
|
|
# option) any later version. See http://www.gnu.org/copyleft/gpl.html for
|
|
# the full text of the license.
|
|
|
|
import sys, os, os.path, gettext, gzip
|
|
from apport import unicode_gettext as _, fatal
|
|
|
|
import problem_report
|
|
|
|
|
|
def help():
|
|
print(_('Usage: %s <report> <target directory>') % sys.argv[0])
|
|
|
|
|
|
gettext.textdomain('apport')
|
|
|
|
if len(sys.argv) >= 2 and sys.argv[1] == '--help':
|
|
help()
|
|
sys.exit(0)
|
|
|
|
if len(sys.argv) != 3:
|
|
help()
|
|
sys.exit(1)
|
|
|
|
report = sys.argv[1]
|
|
dir = sys.argv[2]
|
|
|
|
# ensure that the directory does not yet exist or is empty
|
|
try:
|
|
if os.path.isdir(dir):
|
|
if os.listdir(dir):
|
|
fatal(_('Destination directory exists and is not empty.'))
|
|
else:
|
|
os.mkdir(dir)
|
|
except OSError as e:
|
|
fatal(str(e))
|
|
|
|
bin_keys = []
|
|
pr = problem_report.ProblemReport()
|
|
if report == '-':
|
|
pr.load(sys.stdin, binary=False)
|
|
else:
|
|
try:
|
|
if report.endswith('.gz'):
|
|
with gzip.open(report, 'rb') as f:
|
|
pr.load(f, binary=False)
|
|
else:
|
|
with open(report, 'rb') as f:
|
|
pr.load(f, binary=False)
|
|
except IOError as e:
|
|
fatal(str(e))
|
|
for k in pr:
|
|
if pr[k] is None:
|
|
bin_keys.append(k)
|
|
continue
|
|
with open(os.path.join(dir, k), 'wb') as f:
|
|
if type(pr[k]) == str:
|
|
f.write(pr[k].encode('UTF-8'))
|
|
else:
|
|
f.write(pr[k])
|
|
try:
|
|
if report.endswith('.gz'):
|
|
with gzip.open(report, 'rb') as f:
|
|
pr.extract_keys(f, bin_keys, dir)
|
|
else:
|
|
with open(report, 'rb') as f:
|
|
pr.extract_keys(f, bin_keys, dir)
|
|
except IOError as e:
|
|
fatal(str(e))
|