iktdev/MediaProcessing
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Updated app.py

Browse Source
This commit is contained in:
Brage 2023-07-18 03:05:35 +02:00
parent d2263aaad4
commit 69e938d10e
1 changed files with 6 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
pyMetadata/app.py
View File

@ -114,32 +114,28 @@ class MessageHandlerThread(threading.Thread):
def run(self): def run(self):
logger.info("Handling message: key=%s, value=%s", self.message.key, self.message.value) logger.info("Handling message: key=%s, value=%s", self.message.key, self.message.value)
# Deserialiser meldingsverdien fra JSON til et Python-dictionary
message_value = json.loads(self.message.value)
# Sjekk om meldingen har en Status # Sjekk om meldingen har en Status
if 'status' in message_value: if 'status' in self.message.value:
status_type = message_value['status']['statusType'] status_type = self.message.value['status']['statusType']
# Sjekk om statusen er SUCCESS # Sjekk om statusen er SUCCESS
if status_type == 'SUCCESS': if status_type == 'SUCCESS':
data_value = message_value['data']["title"] data_value = self.message.value['data']["title"]
logger.info("Performing action for data: %s", data_value)
# Utfør handlingen basert på verdien # Utfør handlingen basert på verdien
result = self.perform_action(title=data_value) result = self.perform_action(title=data_value)
producerMessage = self.compose_message(referenceId=message_value["referenceId"], result=result) producerMessage = self.compose_message(referenceId=self.message.value["referenceId"], result=result)
# Serialiser resultatet til JSON # Serialiser resultatet til JSON
result_json = json.dumps(producerMessage.to_json()) result_json = producerMessage.to_json()
# Send resultatet tilbake ved hjelp av Kafka-producer # Send resultatet tilbake ved hjelp av Kafka-producer
producer = KafkaProducer(bootstrap_servers=bootstrap_servers) producer = KafkaProducer(bootstrap_servers=bootstrap_servers)
producer.send(kafka_topic, key="event:metadata:obtained", value=result_json) producer.send(kafka_topic, key="event:metadata:obtained", value=result_json)
producer.close() producer.close()
def perform_action(self, title) -> DataResult: def perform_action(self, title) -> DataResult:
anii = AniiMetadata(title) anii = AniiMetadata(title)
imdb = ImdbMetadata(title) imdb = ImdbMetadata(title)