fix: passing referer when scraping flower (#20)

Co-authored-by: Christopher Frige <christopher.santos@neogrid.com>

Author: christopherfrige

backend/src/domain/use_cases/common/download_image.py

@@ -3,7 +3,7 @@
 
 
 class DownloadImageUseCase:
-    def __init__(self, referer=None):
+    def __init__(self, referer: str | None):
         self.headers = {}
         if referer:
             self.headers = {

backend/src/domain/use_cases/manwha/manage_manwha.py

@@ -26,7 +26,7 @@
 
 
 class ManageManwhaUseCase:
-    def __init__(self, session: Session, storage: S3Service):
+    def __init__(self, session: Session, storage: S3Service, referer: str | None):
         self.manwha_repository = ManwhaRepository(session)
         self.genre_repository = GenreRepository(session)
         self.artist_repository = ArtistRepository(session)
@@ -35,7 +35,7 @@ def __init__(self, session: Session, storage: S3Service):
         self.manwha_genre_repository = ManwhaGenreRepository(session)
         self.manwha_artist_repository = ManwhaArtistRepository(session)
         self.manwha_author_repository = ManwhaAuthorRepository(session)
-        self.download_image = DownloadImageUseCase()
+        self.download_image = DownloadImageUseCase(referer)
         self.storage = storage
 
     def execute(self, manwha_data) -> int:

backend/src/domain/use_cases/scraper/base_scraper.py

@@ -23,23 +23,17 @@
 
 
 class BaseScraperUseCase(ABC):
-    def __init__(
-        self,
-        session: Session,
-        storage: S3Service,
-    ):
-        self.referer = None
-
+    def __init__(self, session: Session, storage: S3Service, referer: str | None):
         self.session = session
 
         self.manwha_repository = ManwhaRepository(session)
         self.scraper_manwha_repository = ScraperManwhaRepository(session)
         self.chapter_repository = ChapterRepository(session)
 
-        self.manage_manwha = ManageManwhaUseCase(session, storage)
+        self.manage_manwha = ManageManwhaUseCase(session, storage, referer)
         self.upload_chapter_pages = UploadChapterPagesUseCase(session, storage)
         self.check_new_chapters = CheckNewChaptersUseCase(session)
-        self.download_image = DownloadImageUseCase(self.referer)
+        self.download_image = DownloadImageUseCase(referer)
 
     def _driver_options(self):
         options = webdriver.ChromeOptions()

backend/src/domain/use_cases/scraper/scrape_flower_manwhas.py

@@ -8,9 +8,9 @@
 
 class ScrapeFlowerManwhasUseCase(BaseScraperUseCase):
     def __init__(self, session: Session, storage: S3Service):
-        super().__init__(session, storage)
-        self.reader_id = ReaderEnum.FLOWER.value
         self.referer = "https://flowermangas.com/"
+        self.reader_id = ReaderEnum.FLOWER.value
+        super().__init__(session, storage, self.referer)
 
     def scrape_manwha_main_page(self, manwha_url: str):
         self.scraper.get(manwha_url)