upconvert.py

#!/usr/bin/env python
""" A universal hardware design file format converter using 
Upverter's Open JSON Interchange Format """

# upconvert.py - A universal hardware design file format converter using
# Upverter's Open JSON Interchange Format
# (http://upverter.com/resources/open-json-format/)
#
# Authors:
#   Alex Ray ajray@ncsu.edu
#   Upverter support@upverter.com
#
# Usage example:
#   ./upconvert.py -i test.upv -o test.json 


import os, re, copy, json
import parser.openjson, parser.kicad, parser.eaglexml
import writer.openjson, writer.kicad
from argparse import ArgumentParser

PARSERS = {
    'openjson': parser.openjson.JSON,
    'kicad': parser.kicad.KiCAD,
    'eaglexml': parser.eaglexml.EagleXML,
}

WRITERS = {
    'openjson': writer.openjson.JSON,
    'kicad': writer.kicad.KiCAD
}


def parse(in_file, in_format='openjson'):
    """ Parse the given input file using the in_format """

    try:
        p = PARSERS[in_format]()
    except KeyError:
        print "ERROR: Unsupported input type:", in_format
        exit(1)
    return p.parse(in_file)


def write(dsgn, out_file, out_format='openjson'):
    """ Write the converted input file to the out_format """

    try:
        w = WRITERS[out_format]()
    except KeyError:
        print "ERROR: Unsupported output type:", out_format
        exit(1)
    return w.write(dsgn, out_file)


if __name__ == "__main__":
    parser = ArgumentParser()
    parser.add_argument("-i", "--input", dest="inputfile",
            help="read INPUT file in", metavar="INPUT")
    parser.add_argument("-f", "--from", dest="inputtype",
            help="read input file as TYPE", metavar="TYPE",
            default="openjson")
    parser.add_argument("-o", "--output", dest="outputfile",
            help="write OUTPUT file out", metavar="OUTPUT")
    parser.add_argument("-t", "--to", dest="outputtype",
            help="write output file as TYPE", metavar="TYPE",
            default="openjson")

    args = parser.parse_args()
    inputtype = args.inputtype
    outputtype = args.outputtype
    inputfile = args.inputfile
    outputfile = args.outputfile
    if None == inputfile:
        args.print_help()
        exit(1)

    # parse and export the data
    design = parse(inputfile, inputtype)
    if design is not None: # we got a good result
        write(design, outputfile, outputtype)
    else: # parse returned None -> something went wrong
        print "Output cancelled due to previous errors."
        exit(1)