def insert_category(category): a = Categories.select(Categories.name).where(Categories.name == category) if not a.exists(): id_company = random.randint(0, 7000000) Categories.create(id=id_company, name=category) return 'Done' return 'Genre already exists'
# Connect to a MySQL database on network. db = peewee.MySQLDatabase('foodstuff', user='******', password='', host='localhost', port=3306) # The tables are created from a model with create_table() Categories.create_table() Products.create_table() Substitutes.create_table() # populate the Category table for item in CATEGORIES_ARRAY: item = Categories.create(name=item) item.save() # get id and name from categories to populate products table def fill_db_from_categories(cat): for index, value in enumerate(cat): temp_var = "var" + str(index) temp_var = requests.get( "https://fr.openfoodfacts.org/cgi/search.pl?action=process&tagtype_0=labels&tag_contains_0=contains&tag_0=sans%20gluten&tagtype_1=categories&tag_contains_1=contains&tag_1={}&sort_by=unique_scans_n&page_size=100&axis_x=energy&axis_y=products_n&action=display&json=1" .format(value)).json() for x, i in enumerate(temp_var['products']):