# -*- coding: utf-8 -*- # Copyright © Cloud Linux GmbH & Cloud Linux Software, Inc 2010-2019 All Rights Reserved # # Licensed under CLOUD LINUX LICENSE AGREEMENT # http://cloudlinux.com/docs/LICENSE.TXT from __future__ import print_function from __future__ import division from __future__ import absolute_import import csv import simplejson import sys from xml.sax.saxutils import escape as _escape from past.builtins import basestring # noqa from future.utils import iteritems HTML_ESCAPE_TABLE = { '"': """, "'": "'" } def escape_string(data): if isinstance(data, basestring): return _escape(data, HTML_ESCAPE_TABLE) elif isinstance(data, (tuple, list)): new_data = [] for value in data: new_data.append(escape_string(value)) return new_data elif isinstance(data, dict): new_dict = {} for k, v in iteritems(data): new_dict[k] = escape_string(v) return new_dict return data def validate_json_message(data): # copies message from 'details' to 'message' # in case there is no 'message' in json if not 'message' in data: data['message'] = data['details'] class clprint(object): def print_data(cls, fmt, data, escape=None): """ Dispatches data to corresponing routine for printing @param fmt: string @param data: dict """ dispatcher = { 'json': cls.print_json, 'perl': cls.print_perl, 'csv': cls.print_csv, 'text': cls.print_text } try: dispatcher[fmt](data, escape=escape) except KeyError: dispatcher['text'](data) print_data = classmethod(print_data) def print_diag(cls, fmt, data): """ Dispatches data to corresponing routine for printing @param fmt: string @param data: dict """ dispatcher = { 'json': cls.print_diag_json, 'perl': cls.print_diag_perl, 'csv': cls.print_diag_csv, 'text': cls.print_diag_text } try: dispatcher[fmt](data) except KeyError: dispatcher['text'](data) print_diag = classmethod(print_diag) def print_csv(data, escape=None): """ Prints data as comma separated values @param data: dict """ csv_out = csv.writer(sys.stdout, quoting=csv.QUOTE_ALL) for opt in sorted(data.keys()): flat_array = [opt] for key, value in iteritems(data[opt]): if escape: value = escape_string(value) flat_array.extend([key, value]) csv_out.writerow(flat_array) print_csv = staticmethod(print_csv) def print_diag_csv(data): """ Prints diagnostic messages as comma separated values @param data: dict """ validate_json_message(data) csv_out = csv.writer(sys.stdout, quoting=csv.QUOTE_ALL) csv_out.writerow([data['status'], data['message']]) print_diag_csv = staticmethod(print_diag_csv) def print_json(data, escape=None): """ Prints data as JSON @param data: dict """ if data: if escape: data = escape_string(data) print(simplejson.dumps({ 'status': 'OK', 'data': data})) else: print(simplejson.dumps({'status': 'OK'})) print_json = staticmethod(print_json) def print_diag_json(data): """ Prints diagnostic messages as JSON @param data: dict """ print(simplejson.dumps(data)) print_diag_json = staticmethod(print_diag_json) def print_text(data, escape=None): """ Prints data as plain text @param data: dict """ for opt in sorted(data.keys()): print("TITLE:%s" % (opt,)) for key, v in iteritems(data[opt]): if escape: v = escape_string(v) print('%s:%s' % (key.upper(), v)) print('') print_text = staticmethod(print_text) @staticmethod def print_diag_text(data): """ Prints diagnostic messages as plain text @param data: dict """ validate_json_message(data) print("%s:%s" % (data['status'], data['message']), file=sys.stderr) if data.get('details'): print("Details:", file=sys.stderr) print(data.get('details', '') % data.get('context', {}), file=sys.stderr) def print_diag_perl(data): """ Prints diagnostic messages as perl data structure @param data: dict """ validate_json_message(data) print("{status=>%s,message=>%s}" % (data['status'], data['message'])) print_diag_perl = staticmethod(print_diag_perl) def print_perl(data, escape=None): """ Prints data as perl data structure @param data: dict """ out = [] for opt in sorted(data.keys()): structure = [] structure.append("title=>'%s'" % (opt, )) for k, v in iteritems(data[opt]): if escape: v = escape_string(v) structure.append("%s=>'%s'" % (k, v)) out.append("{%s}" % (','.join(structure))) print('[%s]' % (','.join(out),)) print_perl = staticmethod(print_perl)