def update_state(file_name):
payload = {
'parent_name': globals.PARENT_NAME,
'group_name': globals.GROUP_NAME,
'container_name': globals.RECEIVE_TOPIC,
'file_name': file_name,
'client_id': globals.CLIENT_ID
}
try:
requests.request("POST", globals.DASHBOARD_URL, data=payload)
except Exception as e:
print(f"{e} EXCEPTION IN UPDATE STATE API CALL......")
err_logger(f"{e} EXCEPTION IN UPDATE STATE API CALL......FILE ID {FILE_ID}")
def save_to_db(db_object, labels, scores):
try:
print("*****************SAVING TO DB******************************")
print("in save")
print(db_object)
print(db_object.id)
db_object.labels = labels
db_object.scores = scores
db_object.save()
print("*****************SAVED TO DB******************************")
except Exception as e:
print(" ERROR IN SAVE TO DB")
err_logger(str(e)+" ERROR IN SAVE TO DB")
def save_to_db(db_object, feature_to_save, file):
try:
print("*****************SAVING TO DB******************************")
print("in save")
feature_obj = Features()
feature_obj.document = db_object
feature_obj.feature = feature_to_save
feature_obj.file = file
feature_obj.save()
print("*****************SAVED TO DB******************************")
except Exception as e:
print(" ERROR IN SAVE TO DB")
err_logger(str(e)+" ERROR IN SAVE TO DB")
global_init()
message = message.value
db_key = str(message)
print(db_key, 'db_key')
FILE_ID = db_key
try:
db_object = Cache.objects.get(pk=db_key)
file_name = db_object.file_name
print("#############################################")
print("########## PROCESSING FILE " + file_name)
print("#############################################")
final_labels = db_object.labels
final_scores = db_object.scores
except Exception as e:
print("EXCEPTION IN FETCHING FROM DATABASE......")
err_logger(str(e) + " EXCEPTION IN FETCHING FROM DATABASE......FILE ID " + FILE_ID)
continue
if db_object.is_doc_type:
"""document"""
if db_object.contains_images:
images_array = []
for image in db_object.files:
pdf_image = str(uuid.uuid4()) + ".jpg"
with open(pdf_image, 'wb') as file_to_save:
file_to_save.write(image.file.read())
images_array.append(pdf_image)
for image in images_array:
try:
response = predict(file_name=image)
for label, score in zip(response["labels"], response['scores']):
if label not in final_labels: