class Applications(object):
def __init__(self, user, password):
self.logger = Logger()
self.xml_parser = xml_parser.XmlParser()
self.user = user
self.password = password
return
def get_apps(self):
apps = []
try:
self.logger.info('Attempting to get applications!!!')
resp = requests.post(endpoints.REQUEST_APPS,
auth=HTTPBasicAuth(self.user, self.password))
if resp.status_code == 200:
response = self.xml_parser.parse_xml(resp.text)
for item in response:
if item.attrib['app_id'] is not None and not item.attrib[
'app_id'] in apps:
apps.append({
'app_id': item.attrib['app_id'],
'app_name': item.attrib['app_name'],
'sandboxes': []
})
self.logger.info('Successfully retrieved app list!')
else:
self.logger.error('Authentication Issue: {}'.format(
resp.status_code))
except Exception as e:
self.logger.exception('Error retrieving app list: {}'.format(e))
return apps
class SpreadsheetCreator(object):
def __init__(self):
self.have_data = False
self.logger = Logger()
self.filename = '{}VeracodeSandboxResults-{}.xlsx'.format(
os.getenv('VC_ATTACHMENT_BASE_PATH'), date.today())
return
def write_spreadsheet(self, apps):
try:
workbook = xlsxwriter.Workbook(self.filename)
worksheet = workbook.add_worksheet('Aggregated Build Data')
bold = workbook.add_format({'bold': True})
worksheet.write('A1', 'User', bold)
worksheet.write('B1', 'Total Builds', bold)
row = 1
col = 0
data = {}
if len(apps) > 0:
for app in apps:
for sandbox in app['sandboxes']:
for build in sandbox['builds']:
if not build['submitter'] in data:
data[build['submitter']] = 0
data[build['submitter']] += 1
if (len(data) > 0):
self.have_data = True
for user, number_of_builds in data.items():
worksheet.write(row, col, user)
worksheet.write(row, col + 1, number_of_builds)
row += 1
else:
raise 'Unable to create spreadsheet, no apps found in data'
workbook.close()
except Exception as e:
self.logger.exception('Error creating spreadsheet: {}'.format(e))