def handle(self, *args, **options): if options['load_time']: extension = os.path.splitext(options['filename'])[1] if extension == '.xlsx' or extension == '.xls': excel_data = MaximoExcelData(stdout=self.stdout) results = excel_data.load(options['filename'], action=MaximoExcelData.LOAD_TIME) self.stdout.write('Parsed: %s' % options['filename']) self.stdout.write('Created Registers: %d of %d' % (results['time_results']['created'], results['time_results']['rows_parsed'])) elif extension == '.csv': excel_data = MaximoCSVData(stdout=self.stdout) results = excel_data.load_time_registers(options['filename']) self.stdout.write('Parsed: %s' % options['filename']) self.stdout.write('Created Registers: %d of %d' % (results['created'], results['rows_parsed'])) elif options['export_time']: excel_data = MaximoExcelData(stdout=self.stdout) registers = MaximoTimeRegister.objects.all() if options['filename']: filename = options['filename'] else: filename = filename_with_datetime(TEST_OUTPUT_PATH, 'Export_Times_%s.xlsx' % ('TINO')) excel_data.export_time_registers(filename, registers) self.stdout.write('Wrote: %s' % filename) elif options['update_projects']: self.update_projects()
def handle(self, *args, **options): if options['load_time']: extension = os.path.splitext(options['filename'])[1] if extension == '.xlsx' or extension == '.xls': excel_data = MaximoExcelData(stdout=self.stdout) results = excel_data.load(options['filename'], action=MaximoExcelData.LOAD_TIME) self.stdout.write('Parsed: %s' % options['filename']) self.stdout.write('Created Registers: %d of %d' % (results['time_results']['created'], results['time_results']['rows_parsed'])) elif extension == '.csv': excel_data = MaximoCSVData(stdout=self.stdout) results = excel_data.load_time_registers(options['filename']) self.stdout.write('Parsed: %s' % options['filename']) self.stdout.write('Created Registers: %d of %d' % (results['created'], results['rows_parsed'])) elif options['export_time']: excel_data = MaximoExcelData(stdout=self.stdout) registers = MaximoTimeRegister.objects.all() if options['filename']: filename = options['filename'] else: filename = filename_with_datetime( TEST_OUTPUT_PATH, 'Export_Times_%s.xlsx' % ('TINO')) excel_data.export_time_registers(filename, registers) self.stdout.write('Wrote: %s' % filename) elif options['update_projects']: self.update_projects()
def test_export_time_registers(self): filename = filename_with_datetime(TEST_OUTPUT_PATH,'maximo_time_data_export.xlsx') #os.path.join(TEST_OUTPUT_PATH, '%s_%s.xlsx' % ('maximo_time_data_export', timezone.now().strftime('%Y%m%d_%H%M'))) excel_data = MaximoExcelData() self.create_time_registers(date(2015, 9, 1), date(2015, 9, 30)) registers = MaximoTimeRegister.objects.all() excel_data.export_time_registers(filename, registers) self.assertTrue(os.path.exists(filename)) logger.debug('Wrote: %s' % filename) os.remove(filename) self.assertFalse(os.path.exists(filename))
def test_export_time_registers(self): filename = filename_with_datetime(TEST_OUTPUT_PATH, 'maximo_time_data_export.xlsx') #os.path.join(TEST_OUTPUT_PATH, '%s_%s.xlsx' % ('maximo_time_data_export', timezone.now().strftime('%Y%m%d_%H%M'))) excel_data = MaximoExcelData() self.create_time_registers(date(2015, 9, 1), date(2015, 9, 30)) registers = MaximoTimeRegister.objects.all() excel_data.export_time_registers(filename, registers) self.assertTrue(os.path.exists(filename)) logger.debug('Wrote: %s' % filename) os.remove(filename) self.assertFalse(os.path.exists(filename))