"""JSON interface class."""
import sys
from glances.globals import json_dumps
from glances.compat import PY3, listkeys
from glances.logger import logger
from glances.exports.glances_export import GlancesExport
class Export(GlancesExport):
"""This class manages the JSON export module."""
def __init__(self, config=None, args=None):
"""Init the JSON export IF."""
super(Export, self).__init__(config=config, args=args)
# JSON file name
self.json_filename = args.export_json_file
# Set the JSON output file
try:
if PY3:
self.json_file = open(self.json_filename, 'w')
self.json_file.close()
else:
self.json_file = open(self.json_filename, 'wb')
self.json_file.close()
except IOError as e:
logger.critical("Cannot create the JSON file: {}".format(e))
sys.exit(2)
logger.info("Exporting stats to file: {}".format(self.json_filename))
self.export_enable = True
# Buffer for dict of stats
self.buffer = {}
def exit(self):
"""Close the JSON file."""
logger.debug("Finalise export interface %s" % self.export_name)
self.json_file.close()
def export(self, name, columns, points):
"""Export the stats to the JSON file."""
# Check for completion of loop for all exports
if name == self.last_exported_list()[0] and self.buffer != {}:
# One whole loop has been completed
# Flush stats to file
logger.debug("Exporting stats ({}) to JSON file ({})".format(listkeys(self.buffer), self.json_filename))
# Export stats to JSON file
if PY3:
with open(self.json_filename, "w") as self.json_file:
self.json_file.write("{}\n".format(json_dumps(self.buffer)))
else:
with open(self.json_filename, "wb") as self.json_file:
self.json_file.write("{}\n".format(json_dumps(self.buffer)))
# Reset buffer
self.buffer = {}
# Add current stat to the buffer
self.buffer[name] = dict(zip(columns, points))
|