#!/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))
Mini Shell