def handle(self, *args, **options): verbosity = int(options['verbosity']) if verbosity == 0: borme.importer.logger.setLevel(logging.ERROR) elif verbosity == 1: # default borme.importer.logger.setLevel(logging.INFO) elif verbosity == 2: borme.importer.logger.setLevel(logging.INFO) elif verbosity > 2: borme.importer.logger.setLevel(logging.DEBUG) if verbosity > 2: logging.getLogger().setLevel(logging.DEBUG) start_time = time.time() if args: for filename in args: print(filename) import_borme_json(filename) config = Config.objects.first() if config: config.last_modified = timezone.now() else: config = Config(last_modified=timezone.now()) config.version = get_git_revision_short_hash() config.save() # Elapsed time elapsed_time = time.time() - start_time print('\nElapsed time: %.2f seconds' % elapsed_time)
def test_nombramientos_ceses(self): companies = Company.objects.all() self.assertEqual(len(companies), 0) json_path = os.path.join(THIS_PATH, 'files', '1_nombramientos.json') ret = import_borme_json(json_path) self.assertTrue(ret) companies = Company.objects.all() self.assertEqual(len(companies), 1) company = companies[0] self.assertEqual(len(company.cargos_actuales), 2) self.assertEqual(len(company.cargos_historial), 0) json_path = os.path.join(THIS_PATH, 'files', '2_ceses.json') ret = import_borme_json(json_path) self.assertTrue(ret) companies = Company.objects.all() self.assertEqual(len(companies), 1) company = companies[0] self.assertEqual(len(company.cargos_actuales), 0) self.assertEqual(len(company.cargos_historial), 2)
def test_nombramientos_ceses(self): companies = Company.objects.all() self.assertEqual(len(companies), 0) json_path = os.path.join(THIS_PATH, 'files', '1_nombramientos.json') ret = import_borme_json(json_path) self.assertTrue(ret) companies = Company.objects.all() self.assertEqual(len(companies), 1) company = companies[0] self.assertEqual(len(company.get_cargos_actuales()[0]), 2) self.assertEqual(len(company.get_cargos_historial()[0]), 0) json_path = os.path.join(THIS_PATH, 'files', '2_ceses.json') ret = import_borme_json(json_path) self.assertTrue(ret) companies = Company.objects.all() self.assertEqual(len(companies), 1) company = companies[0] self.assertEqual(len(company.get_cargos_actuales()[0]), 0) self.assertEqual(len(company.get_cargos_historial()[0]), 2)
def test_nombramientos_ceses(self): companies = Company.objects.all() self.assertEqual(len(companies), 0) json_path = os.path.join(THIS_PATH, 'files', '3_cese_y_nombramiento.json') ret = import_borme_json(json_path) self.assertTrue(ret) companies = Company.objects.all() self.assertEqual(len(companies), 1) company = companies[0] self.assertEqual(company.name, 'EMPRESA TRES') self.assertEqual(len(company.cargos_actuales), 1) self.assertEqual(len(company.cargos_historial), 2)
def test_extincion(self): companies = Company.objects.all() self.assertEqual(len(companies), 0) json_path = os.path.join(THIS_PATH, 'files', '4_extincion.json') ret = import_borme_json(json_path) self.assertTrue(ret) companies = Company.objects.all() self.assertEqual(len(companies), 1) company = companies[0] self.assertEqual(company.name, 'EMPRESA EXTINGUIDA') self.assertEqual(company.is_active, False) self.assertEqual(company.date_extinction, datetime.date(2015, 1, 4)) self.assertEqual(len(company.get_cargos_actuales()[0]), 0) self.assertEqual(len(company.get_cargos_historial()[0]), 1)