Piotr Domański
e2af49046c
All checks were successful
Python lint and test / linttest (push) Successful in 5m30s
52 lines
1.2 KiB
Python
52 lines
1.2 KiB
Python
from dataclasses import dataclass
|
|
from logging import getLogger
|
|
|
|
import requests as r
|
|
|
|
logger = getLogger(__name__)
|
|
|
|
|
|
class NotFound(Exception):
|
|
pass
|
|
|
|
|
|
class ParseError(Exception):
|
|
pass
|
|
|
|
|
|
@dataclass
|
|
class Product:
|
|
name: str
|
|
kcal: float
|
|
fat: float
|
|
protein: float
|
|
carb: float
|
|
fiber: float
|
|
|
|
|
|
def find(bar_code: str) -> Product:
|
|
url = f"https://world.openfoodfacts.org/api/v2/product/{bar_code}.json"
|
|
response = r.get(url)
|
|
|
|
if response.status_code == 404:
|
|
raise NotFound()
|
|
|
|
try:
|
|
data = response.json()
|
|
|
|
name = data["product"]["product_name"]
|
|
|
|
if data["product"]["brands"]:
|
|
name = data["product"]["brands"] + " " + name
|
|
|
|
return Product(
|
|
name=name,
|
|
kcal=data["product"]["nutriments"]["energy-kcal_100g"],
|
|
fat=data["product"]["nutriments"]["fat_100g"],
|
|
protein=data["product"]["nutriments"]["proteins_100g"],
|
|
carb=data["product"]["nutriments"]["carbohydrates_100g"],
|
|
fiber=data["product"]["nutriments"].get("fiber_100g", 0.0),
|
|
)
|
|
except Exception as e:
|
|
logger.error(e)
|
|
raise ParseError()
|