diff --git a/src/main.py b/src/main.py index 01c248f..991aa52 100644 --- a/src/main.py +++ b/src/main.py @@ -64,6 +64,14 @@ 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" + 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" + + return full_filepath + @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"