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

Correction1

Browse Source
This commit is contained in:
bskjon 2024-04-15 17:25:12 +02:00
parent dcf247fe41
commit 5e52385c68
1 changed files with 14 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
apps/pyMetadata/sources/select.py
View File

@ -77,11 +77,16 @@ class UseSource():
def select_result(self) -> Optional[DataResult]: def select_result(self) -> Optional[DataResult]:
"""""" """"""
weightResult = self.__perform_search(title=self.title) scored: List[DataAndScore] = []
titleResult = self.__perform_search(title=self.title)
baseNameResult = self.__perform_search(title=self.baseName) baseNameResult = self.__perform_search(title=self.baseName)
weightResult.extend(baseNameResult)
scored = self.__calculate_score(title=self.title, weightData=weightResult) titleScoreResult = self.__calculate_score(title=self.title, weightData=titleResult)
baseNameScoreResult = self.__calculate_score(title=self.baseName, weightData=baseNameResult)
scored.extend(titleScoreResult)
scored.extend(baseNameScoreResult)
scored.sort(key=lambda x: x.score, reverse=True) scored.sort(key=lambda x: x.score, reverse=True)
jsr = "" jsr = ""
@ -95,12 +100,15 @@ class UseSource():
logger.info(jsr) logger.info(jsr)
try: try:
titledResult = titleResult
titledResult.extend(baseNameResult)
titles: List[str] = [] titles: List[str] = []
for wd in weightResult: for wd in titledResult:
titles.append(wd.result.data.title) titles.append(wd.result.data.title)
titles.extend(wd.result.data.altTitle) titles.extend(wd.result.data.altTitle)
joinedTitles = "\n\t" + "\n\t".join(titles) joinedTitles = "\n\t" + "\n\t".join(titles)
logger.info(f"[Title]: {self.title} \nFound: {joinedTitles} \nTitle selected: \n\t{scored[0].result.data.title}\n") logger.info(f"\nTitle: {self.title} \nBaseName: {self.baseName} \nFound: {joinedTitles} \nTitle selected: \n\t{scored[0].result.data.title}\n")
except Exception as e: except Exception as e:
logger.error(e) logger.error(e)
pass pass