import requests import mysql.connector #barcode -> označení produktu #insert -> jestli chceme přidávat nebo odebírat produkt (1 = přidat; 0 = odebrat) def main(barcode, insert: bool): try: mydb = mysql.connector.connect( host="localhost", user="uzivatel", password="heslo" ) mycursor = mydb.cursor() if insert == True: mycursor.execute( f"SELECT COUNT(*) FROM Bc.Seznam WHERE Barcode = '{barcode}'") je_pritomen = bool(mycursor.fetchone()[0]) if je_pritomen: mycursor.execute( f"SELECT Pocet FROM Bc.Seznam WHERE Barcode = '{barcode}'") aktualni_pocet = mycursor.fetchone()[0] mycursor.execute( f"UPDATE Bc.Seznam SET Pocet = {aktualni_pocet+1} WHERE Barcode = '{barcode}'") else: res = requests.get( f"https://world.openfoodfacts.org/api/v0/product/{barcode}.json") if res.status_code == requests.status_codes.codes.ok: res = res.json() if res["status"] == 1: product_name = res["product"]["product_name"].capitalize() brands = res["product"]["brands"].capitalize() mycursor.execute( f"INSERT INTO Bc.Seznam (Nazev, Pod_Nazev, Pocet, Barcode) VALUES ('{product_name}', '{brands}', 1, '{barcode}')") else: mycursor.execute( f"SELECT Pocet FROM Bc.Seznam WHERE Barcode = '{barcode}'") if mycursor.fetchone() != None: aktualni_pocet = mycursor.fetchone()[0] if aktualni_pocet > 0: mycursor.execute( f"UPDATE Bc.Seznam SET Pocet = {aktualni_pocet-1} WHERE Barcode = '{barcode}'") except mysql.connector.Error as error: print(f"Error ---> {error}") finally: if mydb.is_connected(): mydb.commit() mycursor.close() if __name__ == "__main__": main(8722700919254, 1)