json_to_xml.py

#!/usr/bin/env python
#  vim:ts=4:sts=4:sw=4:et
#
#  Author: Hari Sekhon
#  Date: 2016-01-15 00:07:09 +0000 (Fri, 15 Jan 2016)
#
#  https://github.com/harisekhon/pytools
#
#  License: GNU GPL version 2 (this file only), rest of this repo is licensed as per the adjacent LICENSE file
#
#  If you're using my code you're welcome to connect with me on LinkedIn and optionally send me feedback
#  to help improve or steer this or other code I publish
#
#  http://www.linkedin.com/in/harisekhon
#

"""

Tool to convert JSON to XML

Reads any given files as JSON and prints the equivalent XML to stdout for piping or redirecting to a file.

Directories if given are detected and recursed, processing all files in the directory tree ending in a .json suffix.

Works like a standard unix filter program - if no files are passed as arguments or '-' is passed then reads from
standard input.

"""

from __future__ import absolute_import
from __future__ import division
from __future__ import print_function
#from __future__ import unicode_literals

import json
import os
import re
import sys
# This library is GNU GPL licensed so this program is also GNU GPL licensed
import dicttoxml
#import xml.etree.ElementTree as ET
libdir = os.path.abspath(os.path.join(os.path.dirname(__file__), 'pylib'))
sys.path.append(libdir)
try:
    # pylint: disable=wrong-import-position
    from harisekhon.utils import die, ERRORS, log, log_option
    from harisekhon import CLI
except ImportError as _:
    print('module import failed: %s' % _, file=sys.stderr)
    print("Did you remember to build the project by running 'make'?", file=sys.stderr)
    print("Alternatively perhaps you tried to copy this program out without it's adjacent libraries?", file=sys.stderr)
    sys.exit(4)

__author__ = 'Hari Sekhon'
__version__ = '0.1'


class JsonToXml(CLI):

    def __init__(self):
        # Python 2.x
        super(JsonToXml, self).__init__()
        # Python 3.x
        # super().__init__()
        self.re_json_suffix = re.compile(r'.*\.json$', re.I)

    @staticmethod
    def json_to_xml(content, filepath=None):
        try:
            _ = json.loads(content)
        except (KeyError, ValueError) as _:
            file_detail = ''
            if filepath is not None:
                file_detail = ' in file \'{0}\''.format(filepath)
            die("Failed to parse JSON{0}: {1}".format(file_detail, _))
        return dicttoxml.dicttoxml(_)

    def run(self):
        if not self.args:
            self.args.append('-')
        for arg in self.args:
            if arg == '-':
                continue
            if not os.path.exists(arg):
                print("'%s' not found" % arg)
                sys.exit(ERRORS['WARNING'])
            if os.path.isfile(arg):
                log_option('file', arg)
            elif os.path.isdir(arg):
                log_option('directory', arg)
            else:
                die("path '%s' could not be determined as either a file or directory" % arg)
        for arg in self.args:
            self.process_path(arg)

    def process_path(self, path):
        if path == '-' or os.path.isfile(path):
            self.process_file(path)
        elif os.path.isdir(path):
            for root, _, files in os.walk(path):
                for filename in files:
                    filepath = os.path.join(root, filename)
                    if self.re_json_suffix.match(filepath):
                        self.process_file(filepath)
        else:
            die("failed to determine if path '%s' is a file or directory" % path)

    def process_file(self, filepath):
        log.debug('processing filepath \'%s\'', filepath)
        if filepath == '-':
            filepath = '<STDIN>'
        if filepath == '<STDIN>':
            self.json_to_xml(sys.stdin.read())
        else:
            with open(filepath) as _:
                content = _.read()
                print(self.json_to_xml(content, filepath=filepath))


if __name__ == '__main__':
    JsonToXml().main()