def unknown_user(message):
""" Handle all messages if user is not identified """
time.sleep(1)
# Check if message text has '@' between some non-space symbols
if not message.text or not re.findall(r"\S+@\S+", message.text):
bot.send_message(message.chat.id, messages.send_email)
return
email = re.findall(r"\S+@\S+", message.text)[0].lower()
# Suppose user entered email, look for it in database
email_info = db_find_value("email", email)
# Email not found, insert new row in DB with that email and user ID
if not email_info:
new_client(email, "tg_id", message.chat.id)
initial_buttons(message)
# Email is already used by user with other ID, ask to immediately contact us
elif email_info['tg_id'] != "0":
new_client("-", "tg_id", message.chat.id)
open_dialogue("tg_id", message.chat.id)
initial_buttons(message, messages.email_already_used)
# Email found in DB and not used by other ID, update DB
else:
update_clients(["email", email], ["tg_id", message.chat.id])
initial_buttons(message)
def support(message, urgent=False):
""" Handles every attempt to open support dialogue. Does not open if not urgent and not in working time """
time.sleep(1)
buttons = types.InlineKeyboardMarkup()
# User trying to contact support in non working time
if not urgent:
if not 17 <= datetime.datetime.today(
).hour < 22 or datetime.datetime.today().isoweekday() in [6, 7]:
buttons.add(
types.InlineKeyboardButton(text="Срочная связь",
callback_data="urgent"))
bot.send_message(message.chat.id,
messages.non_working,
reply_markup=buttons)
return
open_dialogue("tg_id", message.chat.id)
buttons = types.InlineKeyboardMarkup()
buttons.add(
types.InlineKeyboardButton(text="Первичная настройка",
callback_data="install"))
buttons.add(
types.InlineKeyboardButton(text="Другое", callback_data="other"))
buttons.add(
types.InlineKeyboardButton(text="ZGC SHOP", callback_data="market"))
# Ask user to choose problem type
msg = messages.type_support
user_info = db_find_value("tg_id", message.chat.id)
sub = user_info['sub']
if sub != '-' and int(user_info['verified']):
msg += f"\U000026A1 Ваша подписка: {sub}"
bot.send_message(message.chat.id, msg, reply_markup=buttons)
def forward_to_support(message):
""" Forward all clients messages to support group if dialogue is open
Also send to support info about client's tariff """
time.sleep(1)
user_info = db_find_value("tg_id", message.chat.id)
user_state = user_info['state']
# transfer client's message to support group
tg_to_tg(config.group_id, message)
# Client sent a message after being reminded about the payment, open dialogue
if user_state == "REMINDED":
open_dialogue("tg_id", message.chat.id, state="PAY")
if message.photo:
filename = save_file(message,
folder=config.pay_imgs_path,
check=True)
autopay(message.chat.id, 'tg_id', filename)
# Client sent a photo after he has chosen the payment option. Consider this photo as payment screenshot
elif user_state == "PAY":
if message.photo:
filename = save_file(message,
folder=config.pay_imgs_path,
check=True)
autopay(message.chat.id, 'tg_id', filename)
# Notify client that his message was received (once per dialogue)
if user_info['received'] == "NO":
bot.send_message(
message.chat.id, "Ваше сообщение передано в поддержку. "
"Мы постараемся ответить как можно быстрее!")
update_clients(["tg_id", message.chat.id], ["received", "YES"])
def start(message):
""" Ask user to send his email address if not identified. Otherwise send default message with reply keyboard """
time.sleep(1)
if not db_find_value("tg_id", message.chat.id):
bot.send_message(message.chat.id, messages.send_email)
else:
initial_buttons(message)
def coop(message):
""" Handles cooperation button """
time.sleep(1)
buttons = types.InlineKeyboardMarkup()
buttons.add(
types.InlineKeyboardButton(text="Сделать предложение",
url='https://zgcvpn.ru/partnership'))
bot.send_message(message.chat.id, messages.coop, reply_markup=buttons)
def blog(message):
""" Handles blog button """
time.sleep(1)
buttons = types.InlineKeyboardMarkup()
buttons.add(
types.InlineKeyboardButton(text="Блог",
url='https://market.zgc.su/zgcvpnblog'))
bot.send_message(
message.chat.id,
"Узнайте как заблокировать рекламу, какие появились сервера и многое другое",
reply_markup=buttons)
def shop(message):
""" Handles zgc shop button """
time.sleep(1)
buttons = types.InlineKeyboardMarkup()
buttons.add(
types.InlineKeyboardButton(text="\U0001F6D2 ZGC SHOP",
url='https://market.zgc.su'))
buttons.add(
types.InlineKeyboardButton(text="\U00002753 Связаться с поддержкой",
callback_data='market'))
bot.send_message(message.chat.id, messages.shop, reply_markup=buttons)
def tm(message):
""" Handles for Turkmenistan button """
time.sleep(1)
buttons = types.InlineKeyboardMarkup()
buttons.add(
types.InlineKeyboardButton(text="Сайт обслуживания", url='tm.zgc.su'))
buttons.add(
types.InlineKeyboardButton(
text="Как подключить?",
url='https://sites.google.com/view/zgcvpn/try?authuser=0'))
bot.send_message(message.chat.id, messages.turk, reply_markup=buttons)
def trial(message):
""" Handles free trial button """
time.sleep(1)
buttons = types.InlineKeyboardMarkup()
buttons.add(
types.InlineKeyboardButton(text="\U0001F4B4 Оплата",
callback_data="pay"))
buttons.add(
types.InlineKeyboardButton(text="\U00002753 Связаться с поддержкой",
callback_data="sup"))
bot.send_message(message.chat.id,
messages.trial_text,
reply_markup=buttons,
parse_mode='Markdown')
def pay(message):
""" Handles payment button """
time.sleep(1)
open_dialogue("tg_id", message.chat.id, state="PAY")
buttons = types.InlineKeyboardMarkup()
buttons.add(
types.InlineKeyboardButton(text="В рублях ₽ или в гривнах ₴",
callback_data="rub"))
buttons.add(
types.InlineKeyboardButton(text="В юанях ¥", callback_data="yuan"))
buttons.add(
types.InlineKeyboardButton(text="\U00002753 Связаться с поддержкой",
callback_data="sup"))
bot.send_message(message.chat.id, messages.pay_type, reply_markup=buttons)
def one_message_pass(message):
""" User pushed the feedback button after previous support conversation was closed.
Suppose user entering one-message review """
time.sleep(1)
time_past = int(time.time()) - int(
get_info("review_time", "tg_id", message.chat.id))
# If user pushed the button more than a day ago, don't send his message to support
if time_past // 3600 >= 24:
bot.send_message(message.chat.id, messages.buttons_menu)
else:
tg_to_tg(config.group_id, message, review=True)
bot.send_message(message.chat.id, "Спасибо за отзыв!")
update_clients(["tg_id", message.chat.id], ["state", "CLOSED"])
def initial_buttons(message, send_text=messages.buttons_menu):
""" Send user message with default reply keyboard"""
time.sleep(1)
markup_buttons = types.ReplyKeyboardMarkup(resize_keyboard=True)
item_pay = types.InlineKeyboardButton("\U0001F4B4 Оплата")
item_shop = types.InlineKeyboardButton("\U0001F6D2 ZGC SHOP")
item_trial = types.InlineKeyboardButton("\U0001F193 Пробный период")
item_promo = types.InlineKeyboardButton("\U0001F4F0 Узнать больше")
item_turk = types.InlineKeyboardButton(
"\U0001F1F9\U0001F1F2Для Туркменистана")
item_coop = types.InlineKeyboardButton("\U0001F91D Сотрудничество")
item_connection = types.InlineKeyboardButton(
"\U00002753 Связаться с поддержкой")
markup_buttons.add(item_pay, item_trial, item_turk, item_promo, item_shop,
item_coop, item_connection)
bot.send_message(message.chat.id, send_text, reply_markup=markup_buttons)
def tg_to_tg(to_id, message, from_support=False, review=False):
""" Transfers messages from client to support and back
Used instead of TG API forward_message method to be able to reply every client, even those who restricted forwarding """
text = message.text or message.caption or ''
# Add info about client to message
if not from_support:
# Upper part of the message with emoji and name
header = f"\U0001F4E2 Отзыв\n" if review \
else f"\U0001F4AC{message.from_user.first_name} {message.from_user.last_name}\n"
# Bottom part of the message with id and social network name, so we can reply back
check = "\U00002705" if get_info("verified", "tg_id",
message.chat.id) else ''
bottom = f"{message.chat.id} Telegram{check}"
text = header + text + "\n\n"
# Add client tariff info
if not info_soon_check(message.chat.id, 'tg_id'):
text += "\n" + client_info_msg("tg_id", message.chat.id)
text += bottom
if message.text:
bot.send_message(to_id, text)
time.sleep(1)
if message.photo:
photo_max_res = sorted(message.photo,
key=lambda x: x.height)[-1].file_id
bot.send_photo(to_id, photo_max_res, caption=text)
time.sleep(1)
if message.video:
bot.send_video(to_id, message.video.file_id, caption=text)
if message.document:
bot.send_document(to_id, message.document.file_id, caption=text)
if message.voice:
bot.send_voice(to_id, message.voice.file_id)
# Voice message has no caption, so send empty text message with top+bottom borders to be able to reply
# Same for audio and stickers
if not from_support:
bot.send_message(to_id, text)
if message.audio:
bot.send_audio(to_id, message.audio.file_id)
if not from_support:
bot.send_message(to_id, text)
if message.sticker:
bot.send_sticker(to_id, message.sticker.file_id)
if not from_support:
bot.send_message(to_id, text)
def support_group(message):
""" Handle all messages in support group """
time.sleep(1)
# Bot info message
if message.text and message.text.lower() == "/info":
bot.send_message(config.group_id, messages.info)
# Message is reply to some message
if message.reply_to_message:
client_text = message.reply_to_message.text or message.reply_to_message.caption
if not client_text:
pass
# Reply object message was forwarded from tg
elif client_text and client_text.endswith(
"Telegram") or client_text.endswith("Telegram\U00002705"):
tg_id = client_text.split()[-2]
# User id does not fit 'one or more numeral' regexp
if not re.fullmatch(r"[0-9]+", str(tg_id)):
bot.send_message(
config.group_id, "Не удалось отправить сообщение. "
"Скорее всего это закрытый аккаунт без айди в подписи")
return
# Close dialogue
if message.text and message.text.lower() in [
"пока", "/пока", "off", "конец", "/q"
]:
close_dialogue("tg_id", tg_id)
# Close payment dialogue
elif message.text and message.text.lower() in ["/оплата", "оп"]:
close_dialogue("tg_id", tg_id, pay=True)
# Close dialogue silently
elif message.text and message.text.lower() == "/закрыть":
close_dialogue("tg_id", tg_id, silent=True)
# Check if message was forwarded by bot, not by other user
elif message.reply_to_message.from_user.id == config.bot_id:
copy_send_message(
tg_id, message.chat.id,
message.message_id) # Finally, send answer to client :)
open_dialogue("tg_id", tg_id)
# Reply object message was forwarded from VK
elif client_text and client_text.endswith(
"Vkontakte") or client_text.endswith("Vkontakte\U00002705"):
vk_id = client_text.split()[-2]
if message.text and message.text.lower() in [
"пока", "off", "конец", "/q"
]:
close_dialogue("vk_id", vk_id)
elif message.text and message.text.lower() in ["/оплата", "оп"]:
close_dialogue("vk_id", vk_id, pay=True)
elif message.text and message.text.lower() == "/закрыть":
close_dialogue("vk_id", vk_id, silent=True)
else:
tg_to_vk(message, vk_id)
open_dialogue("vk_id", vk_id)
# Reply object message was forwarded from FB
elif client_text and client_text.endswith("Facebook"):
fb_id = client_text.split()[-2]
if message.text and message.text.lower() in [
"пока", "/пока", "off", "конец", "/q"
]:
close_dialogue("fb_id", fb_id)
elif message.text and message.text.lower() in ["/оплата", "оп"]:
close_dialogue("fb_id", fb_id, pay=True)
elif message.text and message.text.lower() == "/закрыть":
close_dialogue("fb_id", fb_id, silent=True)
else:
tg_to_fb(message, fb_id)
# my code
elif client_text and client_text.endswith("Web_client"):
import data_structs as ds
import helpers
replied_message = message.reply_to_message.text
reply_email = helpers.get_email_from_message(replied_message)
ws_message = json.dumps({'from': 'bot', 'message': message.text})
import asyncio
asyncio.run(ds.send_ws_msg(reply_email, ws_message))
# Message is not reply to some message
else:
# Message text is dialogue closing command
if message.text and message.text.lower().split()[0] in [
"пока", "/пока", "off", "конец", "/q"
]:
# If not replying to some message, closing message must be exactly 2 words
# and contain ID of user whose dialogue we want to close
if len(message.text.lower().split()) != 2:
bot.send_message(
message.chat.id,
"Отправьте команду и айди через пробел, например:\n"
"Пока 1234567")
return
# We don't know which messenger this id belongs to, so just try to close this ID for every type
for id_type in ["tg_id", "vk_id", "fb_id"]:
close_dialogue(id_type, message.text.lower().split()[1])
elif message.text and message.text.lower().split()[0] in [
"/оплата", "оп"
]:
if len(message.text.lower().split()) != 2:
bot.send_message(
message.chat.id,
"Отправьте команду и айди через пробел, например:\n"
"/оплата 1234567")
return
for id_type in ["tg_id", "vk_id", "fb_id"]:
close_dialogue(id_type,
message.text.lower().split()[1],
pay=True)
elif message.text and message.text.lower().split()[0] == "/закрыть":
if len(message.text.lower().split()) != 2:
bot.send_message(
message.chat.id,
"Отправьте команду и айди через пробел, например:\n"
"/закрыть 1234567")
return
for id_type in ["tg_id", "vk_id", "fb_id"]:
close_dialogue(id_type,
message.text.lower().split()[1],
silent=True)
def mail_edit(message):
""" Allows bot owner to edit database, deleting row with wrong email and copying IDs to row with correct email
Start with bot command '/почта', then reply to every bot message """
time.sleep(1)
if message.text == "/почта":
temp["true_email"] = ""
temp["wrong_email"] = ""
bot.send_message(message.chat.id, messages.edit_wrong_mail)
return
reply_text = message.reply_to_message.text
# First, you must enter wrong email address, existing in DB
if reply_text == messages.edit_wrong_mail:
wrong_email_info = db_find_value("email", message.text.lower())
# Wrong email not found in DB
if not wrong_email_info:
bot.send_message(message.chat.id, "Такой почты нет в базе!")
return
# Save email info to temp
temp['wrong_email'] = list(wrong_email_info.values())
# Message for the next step
bot.send_message(message.chat.id, messages.edit_true_mail)
# Second, you must enter correct email address, existing in DB
elif reply_text == messages.edit_true_mail:
# No info about wrong email in temp
if not temp['wrong_email']:
bot.send_message(message.chat.id, "Не указан неверный email")
return
true_email_info = db_find_value("email", message.text.lower())
# Correct email not found in DB
if not true_email_info:
bot.send_message(message.chat.id, "Такой почты нет в базе!")
return
temp['true_email'] = list(true_email_info.values())
t = temp['true_email']
w = temp['wrong_email']
# Send two messages with DB info about wrong and correct email so we can check everything
bot.send_message(
message.chat.id,
f"Данные неправильной почты:\nemail = {w[0]}\ndate = {w[1]}\n"
f"tariff = {w[2]}\nsub = {w[3]}\ntg_id = {w[4]}\nvk_id = {w[5]}\nfb_id = {w[6]}"
)
bot.send_message(
message.chat.id,
f"Данные правильной почты:\nemail = {t[0]}\ndate = {t[1]}\n"
f"tariff = {t[2]}\nsub = {t[3]}\ntg_id = {t[4]}\nvk_id = {t[5]}\nfb_id = {t[6]}"
)
# Return if wrong and correct emails have different IDs for the same messenger (tg/vk/fb)
if w[4] != "0" != t[4] and w[4] != t[4] \
or w[5] != "0" != t[5] and w[5] != t[5] \
or w[6] != "0" != t[6] and w[6] != t[6]:
bot.send_message(
message.chat.id,
"У этих почт разные айди для одного способа связи, надо разбираться\n"
)
return
# Send message, asking to confirm editing
bot.send_message(message.chat.id, messages.edit_confirm)
elif reply_text == messages.edit_confirm:
# Editing confirmed
if message.text.lower() == "да":
t = temp['true_email']
w = temp['wrong_email']
# No info about wrong or correct email in temp
if not t or not w:
bot.send_message(message.chat.id, "Не указаны почты")
return
# Record this change to log
with open("edit_log.txt", "a+") as file:
file.write(
f"---------------------------\n{datetime.datetime.today()}\n"
f"wrong email = {w[0]}\ntrue email = {t[0]}\n")
tg_id, vk_id, fb_id = w[4], w[5], w[6]
t_mail = t[0]
update_clients(["email", t_mail], ["state", "CLOSED"],
["rate", "5"], ["review_time", "0"])
# Check if ID that we want to copy != 0 and not the same with correct email ID
if tg_id not in ["0", t[4]]:
update_clients(["email", t_mail], ["tg_id", tg_id])
print(f"Заменили tg_id {t[4]} на {tg_id}\n")
if vk_id not in ["0", t[5]]:
update_clients(["email", t_mail], ["vk_id", vk_id])
print(f"Заменили vk_id {t[5]} на {vk_id}\n")
if fb_id not in ["0", t[6]]:
update_clients(["email", t_mail], ["fb_id", fb_id])
print(f"Заменили fb_id {t[6]} на {fb_id}\n")
bot.send_message(message.chat.id, delete_client("email", w[0]))
print(f"Удалили запись с почтой {w[0]}")
temp['wrong_email'], temp['true_email'] = "", ""
# Editing cancelled
elif message.text.lower() == "нет":
bot.send_message(message.chat.id, "Замена почты отменена")
temp['wrong_email'], temp['true_email'] = "", ""
else:
bot.send_message(message.chat.id, "Не понял, повторите")
def mailing(message):
""" Sends a message to clients with selected tariffs """
time.sleep(1)
txt = message.text or message.caption
if txt == "/рассылка":
bot.send_message(message.chat.id, messages.mailing_tariffs)
return
client_text = message.reply_to_message.text
if client_text == messages.mailing_tariffs:
temp['tariffs'] = [i.lower() for i in txt.split()]
bot.send_message(message.chat.id, messages.mailing_message)
elif client_text == messages.mailing_message:
temp['message_id'] = message.message_id
temp['mail_text'] = txt
bot.send_message(
message.chat.id,
f"Следующие тарифы: {', '.join(temp['tariffs'])}\n"
f"Получат сообщение:\n{txt}\n\n"
f"Продолжить? Да/Нет ответом.\n"
f"Чтобы отправить ТОЛЬКО в телеграм или вконтакте, ответьте тг/вк")
elif client_text.startswith("Следующие тарифы:"):
if txt.lower() in ["да", "вк", "тг"]:
if not temp['tariffs'] or not temp['mail_text']:
bot.send_message(message.chat.id, "Не выбраны тарифы/текст!")
return
if txt.lower() == "вк":
spam(temp['tariffs'],
temp['mail_text'],
message.chat.id,
VK_only=True)
elif txt.lower() == 'тг':
spam(temp['tariffs'],
temp['mail_text'],
message.chat.id,
TG_only=True)
else:
spam(temp['tariffs'], temp['mail_text'], message.chat.id)
bot.send_message(message.chat.id, "Рассылка отправлена")
elif txt.lower() == "нет":
bot.send_message(message.chat.id, "Рассылка отменена")
temp['tariffs'], temp['mail_text'] = [], ''
else:
bot.send_message(message.chat.id, "Не понял, повторите")
def react(call):
""" Handles all callback buttons """
time.sleep(1)
buttons = types.InlineKeyboardMarkup()
user_info = db_find_value('tg_id', call.message.chat.id)
if call.data == "rub":
tariff_id = user_info['tariff']
msg = messages.rub_text
# If user has active tariff, send button allowing to extend his current tariff
# Temporarily off
if False and tariff_id in ['1', '2', '3', '22', '23']:
buttons.add(
types.InlineKeyboardButton(text="Продлить текущий тариф",
callback_data=f"Р-{tariff_id}"))
msg += f"\n\nВаш текущий тариф: {tariff_id}"
bot.send_message(call.message.chat.id,
msg,
parse_mode='Markdown',
reply_markup=buttons)
else:
bot.send_message(call.message.chat.id, msg, parse_mode='Markdown')
elif call.data == "yuan":
bot.send_message(call.message.chat.id,
messages.yuan_text,
parse_mode='Markdown')
elif call.data == "install":
bot.send_message(call.message.chat.id, messages.first_install)
elif call.data == "other":
bot.send_message(call.message.chat.id,
messages.support,
parse_mode='Markdown')
elif call.data == "market":
open_dialogue("tg_id", call.message.chat.id)
bot.send_message(
call.message.chat.id,
'Здравствуйте! Укажите, пожалуйста, продукт и вопросы по нему')
elif call.data == "urgent":
support(call.message, urgent=True)
elif call.data == "sup":
support(call.message)
# If user rated quality less than 5 and pushed feedback button, open dialogue for one message only
elif call.data == "get_better":
update_clients(["tg_id", call.message.chat.id],
["state", "ONE MESSAGE"],
["review_time", f"{int(time.time())}"])
bot.send_message(call.message.chat.id, messages.get_better)
elif call.data == "pay":
buttons.add(
types.InlineKeyboardButton(text="В рублях или в гривнах",
callback_data="rub"))
buttons.add(
types.InlineKeyboardButton(text="В юанях", callback_data="yuan"))
buttons.add(
types.InlineKeyboardButton(
text="\U00002753 Связаться с поддержкой", callback_data="sup"))
bot.send_message(call.message.chat.id,
messages.pay_type,
reply_markup=buttons)
elif call.data in ["1", "2", "3", "4", "5"]:
# User has already rated
if get_info("rate", "tg_id", call.message.chat.id) != "0":
bot.send_message(call.message.chat.id,
"Вы уже поставили оценку, спасибо!")
return
rating = call.data
# Ask user to make review if he gave the highest rate
if rating == "5":
buttons.add(
types.InlineKeyboardButton(text="\U0001F49B Оставить отзыв",
url=config.review_link))
bot.send_message(
call.message.chat.id,
"Если вам понравился наш сервис - оставьте отзыв, "
"и мы предоставим вам 10 дней бесплатного VPN!\n\n"
"_Когда оставите отзыв свяжитесь с нами для получения бонуса_",
reply_markup=buttons,
parse_mode='Markdown')
# Ask user to write feedback
else:
buttons.add(
types.InlineKeyboardButton(
text="\U0001F4A1 Оставить пожелание",
callback_data="get_better"))
bot.send_message(call.message.chat.id,
"Мы можем что-то улучшить в обслуживании?",
reply_markup=buttons)
bot.send_message(
config.group_id,
f"Клиент `{call.message.chat.id}` поставил вам {rating}",
parse_mode='Markdown')
update_clients(["tg_id", call.message.chat.id], ["rate", rating])
# User chose tariff he wants to buy, send him buttons to choose tariff duration in days (ascending order)
# Temporarily off
elif False and re.fullmatch(r'[РЮ]-[123]+', call.data):
currency, tariff_id = call.data.split('-')
tariffs_list = get_tariffs(['tariff', tariff_id],
['currency', currency])
for price, tariff_id, days, currency, desc in sorted(
tariffs_list, key=lambda x: int(x[2])):
# Telegram does not support prices lower than 60 rub.
if int(price) < 60:
continue
# Choose correct form of 'день' word
if days[-1] == '1':
day_word = 'день'
elif days[-1] in '234':
day_word = 'дня'
else:
day_word = 'дней'
buttons.add(
types.InlineKeyboardButton(
text=f"{days} {day_word}",
callback_data=f"{currency}-{tariff_id}-{days}"))
bot.send_message(
call.message.chat.id,
f"Выберите, на сколько дней вы хотите продлить тариф {tariff_id}",
reply_markup=buttons)
# User chose tariff duration in days, send him payment button
# Temporarily off
elif False and re.fullmatch(r'[РЮ]-[123]+-[0-9]+', call.data):
currency, tariff_id, days = call.data.split('-')
specific_tariff = get_tariffs(['tariff', tariff_id],
['currency', currency],
['days', days])[0]
price = int(specific_tariff[0])
prices = [
types.LabeledPrice(label=f'Тариф {tariff_id} на {days} дней',
amount=price * 100)
]
bot.send_invoice(call.message.chat.id,
title=f"Тариф {tariff_id}",
description='Plati',
provider_token=config.pay_token,
currency='rub',
prices=prices,
start_parameter='payment',
invoice_payload='HAPPY FRIDAYS COUPON')