def buy_prod(call):
user, c = db.get_user_date(call.message.chat)
save = db.get_user_date(user)
prod = db.Product.get(db.Product.id == save.get("prod_id"))
order, c = db.Order.get_or_create(user=user, product=prod)
if order.count is None:
order.count = 1
else:
order.count += 1
order.save()
bot.answer_callback_query(callback_query_id=call.id,
show_alert=False,
text=u"Товар добавлен в корзину")
def rename_category(call):
user, c = db.get_user(call.message.chat)
save = db.get_user_date(user)
cat = db.Category.get(db.Category.id == save.get("cat_id"))
msg = bot.send_message(call.message.chat.id,
u"Пришлите новое имя категории " + cat.name)
bot.register_next_step_handler(msg, ut.edit_cat_name)
def edit_prod_category(call):
user, c = db.get_user(call.message.chat)
save = db.get_user_date(user)
key = cat_list("edit>prod>setcat>", prod=save.get("prod_id"))
bot.send_message(call.message.chat.id,
"Выбирите категорию",
reply_markup=key)
def edit_prod_set_category(call):
user, c = db.get_user(call.message.chat)
save = db.get_user_date(user)
prod = db.Product.get(db.Product.id == save.get("prod_id"))
cat = db.Category.get(db.Category.id == save.get("cat_id"))
prod.category = cat
prod.save()
ut.send_product(call.message.chat, prod.id)
def category(call):
user, cre = db.get_user(call.message.chat)
save = db.get_user_date(user)
if user.is_admin:
key = prod_list(save.get("id"), is_admin=True)
else:
key = prod_list(save.get('id'))
save = db.get_user_date(user)
db.set_user_data(user, save)
if user.is_admin:
key.add(
telebot.types.InlineKeyboardButton(text="Добавить товар",
callback_data="add>product>" +
save.get("id")))
key.add(
telebot.types.InlineKeyboardButton(
text="Переименовать категорию",
callback_data="edit>category>rename>" + save.get("id")))
key.add(
telebot.types.InlineKeyboardButton(text="Категории",
callback_data="shop"))
bot.send_message(call.message.chat.id, "Выберете товар:", reply_markup=key)
def del_prod_sclad(call):
user, c = db.get_user(call.message.chat)
save = db.get_user_date(user)
prod = db.Product.get(db.Product.id == save.get("prod_id"))
for ord in db.Order.select().where(db.Order.product == prod):
ord.delete_instance()
cat = prod.category.id
name = prod.title
prod.delete_instance()
key = telebot.types.InlineKeyboardMarkup()
key.add(
telebot.types.InlineKeyboardButton(text="К Категории",
callback_data="category>" +
str(cat)))
bot.send_message(call.message.chat.id,
text=u"Товар " + name + u" успешно удален",
reply_markup=key)
def del_category(call):
user, c = db.get_user(call.message.chat)
save = db.get_user_date(user)
print save
cat = db.Category.get(db.Category.id == int(save.get("cat_id")))
name = cat.name
for prod in db.Product.select().where(db.Product.category == cat):
prod.delete_instance()
cat.delete_instance()
key = telebot.types.InlineKeyboardMarkup()
key.add(
telebot.types.InlineKeyboardButton(text="Удалить еще категорию",
callback_data="del>category"))
key.add(
telebot.types.InlineKeyboardButton(text="Категории",
callback_data="shop"))
bot.send_message(call.message.chat.id,
u"Категория " + name + u" и все товары в ней удалены")
def view_prod(call):
user, cre = db.get_user(call.message.chat)
save = db.get_user_date(user)
ut.send_product(call.message.chat, save.get("prod_id"))