diff --git a/db.py b/db.py index fab2211..9a02e6c 100644 --- a/db.py +++ b/db.py @@ -40,7 +40,7 @@ def new_part(cloth_type, weigth, pieces): logging.error("only ints allowed") else: cursor = connection.cursor() - executed = "INSERT INTO luggage (clothes, weigth, pieces) VALUES (?, ?)" + executed = "INSERT INTO luggage (clothes, weigth, pieces) VALUES (?, ?, ?)" executed_data = (cloth_type, weigth, pieces) cursor.execute(executed, executed_data) connection.commit() @@ -96,4 +96,12 @@ def fetch_piece(cloth_type): def fetch_all_pieces(): cursor = connection.cursor() cursor.execute("SELECT sum(pieces) FROM luggage") - pieces = cursor.fetchone()[0] \ No newline at end of file + pieces = cursor.fetchone()[0] + logging.info("fetched all luggage pieces") + return pieces + +def delete(cloth_type): + cursor = connection.cursor() + cursor.execute("DELETE FROM luggage WHERE clothes = ?;", (cloth_type,)) + logging.critical(f"{cloth_type} was deleted") + connection.commit() diff --git a/main.py b/main.py index ac7b32f..953992e 100644 --- a/main.py +++ b/main.py @@ -22,8 +22,8 @@ except Exception as e: def insert(): clothe = input("How is the clothe called? ").upper() - weigth = input("The weigth please? ") - pieces = input("How many pieces? ints only ") + weigth = float(input("The weigth please? ")) + pieces = int(input("How many pieces? ints only ")) db.new_part(clothe, weigth, pieces) print("it is inserted") time.sleep(3) @@ -88,6 +88,19 @@ def get_weigth_pieces(): os.system('cls' if os.name == 'nt' else 'clear') menu() +def delete(): + clothe = input("How is the clothe called? ").upper() + really = input("Do you really want to delete it? Y/N").upper() + if really == "Y": + db.delete(clothe) + print(f"{clothe} was deleted") + time.sleep(3) + else: + time.sleep(0.5) + logging.info("no item deleted") + os.system('cls' if os.name == 'nt' else 'clear') + menu() + def exit_programm(): time.sleep(0.5) print(r""" @@ -118,6 +131,7 @@ def menu(): print("6: Get pieces of luggage") print("7: Get clothes") print("8: Get weigth and pieces of clothes") + print("9: Delete Item") print("0: EXIT") choice = input("Type number here ") os.system('cls' if os.name == 'nt' else 'clear') @@ -141,6 +155,8 @@ def menu(): get_weigth_pieces() if choice == "0": exit_programm() + if choice == "9": + delete()