예제 #1
0
def store_to_db(data, extra_attr=None):
    create = False
    try:
        result = ResultData.get(task_id=data["task_id"])


    except ResultData.DoesNotExist:
        create = True

    if create:
        result = ResultData()
        result.transaction_id = data["transaction_id"]
        result.task_id = data["task_id"]
        result.task_type = data["test_type"]
        result.location = data["location"]
        result.country = data["country"]
        result.url = data["url"]

        if extra_attr:
            result.extra_attr = extra_attr

    result.task_status = data["status"]

    # Difference between status and task_status. task_status is for celery task.
    result.status = str(data["status_code"])

    result.raw_data = data
    result.save()