Skip to content

Commit

Permalink
fix img extension in processer
Browse files Browse the repository at this point in the history
  • Loading branch information
ArtemiyGranat committed Feb 10, 2024
1 parent 065cf59 commit dd62bf7
Showing 1 changed file with 7 additions and 2 deletions.
9 changes: 7 additions & 2 deletions image-processer/src/main.py
Original file line number Diff line number Diff line change
Expand Up @@ -22,6 +22,7 @@ def __init__(self) -> None:

self.config = shared_resources.img_processer
self.orig_img_dir = shared_resources.scraper.img_dir
self.orig_img_ext = shared_resources.scraper.img_save_extension
self.sqlite = Database(
gen_sqlite_address(shared_resources.sqlite_creds)
)
Expand All @@ -41,7 +42,9 @@ def init_scheduler(self, func) -> None:


async def process_image(ctx: Context, image: entities.Image) -> None:
with Image.open(f"{ctx.orig_img_dir}/{image.id}.jpg") as orig_img:
with Image.open(
f"{ctx.orig_img_dir}/{image.id}.{ctx.orig_img_ext}"
) as orig_img:
processed_img = rembg.remove(orig_img)
processed_img.save(f"{ctx.config.img_dir}/{image.id}.png")

Expand All @@ -51,7 +54,9 @@ async def process_image(ctx: Context, image: entities.Image) -> None:
),
fields=["processed"],
)
logger.info(f"Removed background: {ctx.orig_img_dir}/{image.id}.jpg")
logger.info(
f"Removed background: {ctx.orig_img_dir}/{image.id}.{ctx.orig_img_ext}"
)


async def process_images(ctx: Context) -> None:
Expand Down

0 comments on commit dd62bf7

Please sign in to comment.