Delete bot.py
This commit is contained in:
Sergey
GitHub
parent
08a0e1662e
commit
51b3fe5349
642
bot.py
642
bot.py
@ -1,642 +0,0 @@
|
||||
import logging
|
||||
import os
|
||||
import re
|
||||
import redis
|
||||
import requests
|
||||
import telebot
|
||||
import threading
|
||||
import traceback
|
||||
import ujson
|
||||
import vk
|
||||
import wget
|
||||
from PIL import Image
|
||||
from telebot import types
|
||||
|
||||
from credentials import token, vk_app_id
|
||||
from vk_messages import VkMessage, VkPolling
|
||||
|
||||
logging.basicConfig(format='%(levelname)-8s [%(asctime)s] %(message)s', level=logging.WARNING, filename='vk.log')
|
||||
|
||||
vk_threads = {}
|
||||
|
||||
vk_dialogs = {}
|
||||
|
||||
FILE_URL = 'https://api.telegram.org/file/bot{0}/{1}'
|
||||
|
||||
vk_tokens = redis.from_url(os.environ.get("REDIS_URL"))
|
||||
|
||||
currentchat = {}
|
||||
|
||||
bot = telebot.AsyncTeleBot(token)
|
||||
|
||||
link = 'https://oauth.vk.com/authorize?client_id={}&' \
|
||||
'display=page&redirect_uri=https://oauth.vk.com/blank.html&scope=friends,messages,offline,docs,photos,video' \
|
||||
'&response_type=token&v=5.65'.format(vk_app_id)
|
||||
|
||||
|
||||
def get_pages_switcher(markup, page, pages):
|
||||
if page != 0:
|
||||
leftbutton = types.InlineKeyboardButton('◀', callback_data='page{}'.format(page - 1)) # callback
|
||||
else:
|
||||
leftbutton = types.InlineKeyboardButton('Поиск 🔍', callback_data='search')
|
||||
if page + 1 < len(pages):
|
||||
rightbutton = types.InlineKeyboardButton('▶', callback_data='page{}'.format(page + 1))
|
||||
else:
|
||||
rightbutton = None
|
||||
|
||||
if rightbutton:
|
||||
markup.row(leftbutton, rightbutton)
|
||||
else:
|
||||
markup.row(leftbutton)
|
||||
|
||||
|
||||
def replace_shields(text):
|
||||
text = text.replace('<', '<')
|
||||
text = text.replace('>', '>')
|
||||
text = text.replace('&', '&')
|
||||
text = text.replace('©', '©')
|
||||
text = text.replace('®', '®')
|
||||
text = text.replace('«', '«')
|
||||
text = text.replace('»', '«')
|
||||
text = text.replace('°', '°')
|
||||
text = text.replace('™', '™')
|
||||
text = text.replace('±', '±')
|
||||
return text
|
||||
|
||||
|
||||
def request_user_dialogs(session, userid):
|
||||
order = []
|
||||
users_ids = []
|
||||
dialogs = vk.API(session).messages.getDialogs(count=200)
|
||||
for chat in dialogs[1:]:
|
||||
if 'chat_id' in chat:
|
||||
if chat['title'].replace('\\', ''):
|
||||
chat['title'] = chat['title'].replace('\\', '')
|
||||
chat['title'] = replace_shields(chat['title'])
|
||||
order.append({'title': chat['title'], 'id': 'group' + str(chat['chat_id'])})
|
||||
elif chat['uid'] > 0:
|
||||
order.append({'title': None, 'id': chat['uid']})
|
||||
users_ids.append(chat['uid'])
|
||||
users = vk.API(session).users.get(user_ids=users_ids, fields=['first_name', 'last_name', 'uid'])
|
||||
for output in order:
|
||||
if output['title'] == ' ... ' or not output['title']:
|
||||
for x in users:
|
||||
if x['uid'] == output['id']:
|
||||
current_user = x
|
||||
break
|
||||
output['title'] = '{} {}'.format(current_user['first_name'], current_user['last_name'])
|
||||
for button in range(len(order)):
|
||||
order[button] = types.InlineKeyboardButton(order[button]['title'], callback_data=str(order[button]['id']))
|
||||
rows = [order[x:x + 2] for x in range(0, len(order), 2)]
|
||||
pages = [rows[x:x + 4] for x in range(0, len(rows), 4)]
|
||||
vk_dialogs[str(userid)] = pages
|
||||
|
||||
|
||||
def create_markup(message, user, page, edit=False):
|
||||
markup = types.InlineKeyboardMarkup(row_width=2)
|
||||
for i in vk_dialogs[str(user)][page]:
|
||||
markup.row(*i)
|
||||
get_pages_switcher(markup, page, vk_dialogs[str(user)])
|
||||
if edit:
|
||||
bot.edit_message_text(
|
||||
'<b>Выберите Диалог:</b> <code>{}/{}</code> стр.'.format(page + 1, len(vk_dialogs[str(user)])),
|
||||
message.chat.id, message.message_id,
|
||||
parse_mode='HTML', reply_markup=markup).wait()
|
||||
else:
|
||||
bot.send_message(message.from_user.id,
|
||||
'<b>Выберите Диалог:</b> <code>{}/{}</code> стр.'.format(page + 1, len(vk_dialogs[str(user)])),
|
||||
parse_mode='HTML', reply_markup=markup).wait()
|
||||
|
||||
|
||||
def search_users(message, text):
|
||||
session = VkMessage(vk_tokens.get(str(message.from_user.id))).session
|
||||
markup = types.InlineKeyboardMarkup(row_width=1)
|
||||
result = vk.API(session).messages.searchDialogs(q=text, limit=10, fields=[])
|
||||
for chat in result:
|
||||
if chat['type'] == 'profile':
|
||||
markup.add(types.InlineKeyboardButton('{} {}'.format(chat['first_name'], chat['last_name']),
|
||||
callback_data=str(chat['uid'])))
|
||||
elif chat['type'] == 'chat':
|
||||
if chat['title'].replace('\\', ''):
|
||||
chat['title'] = chat['title'].replace('\\', '')
|
||||
markup.add(
|
||||
types.InlineKeyboardButton(replace_shields(chat['title']),
|
||||
callback_data='group' + str(chat['chat_id'])))
|
||||
if markup.keyboard:
|
||||
markup.add(types.InlineKeyboardButton('Поиск 🔍', callback_data='search'))
|
||||
bot.send_message(message.from_user.id, '<b>Результат поиска по</b> <i>{}</i>'.format(text),
|
||||
reply_markup=markup, parse_mode='HTML')
|
||||
else:
|
||||
markup.add(types.InlineKeyboardButton('Поиск 🔍', callback_data='search'))
|
||||
bot.send_message(message.from_user.id, '<b>Ничего не найдено по запросу</b> <i>{}</i>'.format(text),
|
||||
parse_mode='HTML', reply_markup=markup)
|
||||
|
||||
|
||||
@bot.callback_query_handler(func=lambda call: True)
|
||||
def callback_buttons(call):
|
||||
if call.message:
|
||||
if 'page' in call.data:
|
||||
bot.answer_callback_query(call.id).wait()
|
||||
create_markup(call.message, call.from_user.id, int(call.data.split('page')[1]), True)
|
||||
elif 'search' in call.data:
|
||||
markup = types.ForceReply(selective=False)
|
||||
bot.answer_callback_query(call.id, 'Поиск беседы 🔍').wait()
|
||||
bot.send_message(call.from_user.id, '<b>Поиск беседы</b> 🔍',
|
||||
parse_mode='HTML', reply_markup=markup).wait()
|
||||
elif 'group' in call.data:
|
||||
session = VkMessage(vk_tokens.get(str(call.from_user.id))).session
|
||||
chat = vk.API(session).messages.getChat(chat_id=call.data.split('group')[1], fields=[])
|
||||
bot.answer_callback_query(call.id,
|
||||
'Вы в беседе {}'.format(replace_shields(chat['title']))).wait()
|
||||
if chat['title'].replace('\\', ''):
|
||||
chat['title'] = chat['title'].replace('\\', '')
|
||||
bot.send_message(call.from_user.id,
|
||||
'<i>Вы в беседе {}</i>'.format(chat['title']),
|
||||
parse_mode='HTML').wait()
|
||||
currentchat[str(call.from_user.id)] = call.data
|
||||
elif call.data.isdigit():
|
||||
session = VkMessage(vk_tokens.get(str(call.from_user.id))).session
|
||||
user = vk.API(session).users.get(user_ids=call.data, fields=[])[0]
|
||||
bot.answer_callback_query(call.id,
|
||||
'Вы в чате с {} {}'.format(user['first_name'], user['last_name'])).wait()
|
||||
bot.send_message(call.from_user.id,
|
||||
'<i>Вы в чате с {} {}</i>'.format(user['first_name'], user['last_name']),
|
||||
parse_mode='HTML').wait()
|
||||
currentchat[str(call.from_user.id)] = call.data
|
||||
|
||||
|
||||
def create_thread(uid, vk_token):
|
||||
a = VkPolling()
|
||||
t = threading.Thread(name='vk' + str(uid), target=a.run, args=(VkMessage(vk_token), bot, uid,))
|
||||
t.setDaemon(True)
|
||||
t.start()
|
||||
vk_threads[str(uid)] = a
|
||||
vk_tokens.set(str(uid), vk_token)
|
||||
|
||||
|
||||
def check_thread(uid):
|
||||
for th in threading.enumerate():
|
||||
if th.getName() == 'vk' + str(uid):
|
||||
return False
|
||||
return True
|
||||
|
||||
|
||||
# Creating VkPolling threads and dialogs info after bot reboot using existing tokens
|
||||
for uid in vk_tokens.scan_iter():
|
||||
if check_thread(uid.decode("utf-8")):
|
||||
create_thread(uid.decode("utf-8"), vk_tokens.get(uid))
|
||||
request_user_dialogs(VkMessage(vk_tokens.get(uid.decode("utf-8"))).session, uid.decode("utf-8"))
|
||||
|
||||
|
||||
def stop_thread(message):
|
||||
for th in threading.enumerate():
|
||||
if th.getName() == 'vk' + str(message.from_user.id):
|
||||
t = vk_threads[str(message.from_user.id)]
|
||||
t.terminate()
|
||||
th.join()
|
||||
vk_tokens.delete(str(message.from_user.id))
|
||||
vk_dialogs.pop(str(message.from_user.id), None)
|
||||
currentchat.pop(str(message.from_user.id), None)
|
||||
|
||||
|
||||
def extract_unique_code(text):
|
||||
# Extracts the unique_code from the sent /start command.
|
||||
try:
|
||||
return text[45:].split('&')[0]
|
||||
except:
|
||||
return None
|
||||
|
||||
|
||||
def verifycode(code):
|
||||
session = vk.Session(access_token=code)
|
||||
api = vk.API(session)
|
||||
return dict(api.account.getProfileInfo(fields=[]))
|
||||
|
||||
|
||||
def info_extractor(info):
|
||||
info = info[-1].url[8:-1].split('.')
|
||||
return info
|
||||
|
||||
|
||||
@bot.message_handler(commands=['chat'])
|
||||
def chat_command(message):
|
||||
if logged(message):
|
||||
if str(message.from_user.id) in currentchat:
|
||||
if 'group' in currentchat[str(message.from_user.id)]:
|
||||
session = VkMessage(vk_tokens.get(str(message.from_user.id))).session
|
||||
chat = vk.API(session).messages.getChat(
|
||||
chat_id=currentchat[str(message.from_user.id)].split('group')[1],
|
||||
fields=[])
|
||||
if chat['title'].replace('\\', ''):
|
||||
chat['title'] = chat['title'].replace('\\', '')
|
||||
bot.send_message(message.from_user.id,
|
||||
'<i>Вы в беседе {}</i>'.format(chat['title']),
|
||||
parse_mode='HTML').wait()
|
||||
else:
|
||||
session = VkMessage(vk_tokens.get(str(message.from_user.id))).session
|
||||
user = vk.API(session).users.get(user_ids=currentchat[str(message.from_user.id)], fields=[])[0]
|
||||
bot.send_message(message.from_user.id,
|
||||
'<i>Вы в чате с {} {}</i>'.format(user['first_name'], user['last_name']),
|
||||
parse_mode='HTML').wait()
|
||||
else:
|
||||
bot.send_message(message.from_user.id,
|
||||
'<i>Вы не находитесь в чате</i>',
|
||||
parse_mode='HTML').wait()
|
||||
|
||||
|
||||
@bot.message_handler(commands=['leave'])
|
||||
def leave_command(message):
|
||||
if logged(message):
|
||||
if str(message.from_user.id) in currentchat:
|
||||
currentchat.pop(str(message.from_user.id), None)
|
||||
bot.send_message(message.from_user.id,
|
||||
'<i>Вы вышли из чата</i>',
|
||||
parse_mode='HTML').wait()
|
||||
else:
|
||||
bot.send_message(message.from_user.id,
|
||||
'<i>Вы не находитесь в чате</i>',
|
||||
parse_mode='HTML').wait()
|
||||
|
||||
|
||||
@bot.message_handler(commands=['dialogs'])
|
||||
def dialogs_command(message):
|
||||
if logged(message):
|
||||
session = VkMessage(vk_tokens.get(str(message.from_user.id))).session
|
||||
request_user_dialogs(session, message.from_user.id)
|
||||
create_markup(message, message.from_user.id, 0)
|
||||
|
||||
|
||||
@bot.message_handler(commands=['search'])
|
||||
def search_command(message):
|
||||
if logged(message):
|
||||
markup = types.ForceReply(selective=False)
|
||||
if telebot.util.extract_arguments(message.text):
|
||||
search_users(message, telebot.util.extract_arguments(message.text))
|
||||
else:
|
||||
bot.send_message(message.from_user.id, '<b>Поиск беседы</b> 🔍',
|
||||
parse_mode='HTML', reply_markup=markup).wait()
|
||||
|
||||
|
||||
@bot.message_handler(commands=['stop'])
|
||||
def stop_command(message):
|
||||
if not check_thread(message.from_user.id):
|
||||
stop_thread(message)
|
||||
bot.send_message(message.from_user.id, 'Успешный выход!').wait()
|
||||
else:
|
||||
bot.send_message(message.from_user.id, 'Вход не был выполнен!').wait()
|
||||
|
||||
|
||||
@bot.message_handler(commands=['start'])
|
||||
def start_command(message):
|
||||
if check_thread(message.from_user.id):
|
||||
mark = types.InlineKeyboardMarkup()
|
||||
login = types.InlineKeyboardButton('ВХОД', url=link)
|
||||
mark.add(login)
|
||||
bot.send_message(message.from_user.id,
|
||||
'Привет, этот бот поможет тебе общаться ВКонтакте, войди по кнопке ниже'
|
||||
' и отправь мне то, что получишь в адресной строке.',
|
||||
reply_markup=mark).wait()
|
||||
else:
|
||||
bot.send_message(message.from_user.id, 'Вход уже выполнен!\n/stop для выхода.').wait()
|
||||
|
||||
|
||||
def form_request(message, method, info):
|
||||
if int(info[2]):
|
||||
if message.text and message.text.startswith('!'):
|
||||
if len(message.text) - 1:
|
||||
message.text = message.text[1:]
|
||||
if info[2] != 'None':
|
||||
method(message, info[1], group=True, forward_messages=info[2])
|
||||
else:
|
||||
method(message, info[1], group=True)
|
||||
elif message.caption and message.caption.startswith('!'):
|
||||
if len(message.caption) - 1:
|
||||
message.caption = message.caption[1:]
|
||||
if info[2] != 'None':
|
||||
method(message, info[1], group=True, forward_messages=info[2])
|
||||
else:
|
||||
method(message, info[1], group=True)
|
||||
else:
|
||||
if message.text and message.text.startswith('!'):
|
||||
if len(message.text) - 1:
|
||||
message.text = message.text[1:]
|
||||
if info[1] != 'None':
|
||||
method(message, info[0], group=False, forward_messages=info[1])
|
||||
else:
|
||||
method(message, info[0], group=False)
|
||||
elif message.caption and message.caption.startswith('!'):
|
||||
if len(message.caption) - 1:
|
||||
message.caption = message.caption[1:]
|
||||
if info[1] != 'None':
|
||||
method(message, info[0], group=False, forward_messages=info[1])
|
||||
else:
|
||||
method(message, info[0], group=False)
|
||||
else:
|
||||
method(message, info[0], group=False)
|
||||
|
||||
|
||||
def logged(message):
|
||||
if vk_tokens.get(str(message.from_user.id)):
|
||||
return True
|
||||
else:
|
||||
bot.send_message(message.from_user.id, 'Вход не выполнен! /start для входа').wait()
|
||||
return False
|
||||
|
||||
|
||||
def vk_sender(message, method):
|
||||
if logged(message):
|
||||
if message.reply_to_message:
|
||||
info = info_extractor(message.reply_to_message.entities)
|
||||
if info is not None:
|
||||
form_request(message, method, info)
|
||||
|
||||
elif str(message.from_user.id) in currentchat:
|
||||
info = []
|
||||
if 'group' in currentchat[str(message.from_user.id)]:
|
||||
info.append('0')
|
||||
info.append(currentchat[str(message.from_user.id)].split('group')[1])
|
||||
info.append('1')
|
||||
else:
|
||||
info.append(currentchat[str(message.from_user.id)])
|
||||
info.append('0')
|
||||
info.append('0')
|
||||
form_request(message, method, info)
|
||||
|
||||
|
||||
def audio_title_creator(message, performer=None, title=None):
|
||||
if not performer and not title:
|
||||
return 'Аудио_{}'.format(str(message.date)[5:])
|
||||
else:
|
||||
return '{} - {}'.format(performer, title)
|
||||
|
||||
|
||||
def send_text(message, userid, group, forward_messages=None):
|
||||
session = VkMessage(vk_tokens.get(str(message.from_user.id))).session
|
||||
if group:
|
||||
vk.API(session).messages.send(chat_id=userid, message=message.text, forward_messages=forward_messages)
|
||||
else:
|
||||
vk.API(session).messages.send(user_id=userid, message=message.text, forward_messages=forward_messages)
|
||||
|
||||
|
||||
def send_doc(message, userid, group, forward_messages=None):
|
||||
filetype = message.content_type
|
||||
session = VkMessage(vk_tokens.get(str(message.from_user.id))).session
|
||||
file = wget.download(
|
||||
FILE_URL.format(token, bot.get_file(getattr(message, filetype).file_id).wait().file_path))
|
||||
if filetype == 'document':
|
||||
openedfile = open(file, 'rb')
|
||||
files = {'file': openedfile}
|
||||
fileonserver = ujson.loads(requests.post(vk.API(session).docs.getUploadServer()['upload_url'],
|
||||
files=files).text)
|
||||
attachment = vk.API(session).docs.save(file=fileonserver['file'],
|
||||
title=getattr(message, filetype).file_name,
|
||||
tags='')
|
||||
openedfile.close()
|
||||
os.remove(file)
|
||||
|
||||
elif filetype == 'voice':
|
||||
openedfile = open(file, 'rb')
|
||||
files = {'file': openedfile}
|
||||
fileonserver = ujson.loads(requests.post(vk.API(session).docs.getUploadServer()['upload_url'],
|
||||
files=files).text)
|
||||
attachment = vk.API(session).docs.save(file=fileonserver['file'], title='Аудиосообщение',
|
||||
tags='')
|
||||
openedfile.close()
|
||||
os.remove(file)
|
||||
|
||||
else: # filetype == 'audio':
|
||||
newfile = file.split('.')[0] + '.aac'
|
||||
os.rename(file, newfile)
|
||||
openedfile = open(newfile, 'rb')
|
||||
files = {'file': openedfile}
|
||||
fileonserver = ujson.loads(requests.post(vk.API(session).docs.getUploadServer()['upload_url'],
|
||||
files=files).text)
|
||||
attachment = vk.API(session).docs.save(file=fileonserver['file'],
|
||||
title=audio_title_creator(message, message.audio.performer,
|
||||
message.audio.title), tags='')
|
||||
openedfile.close()
|
||||
os.remove(newfile)
|
||||
|
||||
if group:
|
||||
if message.caption:
|
||||
|
||||
vk.API(session).messages.send(chat_id=userid, message=message.caption,
|
||||
attachment='doc{}_{}'.format(attachment[0]['owner_id'],
|
||||
attachment[0]['did']),
|
||||
forward_messages=forward_messages)
|
||||
else:
|
||||
vk.API(session).messages.send(chat_id=userid,
|
||||
attachment='doc{}_{}'.format(attachment[0]['owner_id'],
|
||||
attachment[0]['did']),
|
||||
forward_messages=forward_messages)
|
||||
else:
|
||||
if message.caption:
|
||||
vk.API(session).messages.send(user_id=userid, message=message.caption,
|
||||
attachment='doc{}_{}'.format(attachment[0]['owner_id'],
|
||||
attachment[0]['did']),
|
||||
forward_messages=forward_messages)
|
||||
else:
|
||||
vk.API(session).messages.send(user_id=userid,
|
||||
attachment='doc{}_{}'.format(attachment[0]['owner_id'],
|
||||
attachment[0]['did']),
|
||||
forward_messages=forward_messages)
|
||||
|
||||
|
||||
def send_photo(message, userid, group, forward_messages=None):
|
||||
filetype = message.content_type
|
||||
session = VkMessage(vk_tokens.get(str(message.from_user.id))).session
|
||||
file = wget.download(
|
||||
FILE_URL.format(token, bot.get_file(getattr(message, filetype)[-1].file_id).wait().file_path))
|
||||
openedfile = open(file, 'rb')
|
||||
files = {'file': openedfile}
|
||||
fileonserver = ujson.loads(requests.post(vk.API(session).photos.getMessagesUploadServer()['upload_url'],
|
||||
files=files).text)
|
||||
attachment = vk.API(session).photos.saveMessagesPhoto(server=fileonserver['server'], photo=fileonserver['photo'],
|
||||
hash=fileonserver['hash'])
|
||||
if group:
|
||||
if message.caption:
|
||||
vk.API(session).messages.send(chat_id=userid, message=message.caption, attachment=attachment[0]['id'],
|
||||
forward_messages=forward_messages)
|
||||
else:
|
||||
vk.API(session).messages.send(chat_id=userid, attachment=attachment[0]['id'],
|
||||
forward_messages=forward_messages)
|
||||
else:
|
||||
if message.caption:
|
||||
vk.API(session).messages.send(user_id=userid, message=message.caption, attachment=attachment[0]['id'],
|
||||
forward_messages=forward_messages)
|
||||
else:
|
||||
vk.API(session).messages.send(user_id=userid, attachment=attachment[0]['id'],
|
||||
forward_messages=forward_messages)
|
||||
openedfile.close()
|
||||
os.remove(file)
|
||||
|
||||
|
||||
def send_sticker(message, userid, group, forward_messages=None):
|
||||
filetype = message.content_type
|
||||
session = VkMessage(vk_tokens.get(str(message.from_user.id))).session
|
||||
file = wget.download(
|
||||
FILE_URL.format(token, bot.get_file(getattr(message, filetype).file_id).wait().file_path))
|
||||
Image.open(file).save("{}.png".format(file))
|
||||
openedfile = open('{}.png'.format(file), 'rb')
|
||||
files = {'file': openedfile}
|
||||
fileonserver = ujson.loads(requests.post(vk.API(session).photos.getMessagesUploadServer()['upload_url'],
|
||||
files=files).text)
|
||||
attachment = vk.API(session).photos.saveMessagesPhoto(server=fileonserver['server'], photo=fileonserver['photo'],
|
||||
hash=fileonserver['hash'])
|
||||
if group:
|
||||
if message.caption:
|
||||
vk.API(session).messages.send(chat_id=userid, message=message.caption, attachment=attachment[0]['id'],
|
||||
forward_messages=forward_messages)
|
||||
else:
|
||||
vk.API(session).messages.send(chat_id=userid, attachment=attachment[0]['id'],
|
||||
forward_messages=forward_messages)
|
||||
else:
|
||||
if message.caption:
|
||||
vk.API(session).messages.send(user_id=userid, message=message.caption, attachment=attachment[0]['id'],
|
||||
forward_messages=forward_messages)
|
||||
else:
|
||||
vk.API(session).messages.send(user_id=userid, attachment=attachment[0]['id'],
|
||||
forward_messages=forward_messages)
|
||||
openedfile.close()
|
||||
os.remove('{}.png'.format(file))
|
||||
os.remove(file)
|
||||
|
||||
|
||||
def send_video(message, userid, group, forward_messages=None):
|
||||
filetype = message.content_type
|
||||
session = VkMessage(vk_tokens.get(str(message.from_user.id))).session
|
||||
|
||||
file = wget.download(
|
||||
FILE_URL.format(token, bot.get_file(getattr(message, filetype).file_id).wait().file_path))
|
||||
openedfile = open(file, 'rb')
|
||||
files = {'video_file': openedfile}
|
||||
|
||||
if group:
|
||||
attachment = vk.API(session).video.save(is_private=1)
|
||||
fileonserver = ujson.loads(requests.post(attachment['upload_url'],
|
||||
files=files).text)
|
||||
video = 'video{}_{}'.format(attachment['owner_id'], attachment['owner_id']['video_id'])
|
||||
if message.caption:
|
||||
vk.API(session).messages.send(chat_id=userid, message=message.caption, attachment=video,
|
||||
forward_messages=forward_messages)
|
||||
else:
|
||||
vk.API(session).messages.send(chat_id=userid, attachment=video, forward_messages=forward_messages)
|
||||
else:
|
||||
attachment = vk.API(session).video.save(is_private=1)
|
||||
fileonserver = ujson.loads(requests.post(attachment['upload_url'],
|
||||
files=files).text)
|
||||
video = 'video{}_{}'.format(attachment['owner_id'], attachment['vid'])
|
||||
if message.caption:
|
||||
vk.API(session).messages.send(user_id=userid, message=message.caption, attachment=video,
|
||||
forward_messages=forward_messages)
|
||||
else:
|
||||
vk.API(session).messages.send(user_id=userid, attachment=video, forward_messages=forward_messages)
|
||||
openedfile.close()
|
||||
os.remove(file)
|
||||
|
||||
|
||||
def send_contact(message, userid, group, forward_messages=None):
|
||||
session = VkMessage(vk_tokens.get(str(message.from_user.id))).session
|
||||
if message.contact.last_name:
|
||||
text = 'Контакт: {} {}'.format(message.contact.first_name, message.contact.last_name)
|
||||
else:
|
||||
text = 'Контакт: {}'.format(message.contact.first_name)
|
||||
if group:
|
||||
vk.API(session).messages.send(chat_id=userid, message=text, forward_messages=forward_messages)
|
||||
vk.API(session).messages.send(chat_id=userid, message=message.contact, forward_messages=forward_messages)
|
||||
else:
|
||||
vk.API(session).messages.send(user_id=userid, message=text, forward_messages=forward_messages)
|
||||
vk.API(session).messages.send(chat_id=userid, message=message.contact, forward_messages=forward_messages)
|
||||
|
||||
|
||||
@bot.message_handler(content_types=['document', 'voice', 'audio'])
|
||||
def reply_document(message):
|
||||
try:
|
||||
vk_sender(message, send_doc)
|
||||
except:
|
||||
bot.reply_to(message, 'Файл слишком большой, максимально допустимый размер *20мб*!',
|
||||
parse_mode='Markdown').wait()
|
||||
|
||||
|
||||
@bot.message_handler(content_types=['sticker'])
|
||||
def reply_sticker(message):
|
||||
try:
|
||||
vk_sender(message, send_sticker)
|
||||
except Exception:
|
||||
bot.reply_to(message, '*Произошла неизвестная ошибка при отправке*',
|
||||
parse_mode='Markdown').wait() # TODO?: Bugreport system
|
||||
print('Error: {}'.format(traceback.format_exc()))
|
||||
|
||||
|
||||
@bot.message_handler(content_types=['photo'])
|
||||
def reply_photo(message):
|
||||
try:
|
||||
vk_sender(message, send_photo)
|
||||
except:
|
||||
bot.send_message(message.from_user.id, 'Фото слишком большое, максимально допустимый размер *20мб*!',
|
||||
parse_mode='Markdown').wait()
|
||||
|
||||
|
||||
@bot.message_handler(content_types=['video', 'video_note'])
|
||||
def reply_video(message):
|
||||
try:
|
||||
vk_sender(message, send_video)
|
||||
except:
|
||||
bot.reply_to(message, 'Файл слишком большой, максимально допустимый размер *20мб*!',
|
||||
parse_mode='Markdown').wait()
|
||||
|
||||
|
||||
@bot.message_handler(content_types=['contact'])
|
||||
def reply_contact(message):
|
||||
try:
|
||||
vk_sender(message, send_contact)
|
||||
except Exception:
|
||||
bot.reply_to(message, '*Произошла неизвестная ошибка при отправке*',
|
||||
parse_mode='Markdown').wait()
|
||||
print('Error: {}'.format(traceback.format_exc()))
|
||||
|
||||
|
||||
@bot.message_handler(content_types=['text'])
|
||||
def reply_text(message):
|
||||
m = re.search('https://oauth\.vk\.com/blank\.html#access_token=[a-z0-9]*&expires_in=[0-9]*&user_id=[0-9]*',
|
||||
message.text)
|
||||
if m:
|
||||
code = extract_unique_code(m.group(0))
|
||||
if check_thread(message.from_user.id):
|
||||
try:
|
||||
verifycode(code)
|
||||
create_thread(message.from_user.id, code)
|
||||
bot.send_message(message.from_user.id, 'Вход выполнен!').wait()
|
||||
# ---------------- INSTRUCTIONS ---------------- #
|
||||
bot.send_message(message.from_user.id, 'Бот позволяет получать и отвечать на текстовые сообщения'
|
||||
' из ВКонтакте\nПример личного сообщения:').wait()
|
||||
bot.send_message(message.from_user.id, '*Иван Петров:*\nПривет, я тут классный мессенджер нашёл,'
|
||||
' попробуешь? telegram.org/download',
|
||||
parse_mode='Markdown').wait()
|
||||
bot.send_message(message.from_user.id, 'Для сообщений из групповых чатов будет указываться'
|
||||
' чат после имени отправителя:').wait()
|
||||
bot.send_message(message.from_user.id, '*Ник Невидов @ My English is perfect:*\n'
|
||||
'London is the capital of Great Britain',
|
||||
parse_mode='Markdown').wait()
|
||||
bot.send_message(message.from_user.id, 'Чтобы ответить, используй Reply на нужное сообщение.'
|
||||
' (нет, на эти не сработает, нужно реальное)',
|
||||
parse_mode='Markdown').wait()
|
||||
# ---------------- INSTRUCTIONS ---------------- #
|
||||
except:
|
||||
bot.send_message(message.from_user.id, 'Неверная ссылка, попробуй ещё раз!').wait()
|
||||
else:
|
||||
bot.send_message(message.from_user.id, 'Вход уже выполнен!\n/stop для выхода.').wait()
|
||||
return
|
||||
if message.reply_to_message and message.reply_to_message.text == 'Поиск беседы 🔍':
|
||||
search_users(message, message.text)
|
||||
return
|
||||
|
||||
try:
|
||||
vk_sender(message, send_text)
|
||||
except Exception:
|
||||
bot.reply_to(message, 'Произошла неизвестная ошибка при отправке',
|
||||
parse_mode='Markdown').wait()
|
||||
print('Error: {}'.format(traceback.format_exc()))
|
||||
|
||||
|
||||
bot.polling(none_stop=True)
|
||||
Loading…
x
Reference in New Issue
Block a user