Writes code for function get_product() #4

This commit is contained in:
Fred Z 2018-07-24 20:40:31 +02:00
parent a4faa0631d
commit ea26876509

View File

@ -13,6 +13,16 @@ nurition grade.
import json
import requests
# CONFIG
FIELD_KEPT = [
'product_name',
'stores',
'nutrition_grades',
'categories_tags',
'id'
]
def get_product(code):
"""
Call OpenFF API to get data of a single product
@ -35,6 +45,19 @@ def get_product(code):
['en:sugary-snacks', 'en:biscuits-and-cakes', 'en:biscuits', 'fr:petits-beurres']
"""
response = requests.get(
"https://fr.openfoodfacts.org/api/v0/product/{}.json".format(code)
)
product_data = json.loads(response.text)
product = {}
for field in FIELD_KEPT:
product[field] = product_data['product'][field]
return product
if __name__ == "__main__":
""" Starting doctests """