#
# Metrix++, Copyright 2009-2013, Metrix++ Project
# Link: http://metrixplusplus.sourceforge.net
#
# This file is a part of Metrix++ Tool.
#
# Metrix++ 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, version 3 of the License.
#
# Metrix++ 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 Metrix++. If not, see .
#
import mpp.api
import mpp.utils
import csv
class Plugin(mpp.api.Plugin, mpp.api.IConfigurable, mpp.api.IRunable):
def declare_configuration(self, parser):
parser.add_option("--format", "--ft", default='csv', choices=['csv', 'xml'], help="Format of the output data. "
"Possible values are 'xml' and 'csv' [default: %default]")
def configure(self, options):
self.out_format = options.__dict__['format']
def run(self, args):
loader_prev = self.get_plugin_loader().get_plugin('mpp.dbf').get_loader_prev()
loader = self.get_plugin_loader().get_plugin('mpp.dbf').get_loader()
paths = None
if len(args) == 0:
paths = [""]
else:
paths = args
exit_code = export_to_stdout(self.out_format, paths, loader, loader_prev)
return exit_code
def export_to_stdout(out_format, paths, loader, loader_prev):
class StdoutWriter(object):
def write(self, *args, **kwargs):
print args[0],
exit_code = 0
columnNames = ["file", "region", ]
columns = []
for name in loader.iterate_namespace_names():
namespace = loader.get_namespace(name)
for field in namespace.iterate_field_names():
columns.append((name, field, namespace.are_regions_supported()))
columnNames.append(name + ":" + field)
writer = StdoutWriter()
csvWriter = csv.writer(writer)
csvWriter.writerow(columnNames)
if out_format == 'xml':
print "\n"
elif out_format == 'csv':
print "CSV"
else:
assert False, "Unknown output format " + out_format
for path in paths:
path = mpp.utils.preprocess_path(path)
files = loader.iterate_file_data(path)
if files != None:
for file_data in files:
for reg in file_data.iterate_regions():
per_reg_data = []
for column in columns:
per_reg_data.append(reg.get_data(column[0], column[1]))
csvWriter.writerow([file_data.get_path(), reg.get_name()] + per_reg_data)
per_file_data = []
for column in columns:
per_file_data.append(file_data.get_data(column[0], column[1]))
csvWriter.writerow([file_data.get_path(), None] + per_file_data)
else:
mpp.utils.report_bad_path(path)
exit_code += 1
if out_format == 'xml':
print "XML"
return 0