import traceback from datetime import datetime from datetime import timedelta from geruecht.exceptions import DatabaseExecption class Base: def get_free_drink_list_config(self, id): try: cursor = self.db.connection.cursor() cursor.execute(f'select * from free_drink_list_config where id={id}') data = cursor.fetchone() if data['drink_id'] != None: data['drink'] = self.getDrinkPrice(data['drink_id']) data['free_drink_types'] = self.get_free_drink_list_types_for_drink(data['id']) return data except Exception as err: traceback.print_exc() self.db.connection.rollback() raise DatabaseExecption("Something went wrong with Database: {}".format(err)) def get_free_drink_list_configs(self): try: cursor = self.db.connection.cursor() cursor.execute("select * from free_drink_list_config") retVal = cursor.fetchall() for data in retVal: if data['drink_id'] != None: data['drink'] = self.getDrinkPrice(data['drink_id']) data['free_drink_types'] = self.get_free_drink_list_types_for_drink(data['id']) return retVal except Exception as err: traceback.print_exc() self.db.connection.rollback() raise DatabaseExecption("Something went wrong with Database: {}".format(err)) def set_free_drink_list_config(self, free_drink_list_config): try: cursor = self.db.connection.cursor() cursor.execute(f'insert into free_drink_list_config (drink_id, label, price) values ({free_drink_list_config["drink"]["id"]}, "{free_drink_list_config["label"]}", {free_drink_list_config["price"]})') self.db.connection.commit() return self.get_free_drink_list_configs() except Exception as err: traceback.print_exc() self.db.connection.rollback() raise DatabaseExecption("Something went wrong with Database: {}".format(err)) def get_free_drink_list_types(self): try: cursor = self.db.connection.cursor() cursor.execute('select * from free_drink_list_type') return cursor.fetchall() except Exception as err: traceback.print_exc() self.db.connection.rollback() raise DatabaseExecption("Something went wrong with Database: {}".format(err)) def get_free_drink_list_types_for_drink(self, id): try: cursor = self.db.connection.cursor() cursor.execute(f'select a.* from free_drink_list_type a, free_drink_list_type_config b where free_drink_list_config_id={id} and b.free_drink_list_type_id=a.id') return cursor.fetchall() except Exception as err: traceback.print_exc() self.db.connection.rollback() raise DatabaseExecption("Something went wrong with Database: {}".format(err)) def get_free_drink_list_type(self, name): try: cursor = self.db.connection.cursor() if type(name) == str: sql = f'select * from free_drink_list_type where name={name}' elif type(name) == int: sql = f'select * from free_drink_list_type where id={name}' else: raise DatabaseExecption("name as no type int or str. name={}, type={}".format(name, type(name))) cursor.execute(sql) return cursor.fetchone() except Exception as err: traceback.print_exc() self.db.connection.rollback() raise DatabaseExecption("Something went wrong with Database: {}".format(err)) def set_free_drink_list_history(self, user, free_drink_list_config): try: cursor = self.db.connection.cursor() cursor.execute(f'insert into free_drink_list_history (timestamp, free_drink_config_id, user_id, free_drink_type_id) values ("{datetime.now()}", {free_drink_list_config["id"]}, {user.id}, {free_drink_list_config["free_drink_type_id"]})') self.db.connection.commit() return self.get_free_drink_list_history_by_user(user) except Exception as err: traceback.print_exc() self.db.connection.rollback() raise DatabaseExecption("Something went wrong with Database: {}".format(err)) def get_free_drink_list_history_by_user(self, user): try: cursor = self.db.connection.cursor() now = datetime.now() worker = self.getWorker(user, now) if worker: timestamp = worker["startdatetime"] else: timestamp = datetime.now() - timedelta(minutes=30) cursor.execute(f'select * from free_drink_list_history where timestamp>="{timestamp}" and (user_id={user.id} or free_drink_type_id=3)') retVal = cursor.fetchall() for data in retVal: data['timestamp'] = {'year': data['timestamp'].year, 'month': data['timestamp'].month, 'day': data['timestamp'].day, 'hour': data['timestamp'].hour, 'minute': data['timestamp'].minute, 'second': data['timestamp'].second} data['free_drink_config'] = self.get_free_drink_list_config(data['free_drink_config_id']) data['free_drink_type'] = self.get_free_drink_list_type(data['free_drink_type_id']) return retVal except Exception as err: traceback.print_exc() self.db.connection.rollback() raise DatabaseExecption("Something went wrong with Database: {}".format(err)) def update_free_drink_list_history(self, free_drink_list_history): try: cursor = self.db.connection.cursor() cursor.execute(f'update free_drink_list_history set canceled={free_drink_list_history["canceled"]} where id={free_drink_list_history["id"]}') self.db.connection.commit() return True except Exception as err: traceback.print_exc() self.db.connection.rollback() raise DatabaseExecption("Something went wrong with Database: {}".format(err)) def delete_free_drink_list_history(self, free_drink_list_history): try: cursor = self.db.connection.cursor() cursor.execute(f'delete from free_drink_list_history where id={free_drink_list_history["id"]}') self.db.connection.commit() return True except Exception as err: traceback.print_exc() self.db.connection.rollback() raise DatabaseExecption("Something went wrong with Database: {}".format(err))