def sync_bulk_create(cls, reg_details, reg_device_id, app, ussd=None):
"""Create devices in bulk."""
try:
flatten_imeis = []
if ussd is None:
imeis_lists = ast.literal_eval(reg_details.imeis)
else:
imeis_lists = []
ims = reg_details.imeis
imeis_lists.append(list(ims.strip('}{').split(",")))
for device_imeis in imeis_lists:
device = cls(device_imeis[0][:8], reg_details.id,
reg_device_id)
device.save()
for imei in device_imeis:
flatten_imeis.append(imei)
imei_device = ImeiDevice(imei, device.id)
db.session.add(imei_device)
imei_object = ApprovedImeis.query.filter_by(
imei=imei[:14]).first()
if not imei_object:
approved_imei = ApprovedImeis(imei[:14],
reg_details.id,
'pending', 'add')
db.session.add(approved_imei)
elif imei_object and imei_object.status == 'removed':
imei_object.status = 'pending'
imei_object.removed = False
imei_object.request_id = reg_details.id
db.session.add(imei_object)
reg_details.update_processing_status('Processed')
reg_details.update_report_status('Processing')
db.session.commit()
task_id = Utilities.generate_summary(flatten_imeis,
reg_details.tracking_id)
if task_id:
Utilities.pool_summary_request(task_id, reg_details, app)
else:
reg_details.update_report_status('Failed')
db.session.commit()
exit()
if app.config['AUTOMATE_IMEI_CHECK'] or ussd:
if Device.auto_approve(task_id, reg_details, flatten_imeis,
app):
app.logger.info(
"Auto Approved/Rejected Registration Application Id:" +
str(reg_details.id))
except Exception as e: # pragma: no cover
reg_details.update_processing_status('Failed')
reg_details.update_report_status('Failed')
app.logger.exception(e)
db.session.commit()
def sync_bulk_create(cls, reg_details, reg_device_id, app):
"""Create devices in bulk."""
try:
flatten_imeis = []
imeis_lists = ast.literal_eval(reg_details.imeis)
for device_imeis in imeis_lists:
device = cls(device_imeis[0][:8], reg_details.id, reg_device_id)
device.save()
for imei in device_imeis:
flatten_imeis.append(imei)
imei_device = ImeiDevice(imei, device.id)
db.session.add(imei_device)
imei_object = ApprovedImeis.query.filter_by(imei=imei[:14]).first()
if not imei_object:
approved_imei = ApprovedImeis(imei[:14], reg_details.id, 'pending', 'add')
db.session.add(approved_imei)
elif imei_object and imei_object.status == 'removed':
imei_object.status = 'pending'
imei_object.removed = False
imei_object.request_id = reg_details.id
db.session.add(imei_object)
reg_details.update_processing_status('Processed')
reg_details.update_report_status('Processing')
db.session.commit()
task_id = Utilities.generate_summary(flatten_imeis, reg_details.tracking_id)
if task_id:
Utilities.pool_summary_request(task_id, reg_details, app)
else:
reg_details.update_report_status('Failed')
db.session.commit()
except Exception:
reg_details.update_processing_status('Failed')
db.session.commit()
def async_bulk_create(cls, reg_details, reg_device_id,
app): # pragma: no cover
"""Create devices async."""
with app.app_context():
from app import db
try:
filename = reg_details.file
tracking_id = reg_details.tracking_id
args = {
'imei_per_device': reg_details.imei_per_device,
'device_count': reg_details.device_count
}
response = Utilities.process_reg_file(filename, tracking_id,
args)
imeis = []
for device_imeis in response:
tac = Utilities.get_imei_tac(device_imeis[0])
device = cls(tac, reg_details.id, reg_device_id)
db.session.add(device)
db.session.flush()
imeis = imeis + device_imeis
for imei in device_imeis:
imei_device = ImeiDevice(imei, device.id)
db.session.add(imei_device)
imei_object = ApprovedImeis.query.filter_by(
imei=imei[:14]).first()
if not imei_object:
approved_imei = ApprovedImeis(
imei[:14], reg_details.id, 'pending', 'add')
db.session.add(approved_imei)
elif imei_object and imei_object.status == 'removed':
imei_object.status = 'pending'
imei_object.removed = False
imei_object.request_id = reg_details.id
db.session.add(imei_object)
db.session.commit()
reg_details.update_processing_status('Processed')
db.session.commit()
task_id = Utilities.generate_summary(imeis,
reg_details.tracking_id)
app.logger.info(
'task with task_id: {0} initiated'.format(task_id))
if not task_id:
reg_details.update_report_status('Failed')
app.logger.info(
'task with task_id: {0} failed'.format(task_id))
db.session.commit()
else:
Utilities.pool_summary_request(task_id, reg_details, app)
except Exception as e:
app.logger.exception(e)
db.session.rollback()
reg_details.update_report_status('Failed')
reg_details.update_processing_status('Failed')
db.session.commit()
raise e
def test_pool_summary_request(app, session, dirbs_dvs): # pylint: disable=unused-argument
"""Verify that the poll_summary function works correctly."""
tracking_id = uuid.uuid4()
request = create_registration(REG_REQ_DATA, tracking_id)
Utilities.create_directory(tracking_id)
assert request.tracking_id == tracking_id
response = Utilities.pool_summary_request('mock-task-id', request, app)
assert request.tracking_id == str(tracking_id)
assert response is None
def async_create(cls, devices, imei_tac_map, old_devices, dereg_id,
imeis_list, app):
"""Async create a new device."""
with app.app_context():
from app import db
dereg = DeRegDetails.get_by_id(dereg_id)
try:
DeRegDevice.clear_devices(old_devices)
for device in devices:
device_imeis = imei_tac_map.get(device.get('tac'))
dereg_imei_list = DeRegImei.get_deregimei_list(
device.get('id'), device_imeis)
res = db.engine.execute(DeRegImei.__table__.insert(),
dereg_imei_list)
res.close()
dereg.update_processing_status('Processed')
db.session.commit()
task_id = Utilities.generate_summary(imeis_list,
dereg.tracking_id)
if task_id:
Utilities.pool_summary_request(task_id, dereg, app)
else:
dereg.update_processing_status('Failed')
db.session.commit()
if app.config['AUTOMATE_IMEI_CHECK']:
if DeRegDevice.auto_approve(task_id, dereg):
print(
"Auto Approved/Rejected DeRegistration Application Id:"
+ str(dereg.id))
except Exception as e:
app.logger.exception(e)
db.session.rollback()
dereg.update_processing_status('Failed')
dereg.update_report_status('Failed')
db.session.commit()
def async_create(cls, devices, imei_tac_map, old_devices, dereg_id,
imeis_list, app):
"""Async create a new device."""
with app.app_context():
from app import db
dereg = DeRegDetails.get_by_id(dereg_id)
try:
DeRegDevice.clear_devices(old_devices)
for device in devices:
device_imeis = imei_tac_map.get(device.get('tac'))
dereg_imei_list = DeRegImei.get_deregimei_list(
device.get('id'), device_imeis)
res = db.engine.execute(DeRegImei.__table__.insert(),
dereg_imei_list)
res.close()
dereg.update_processing_status('Processed')
db.session.commit()
task_id = Utilities.generate_summary(imeis_list,
dereg.tracking_id)
Utilities.pool_summary_request(task_id, dereg, app)
except Exception as e:
app.logger.exception(e)
dereg.update_processing_status('Failed')
db.session.commit()
