def edit_stock(pk_id, stock_id, description, quantity, category, price): """Takes all stock arguments and save it in the selected stock record. (EDIT THE STOCK) pk_id: primary key id, or stock to be edited. The rest of the arguments are going to be the new data to the specified record. """ stock = Stock.update( stock_id=stock_id, description=description, quantity=quantity, category=category, price=price ).where( Stock.id == pk_id ) return stock.execute()
def add_existing_stock(pk_id, to_add): """Increase quantity for a specified record by pk_id. pk_id: primary key id """ stock = Stock.get( Stock.id == pk_id ) current_quantity = stock.quantity new_quantity = current_quantity + to_add query = Stock.update( quantity=new_quantity ).where( Stock.id == pk_id ) return query.execute()