diff --git a/src/main.py b/src/main.py index 645c48e..01c248f 100644 --- a/src/main.py +++ b/src/main.py @@ -64,9 +64,9 @@ def get_all_products(): def get_single_product(barcode: int): return data_connection.get_single_product(barcode) -@app.get("/icons/{icon_filename}", response_class=FileResponse) -def get_icon(icon_filename: str): - full_filepath = os.path.join(definitions.ICONS_PATH, icon_filename) + ".svg" +@app.get("/icons/{icon_category}/{icon_filename}", response_class=FileResponse) +def get_icon(icon_category: str, icon_filename: str): + full_filepath = os.path.join(definitions.ICONS_PATH, icon_category, icon_filename) + ".svg" assert os.path.exists(full_filepath), f"File {full_filepath} not found" assert image_validator.is_valid_svg(file_name=full_filepath), f"File {full_filepath} is not a valid image"