Updated python

This commit is contained in:
Brage 2023-07-17 01:15:27 +02:00
parent 3f53510fd9
commit f3e66a1a6b
4 changed files with 14 additions and 14 deletions

View File

@ -4,7 +4,7 @@ import threading
import json import json
from kafka import KafkaConsumer, KafkaProducer from kafka import KafkaConsumer, KafkaProducer
from fuzzywuzzy import fuzz from fuzzywuzzy import fuzz
from sources.result import Result, Metadata from sources.result import DataResult, Metadata
from sources.anii import metadata as AniiMetadata from sources.anii import metadata as AniiMetadata
from sources.imdb import metadata as ImdbMetadata from sources.imdb import metadata as ImdbMetadata
@ -103,7 +103,7 @@ class MessageHandlerThread(threading.Thread):
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) -> Result: def perform_action(self, title) -> DataResult:
anii = AniiMetadata(title) anii = AniiMetadata(title)
imdb = ImdbMetadata(title) imdb = ImdbMetadata(title)
@ -132,13 +132,13 @@ class MessageHandlerThread(threading.Thread):
else: else:
# Begge registrene feilet, håndter etter eget behov # Begge registrene feilet, håndter etter eget behov
most_likely_result = Result(statusType="ERROR", errorMessage="No Result") most_likely_result = DataResult(statusType="ERROR", errorMessage="No Result")
# Returner det mest sannsynlige resultatet # Returner det mest sannsynlige resultatet
return most_likely_result return most_likely_result
def compose_message(self, referenceId: str, result: Result) -> ProducerDataValueSchema: def compose_message(self, referenceId: str, result: DataResult) -> ProducerDataValueSchema:
"""""" """"""
return ProducerDataValueSchema( return ProducerDataValueSchema(
referenceId=referenceId, referenceId=referenceId,

View File

@ -1,5 +1,5 @@
from AnilistPython import Anilist from AnilistPython import Anilist
from result import Metadata, Result from result import Metadata, DataResult
class metadata(): class metadata():
name: str = None name: str = None
@ -8,7 +8,7 @@ class metadata():
def __init__(self, name) -> None: def __init__(self, name) -> None:
self.name = name self.name = name
def lookup(self) -> Result: def lookup(self) -> DataResult:
"""""" """"""
try: try:
result = self.anilist.get_anime(self.name) result = self.anilist.get_anime(self.name)
@ -24,10 +24,10 @@ class metadata():
else: else:
meta.type = 'serie' meta.type = 'serie'
meta.genres = result.get('genres', []) meta.genres = result.get('genres', [])
return Result("SUCCESS", None, meta) return DataResult("SUCCESS", None, meta)
except IndexError as ingore: except IndexError as ingore:
return Result(statusType="IGNORE", errorMessage=f"No result for {self.name}") return DataResult(statusType="IGNORE", errorMessage=f"No result for {self.name}")
except Exception as e: except Exception as e:
return Result(statusType="ERROR", errorMessage=str(e)) return DataResult(statusType="ERROR", errorMessage=str(e))

View File

@ -1,5 +1,5 @@
import imdb import imdb
from result import Metadata, Result from result import Metadata, DataResult
class metadata(): class metadata():
name: str = None name: str = None
@ -9,7 +9,7 @@ class metadata():
self.name = name self.name = name
def lookup(self) -> Result: def lookup(self) -> DataResult:
"""""" """"""
try: try:
query = self.imdb.search_movie(self.name) query = self.imdb.search_movie(self.name)
@ -29,6 +29,6 @@ class metadata():
meta.genres = result.get('genres', []) meta.genres = result.get('genres', [])
return Result("SUCCESS", None, meta) return DataResult("SUCCESS", None, meta)
except Exception as e: except Exception as e:
return Result(statusType="ERROR", errorMessage=str(e)) return DataResult(statusType="ERROR", errorMessage=str(e))

View File

@ -11,7 +11,7 @@ class Metadata:
genres: List[str] genres: List[str]
@dataclass @dataclass
class Result: class DataResult:
statusType: str statusType: str
errorMessage: str errorMessage: str
data: Metadata data: Metadata