#!/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 <[email protected]>
#
# 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
from apport import unicode_gettext as _, fatal

import problem_report


def help():
print(_('Usage: %s <report> <target directory>') % sys.argv[0])

if sys.version[0] < '3':
t_str = unicode
else:
t_str = str

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))

pr = problem_report.ProblemReport()
if report == '-':
pr.load(sys.stdin)
else:
try:
with open(report, 'rb') as f:
pr.load(f)
except IOError as e:
fatal(str(e))
for k in pr:
with open(os.path.join(dir, k), 'wb') as f:
if type(pr[k]) == t_str:
f.write(pr[k].encode('UTF-8'))
else:
f.write(pr[k])