Updated python
This commit is contained in:
parent
3f53510fd9
commit
f3e66a1a6b
@ -4,7 +4,7 @@ import threading
|
||||
import json
|
||||
from kafka import KafkaConsumer, KafkaProducer
|
||||
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.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.close()
|
||||
|
||||
def perform_action(self, title) -> Result:
|
||||
def perform_action(self, title) -> DataResult:
|
||||
anii = AniiMetadata(title)
|
||||
imdb = ImdbMetadata(title)
|
||||
|
||||
@ -132,13 +132,13 @@ class MessageHandlerThread(threading.Thread):
|
||||
|
||||
else:
|
||||
# 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
|
||||
return most_likely_result
|
||||
|
||||
|
||||
def compose_message(self, referenceId: str, result: Result) -> ProducerDataValueSchema:
|
||||
def compose_message(self, referenceId: str, result: DataResult) -> ProducerDataValueSchema:
|
||||
""""""
|
||||
return ProducerDataValueSchema(
|
||||
referenceId=referenceId,
|
||||
|
||||
@ -1,5 +1,5 @@
|
||||
from AnilistPython import Anilist
|
||||
from result import Metadata, Result
|
||||
from result import Metadata, DataResult
|
||||
|
||||
class metadata():
|
||||
name: str = None
|
||||
@ -8,7 +8,7 @@ class metadata():
|
||||
def __init__(self, name) -> None:
|
||||
self.name = name
|
||||
|
||||
def lookup(self) -> Result:
|
||||
def lookup(self) -> DataResult:
|
||||
""""""
|
||||
try:
|
||||
result = self.anilist.get_anime(self.name)
|
||||
@ -24,10 +24,10 @@ class metadata():
|
||||
else:
|
||||
meta.type = 'serie'
|
||||
meta.genres = result.get('genres', [])
|
||||
return Result("SUCCESS", None, meta)
|
||||
return DataResult("SUCCESS", None, meta)
|
||||
|
||||
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:
|
||||
return Result(statusType="ERROR", errorMessage=str(e))
|
||||
return DataResult(statusType="ERROR", errorMessage=str(e))
|
||||
|
||||
@ -1,5 +1,5 @@
|
||||
import imdb
|
||||
from result import Metadata, Result
|
||||
from result import Metadata, DataResult
|
||||
|
||||
class metadata():
|
||||
name: str = None
|
||||
@ -9,7 +9,7 @@ class metadata():
|
||||
self.name = name
|
||||
|
||||
|
||||
def lookup(self) -> Result:
|
||||
def lookup(self) -> DataResult:
|
||||
""""""
|
||||
try:
|
||||
query = self.imdb.search_movie(self.name)
|
||||
@ -29,6 +29,6 @@ class metadata():
|
||||
|
||||
meta.genres = result.get('genres', [])
|
||||
|
||||
return Result("SUCCESS", None, meta)
|
||||
return DataResult("SUCCESS", None, meta)
|
||||
except Exception as e:
|
||||
return Result(statusType="ERROR", errorMessage=str(e))
|
||||
return DataResult(statusType="ERROR", errorMessage=str(e))
|
||||
@ -11,7 +11,7 @@ class Metadata:
|
||||
genres: List[str]
|
||||
|
||||
@dataclass
|
||||
class Result:
|
||||
class DataResult:
|
||||
statusType: str
|
||||
errorMessage: str
|
||||
data: Metadata
|
||||
|
||||
Loading…
Reference in New Issue
Block a user