mirror of
https://github.com/flynx/photographer.ru-bot-tg2vk.git
synced 2025-10-28 02:30:09 +00:00
190 lines
4.7 KiB
Python
Executable File
190 lines
4.7 KiB
Python
Executable File
#!/usr/bin/python3
|
|
#----------------------------------------------------------------------
|
|
#
|
|
# systemd service...
|
|
#
|
|
# see:
|
|
# https://tecadmin.net/run-shell-script-as-systemd-service/
|
|
# https://tecadmin.net/setup-autorun-python-script-using-systemd/
|
|
#
|
|
#
|
|
#----------------------------------------------------------------------
|
|
|
|
import telebot
|
|
import vk_api
|
|
import requests
|
|
import os
|
|
import io
|
|
import datetime
|
|
import re
|
|
|
|
|
|
#----------------------------------------------------------------------
|
|
# Helpers...
|
|
|
|
# XXX HACK -- mini/cute html2 -> markdown parser...
|
|
def html2md(text):
|
|
text = re.sub(r'<b>(.*)</b>', r'*\1*', text, flags=re.I|re.M)
|
|
text = re.sub(r'<i>(.*)</i>', r'_\1_', text, flags=re.I|re.M)
|
|
text = re.sub(r'<a\s+[^>]*href="(.*)"[^>]*>(.*)</a>', r'[\2](\1)', text, flags=re.I|re.M)
|
|
return text
|
|
|
|
def load(name):
|
|
return open(os.path.join(BASE_PATH, name), 'r').read().strip()
|
|
|
|
def getDate():
|
|
return datetime.datetime.now().strftime('%Y%m%d %H:%M:%S')
|
|
|
|
|
|
#----------------------------------------------------------------------
|
|
|
|
BASE_PATH = os.path.dirname(
|
|
os.path.abspath(__file__))
|
|
|
|
|
|
telegram_token = load('telegram.token')
|
|
vk_token = load('vk.token')
|
|
vk_group_id = load('vk_group.id')
|
|
|
|
|
|
TEXT_SNIP = 50
|
|
|
|
|
|
# Content types supported:
|
|
# "text",
|
|
# "audio",
|
|
# "document",
|
|
# "photo",
|
|
# "sticker",
|
|
# "video",
|
|
# "video_note",
|
|
# "voice",
|
|
# "location",
|
|
# "contact",
|
|
# "new_chat_members",
|
|
# "left_chat_member",
|
|
# "new_chat_title",
|
|
# "new_chat_photo",
|
|
# "delete_chat_photo",
|
|
# "group_chat_created",
|
|
# "supergroup_chat_created",
|
|
# "channel_chat_created",
|
|
# "migrate_to_chat_id",
|
|
# "migrate_from_chat_id",
|
|
# "pinned_message",
|
|
TRACK_TYPES = [
|
|
"text",
|
|
"photo",
|
|
]
|
|
|
|
|
|
bot = telebot.TeleBot(telegram_token)
|
|
vk_session = vk_api.VkApi(token=vk_token)
|
|
vk = vk_session.get_api()
|
|
|
|
|
|
|
|
|
|
#----------------------------------------------------------------------
|
|
# Text post handler...
|
|
#
|
|
# XXX TODO:
|
|
# - for some reason this does not see attached images at all...
|
|
# message.photo == None...
|
|
# - get telegram formatting (markdown???)
|
|
# (see notes for: repost_photo_to_vk(..) below)
|
|
@bot.channel_post_handler(content_types=['text'])
|
|
def repost_text_to_vk(message):
|
|
print(f"{getDate()}: Received message: {(message.text or '')[:TEXT_SNIP]}")
|
|
|
|
attachments = []
|
|
|
|
# XXX for some reason we are not getting any photos here...
|
|
print(f" --- PHOTOS: {message.photo}")
|
|
if message.photo:
|
|
for photo in message.photo:
|
|
# Скачиваем фото
|
|
file_info = bot.get_file(photo.file_id)
|
|
photo_url = "https://api.telegram.org/file/bot{}/{}".format(
|
|
telegram_token, file_info.file_path)
|
|
|
|
# Загружаем фото в ВКонтакте
|
|
photo_id = upload_photo_to_vk(vk, photo_url)
|
|
attachments.append(photo_id)
|
|
|
|
print(f" --- PHOTO {len(attachments)}: {attachments[-1]}")
|
|
|
|
# Публикуем сообщение в группе ВКонтакте
|
|
vk.wall.post(
|
|
owner_id="-" + vk_group_id,
|
|
message=message.text,
|
|
attachments=",".join(attachments),
|
|
)
|
|
print(f"{getDate()}: Posted message: {(message.text or '')[:TEXT_SNIP]}")
|
|
|
|
|
|
|
|
#----------------------------------------------------------------------
|
|
# Image post handler...
|
|
#
|
|
# see:
|
|
# https://github.com/python273/vk_api/blob/master/examples/upload_photo.py
|
|
# https://vk-api.readthedocs.io/en/latest/upload.html (see: photo_wall(..))
|
|
#
|
|
# XXX TODO
|
|
# - multiple images per post are treated as separate posts for some reason -- need
|
|
# to find a way to group...
|
|
# - get telegram formatting:
|
|
# - links
|
|
# - header
|
|
# - ...
|
|
# - Q: is the largest image alwways the last??
|
|
# ...check telegram API
|
|
@bot.channel_post_handler(content_types=['photo'])
|
|
def repost_photo_to_vk(message):
|
|
print(f"{getDate()}: Received photo: {(message.caption or '')[:TEXT_SNIP]}")
|
|
|
|
# download images (memory)...
|
|
upload = vk_api.VkUpload(vk_session)
|
|
ids = []
|
|
photo = message.photo[-1]
|
|
file_info = bot.get_file(photo.file_id)
|
|
photo_url = "https://api.telegram.org/file/bot{}/{}".format(
|
|
telegram_token,
|
|
file_info.file_path)
|
|
response = requests.get(photo_url)
|
|
|
|
# upload images...
|
|
photo_info = upload.photo_wall(
|
|
io.BytesIO(response.content),
|
|
group_id=vk_group_id)[0]
|
|
|
|
ids += [
|
|
"photo{}_{}".format(
|
|
photo_info["owner_id"],
|
|
photo_info["id"]) ]
|
|
|
|
print(f" --- PHOTO {len(ids)}: {ids[-1]}")
|
|
|
|
vk.wall.post(
|
|
owner_id="-" + vk_group_id,
|
|
message=message.caption,
|
|
#message=message.html_caption,
|
|
attachments=",".join(ids))
|
|
|
|
print(f"{getDate()}: Posted photo: {(message.caption or '')[:TEXT_SNIP]}")
|
|
|
|
|
|
|
|
#----------------------------------------------------------------------
|
|
# Start...
|
|
|
|
print(f'{getDate()}: Bot started...')
|
|
|
|
bot.polling(none_stop=False, interval=0, timeout=60)
|
|
|
|
|
|
|
|
#----------------------------------------------------------------------
|
|
# vim:set ts=4 sw=4 :
|