diff --git a/scrapinglib/api.py b/scrapinglib/api.py index 953aca1..454a157 100644 --- a/scrapinglib/api.py +++ b/scrapinglib/api.py @@ -147,6 +147,8 @@ class Scraping: def searchAdult(self, number, sources): if self.specifiedSource: sources = [self.specifiedSource] + elif type(sources) is list: + pass else: sources = self.checkAdultSources(sources, number) json_data = {} @@ -171,7 +173,17 @@ class Scraping: break except: continue - + + # javdb的封面有水印,如果可以用其他源的封面来替换javdb的封面 + if json_data['source'] == 'javdb': + other_sources = sources[sources.index('javdb') + 1:] + # search other sources + json_data_other = self.searchAdult(number, other_sources) + if json_data_other is not None: + json_data['cover'] = json_data_other['cover'] + if self.debug: + print(f"[+]Find movie [{number}] cover on website '{json_data_other['cover']}'") + # Return if data not found in all sources if not json_data: print(f'[-]Movie Number [{number}] not found!')