| import requests |
| from data.task import Task |
| from util.slack import Slack |
|
|
| comic_url = "http://internal-k8s-gamma-internal-ea8e32da94-1997933257.ap-south-1.elb.amazonaws.com:80" |
|
|
|
|
| def updateSource(sourceId, userId, state): |
| print("update source is called") |
| url = comic_url + f"/comic-crecoai/source/{sourceId}" |
| headers = {"Content-Type": "application/json", "user-id": str(userId)} |
|
|
| data = {"state": state} |
|
|
| try: |
| response = requests.patch(url, headers=headers, json=data, timeout=10) |
| print("update source response", response) |
| except requests.exceptions.Timeout: |
| print("Request timed out while updating source") |
| except requests.exceptions.RequestException as e: |
| print(f"Error while updating source: {e}") |
|
|
| return |
|
|
|
|
| def saveGeneratedImages(sourceId, userId): |
| print("save generation called") |
| url = comic_url + "/comic-crecoai/source/" + str(sourceId) + "/generatedImages" |
| headers = {"Content-Type": "application/json", "user-id": str(userId)} |
| data = {"state": "ACTIVE"} |
|
|
| try: |
| requests.patch(url, headers=headers, json=data) |
| |
| except requests.exceptions.Timeout: |
| print("Request timed out while saving image") |
| except requests.exceptions.RequestException as e: |
| print("Failed to mark source as active: ", e) |
| return |
| return |
|
|
|
|
| def update_db(func): |
| def caller(*args, **kwargs): |
| if type(args[0]) is not Task: |
| raise Exception("First argument must be a Task object") |
| task = args[0] |
| try: |
| updateSource(task.get_sourceId(), task.get_userId(), "INPROGRESS") |
| rargs = func(*args, **kwargs) |
| updateSource(task.get_sourceId(), task.get_userId(), "COMPLETED") |
| saveGeneratedImages(task.get_sourceId(), task.get_userId()) |
| return rargs |
| except Exception as e: |
| print("Error processing image: {}".format(str(e))) |
| slack = Slack() |
| slack.error_alert(task, e) |
| updateSource(task.get_sourceId(), task.get_userId(), "FAILED") |
|
|
| return caller |
|
|
