element_id = None

command = 'normal'

segment_value_id = None

process_dataset_ids = None

initial_measurement_time = None

final_measurement_time = None

pivot_id = None

chart_id = None

update_commands = ['normal', 'full_gen', 'soft_gen', 'composite', 'delete_data']

delete_commands = ['delete_all', 'delete_segment', 'delete_pivot', 'delete_chart']

_elements = []

def __init__(self) :

logger = MLogger('report generator')

self._logger = logger.get_logger()

try:

self._db = DBManager().get_query()

except MySQLdb.Error, message:

self._logger.error("Report generator. Init mysql db manager error - %s" % message)

log_traceback(self._logger, traceback)

print simplejson.dumps({'status': 'ERROR', 'message': '%s' % message})

sys.exit()

except Exception, exc:

self._logger.error("Report generator. Init mysql db manager error - %s." % exc)

log_traceback(self._logger, traceback)

print simplejson.dumps({'status': 'ERROR', 'message': 'report update failed. %s' % exc})

sys.exit()

def run(self):

"""

Process command

"""

self._logger.info('Generator run: %s' % ' '.join(sys.argv))

if self.element_id:

# process specified report

if self.command in self.delete_commands:

# process delete command for specified report

report = ReportPurge(self.element_id, self._logger)

message = ''

try:

if self.command == 'delete_all':

# delete whole report

report.delete_all()

message = 'Report %s deleted successfully.' % self.element_id

elif self.command == 'delete_segment':

# delete segment

if self.segment_value_id is None:

raise Exception('segment is not specified')

report.delete_segment(self.segment_value_id)

message = 'Segment %s of report %s deleted successfully.' % (self.segment_value_id, self.element_id)

elif self.command == 'delete_pivot':

# delete pivot

if self.pivot_id is None:

raise Exception('pivot is not specified')

report.delete_pivot(self.pivot_id)

message = 'Pivot %s of report %s deleted successfully.' % (self.pivot_id, self.element_id)

elif self.command == 'delete_chart':

# delete chart

if self.chart_id is None:

raise Exception('chart is not specified')

report.delete_chart(self.chart_id)

message = 'Chart %s of report %s deleted successfully.' % (self.chart_id, self.element_id)

except MySQLdb.Error, message:

# process Mysql error

self._logger.error("Report generator. Report %s update failed. SQL error %s" % (self.element_id, message))

log_traceback(self._logger, traceback)

print simplejson.dumps({'status': 'ERROR', 'message': 'SQL error %s' % message})

sys.exit()

except Exception, exc:

# process any other error

self._logger.error("Report generator. Report %s update failed. Exception %s" % (self.element_id, exc))

log_traceback(self._logger, traceback)

print simplejson.dumps({'status': 'ERROR', 'message': 'report %s update failed. %s' % (self.element_id, exc)})

sys.exit()

print simplejson.dumps({'status': 'OK', 'message': message})

else:

# process update command for specified report

self._run_one()

else:

# process update command for all reports

self._run_all()

def _run_one(self):

"""

fetch specified report

"""

try:

res = self._db.Query("""SELECT dashboard_element.element_id

FROM dashboard_element

WHERE

dashboard_element.`type`='internal report'

AND element_id = %s

""", (self.element_id, )) #AND enabled_ind = 'Y'

except MySQLdb.Error, message:

self._logger.error("Report generator. Try to get element %s" % self.element_id)

log_traceback(self._logger, traceback)

print simplejson.dumps({'status': 'ERROR', 'message': 'SQL error %s' % message})

sys.exit()

if res:

self._elements = [row for row in self._db.record]

self._run()

else:

print simplejson.dumps({'status': 'ERROR', 'message': 'report %s updated failed. element not found' % self.element_id})

def _run_all(self):

"""

fetch all enabled reports

"""

try:

res = self._db.Query("""SELECT dashboard_element.element_id

FROM dashboard_element

WHERE

dashboard_element.`type`='internal report'

AND enabled_ind = 'Y'

ORDER BY element_id""")

except MySQLdb.Error, message:

self._logger.error("Report generator. Try to get all elements sql error - %s" % message)

log_traceback(self._logger, traceback)

print simplejson.dumps({'status': 'ERROR', 'message': 'SQL error %s' % message})

sys.exit()

if res:

self._elements = [row for row in self._db.record]

self._run()

else:

print simplejson.dumps({'status': 'ERROR', 'message': 'No reports found'})

def _run(self):

"""

process report(s)

"""

for row in self._elements:

report = None

try:

report = Report(self.command, self.process_dataset_ids, self.initial_measurement_time, self.final_measurement_time)

report.set_logger(self._logger)

#start = time.time()

report.init(row['element_id'])

res = report.update(self.segment_value_id)

#print "it took", time.time() - start, "seconds."

except MySQLdb.Error, message:

self._logger.error("Report generator. Report %s update failed. SQL error %s" % (row['element_id'], message))

log_traceback(self._logger, traceback)

print simplejson.dumps({'status': 'ERROR', 'message': 'report %s update failed. SQL error %s' % (row['element_id'], message)})

if report:

report.unlock()

continue

except Exception, exc:

self._logger.error("Report generator. Report %s update failed. Exception %s" % (row['element_id'], exc))

log_traceback(self._logger, traceback)

print simplejson.dumps({'status': 'ERROR', 'message': 'report %s update failed. %s' % (row['element_id'], exc)})

if report:

report.unlock()

continue

if res:

print simplejson.dumps({'status': 'OK', 'message': 'report %s updated successfully. %s' % (row['element_id'], res)})

else: