diff --git a/src/main.py b/src/main.py index c53f3d6..55158f1 100644 --- a/src/main.py +++ b/src/main.py @@ -1,7 +1,9 @@ from fastapi import FastAPI from fastapi.middleware.cors import CORSMiddleware +from fastapi.responses import FileResponse from src.classes import product_classes +from src.config import definitions from src import data_connection app = FastAPI() @@ -28,4 +30,9 @@ def get_all_products(): @app.get("/products/{barcode}", response_model=product_classes.Product) def get_single_product(barcode: int): - return data_connection.get_single_product(barcode) \ No newline at end of file + return data_connection.get_single_product(barcode) + +@app.get("/icons/{icon_filename}", response_class=FileResponse) +def get_icon(icon_filename: str): + full_filepath = f"{definitions.ICONS_PATH}/{icon_filename}" + return full_filepath \ No newline at end of file