def __init__(self, output, error_level):
# try:
# self.output = open(output, "w")
# except IOError:
# print('cannot open file')
# sys.exit(1)
self.output = output
self.filetype = 0
try:
output.write()
except AttributeError:
self.output = open(output, 'w')
self.filetype = 1
if error_level == 1 or error_level == 0:
self.error_level = error_level
else:
self.error_level = 2
self.pretty = 0
self.writer = YajlGen(beautify=False)
self.curr = "initial"
self.bugID = 1
self.metricID = 1
self.start = 0
self.metricSummaries = {}
self.bugSummaries = {}
def parse_start(self):
self.g = YajlGen(
beautify=self.beautify,
indent_string=self.indent_string,
)
def setPrettyPrint(self, pretty_enable):
self.pretty = pretty_enable
if self.pretty:
self.writer = YajlGen(beautify=True)
else:
self.writer = YajlGen(beautify=False)
