#! /usr/bin/env python3 # -*- coding: utf-8 -*- # filename: totalopenstation-cli-parser.py # Copyright 2008-2013 Stefano Costa # Copyright 2015-2016 Damien Gaignon # This file is part of Total Open Station. # Total Open Station 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 3 of the # License, or (at your option) any later version. # Total Open Station is distributed in the hope that it will be # useful, but WITHOUT ANY WARRANTY; without even the implied warranty # of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU # General Public License for more details. # You should have received a copy of the GNU General Public License # along with Total Open Station. If not, see # . import sys import os import gettext import importlib import logging from optparse import OptionParser import totalopenstation.formats import totalopenstation.output t = gettext.translation('totalopenstation', './locale', fallback=True) _ = t.gettext usage = _("Usage: %prog [option] arg1 [option] arg2 ...") parser = OptionParser(usage=usage) parser.add_option("-i", "--infile", action="store", type="string", dest="infile", help=_("Select input FILE (do not specify for stdin)"), metavar="FILE") parser.add_option("-o", "--outfile", action="store", type="string", dest="outfile", help=_("Select output FILE (do not specify for stdout)"), metavar="FILE") parser.add_option("-f", "--input-format", action="store", type="string", dest="informat", help=_("Select input FORMAT"), metavar="FORMAT") parser.add_option("--2d", action="store_true", dest="xy_only", help=_("Exclude Z coordinates, output only 2D data"), metavar="ONLY2D") parser.add_option("-t", "--output-format", action="store", type="string", dest="outformat", help=_("Select input FORMAT"), metavar="FORMAT") parser.add_option("-r", "--raw", action="store_true", dest="raw", help=_("Enhanced parsed file process")) parser.add_option( "--overwrite", action="store_true", dest="overwrite", default=False, help=_("Overwrite existing output file")) parser.add_option( "--list", action="store_true", dest="list", default=False, help=_("List the available input and output formats")) parser.add_option( "--log", action="store", dest="loglevel", default="WARNING", type="string", help=_("Minimum log level")) parser.add_option( "--logtofile", action="store_true", dest="logotfile", default=False, help=_("Log to file")) (options, args) = parser.parse_args() logger = logging.getLogger() logger.setLevel(options.loglevel.upper()) if options.logotfile: handler = logging.FileHandler("tops.log") else: handler = logging.StreamHandler() formatter = logging.Formatter('%(asctime)s -- %(filename)s -- %(levelname)s -- %(funcName)s -- %(message)s') handler.setFormatter(formatter) logger.addHandler(handler) def list_formats(): '''Print a list of the supported input and output formats.''' mod_string = _("List of supported input formats:\n%s\n") % ('-' * 30) for k, v in sorted(totalopenstation.formats.BUILTIN_INPUT_FORMATS.items()): mod_string += "%s%s\n" % (k.ljust(20), v[2]) mod_string += "\n\n" mod_string += _("List of supported output formats:\n%s\n") % ('-' * 30) for k, v in sorted(totalopenstation.output.BUILTIN_OUTPUT_FORMATS.items()): mod_string += "%s%s\n" % (k.ljust(20), v[2]) mod_string += "\n" return mod_string if options.list: sys.stdout.write(list_formats()) sys.exit() def exit_with_error(message): sys.exit(_("\nError:\n%(message)s\n\n%(formats)s") % {'message': message, 'formats': list_formats()}) if options.informat: try: inputclass = totalopenstation.formats.BUILTIN_INPUT_FORMATS[options.informat] except KeyError as message: exit_with_error(_('%s is not a valid input format') % message) else: if isinstance(inputclass, tuple): try: # builtin format parser mod, cls, name = inputclass inputclass = getattr(importlib.import_module('totalopenstation.formats.%s' % mod), cls) except ImportError as message: exit_with_error(message) else: sys.exit(_("Please specify an input format")) if options.outformat: try: outputclass = totalopenstation.output.BUILTIN_OUTPUT_FORMATS[options.outformat] except KeyError as message: exit_with_error(_('%s is not a valid output format') % message) else: if isinstance(outputclass, tuple): try: # builtin output builder mod, cls, name = outputclass outputclass = getattr(importlib.import_module('totalopenstation.output.%s' % mod), cls) except ImportError as message: exit_with_error(message) if options.infile: infile = open(options.infile, 'r').read() else: if sys.stdin.isatty(): sys.exit(_('No input data!')) else: infile = sys.stdin.read() def main(infile): '''After setting up all parameters, finally try to process input data.''' parsed_data = inputclass(infile) if options.raw: parsed_points = parsed_data.raw_line else: parsed_points = parsed_data.points # processing options if options.xy_only: for feature in parsed_points: geom_cls = getattr(totalopenstation.formats, feature.geometry.geom_type) try: feature.geometry = geom_cls(feature.geometry.x, feature.geometry.y) except: feature.geometry = geom_cls([(p.x, p.y) for p in feature.geometry.geoms]) output = outputclass(parsed_points) def write_to_file(outfile): e = open(outfile, 'w') e.write(output.process()) e.close() if options.outfile: if not os.path.exists(options.outfile): write_to_file(options.outfile) logger.info(_("Downloaded data saved to out file %s") % options.outfile) else: if options.overwrite: write_to_file(options.outfile) logger.info(_("Downloaded data saved to file %s,") % options.outfile) logger.info(_("Overwriting the existing file")) else: sys.exit(_("Specified output file already exists\n")) else: sys.stdout.write(output.process()) if __name__ == '__main__': main(infile)