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)
