Updated python
This commit is contained in:
parent
3f53510fd9
commit
f3e66a1a6b
@ -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,
|
||||||
|
|||||||
@ -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))
|
||||||
|
|
||||||
@ -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))
|
||||||
@ -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
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user