mirror of
https://github.com/EDeev/vkrepost_to_tg.git
synced 2026-10-07 20:49:50 +03:00
Egor Deev
490ea8b5d6
В репозитории оставался код на aiogram 2. На сервере с апреля–сентября 2024 работала переписанная версия: aiogram 3, разметка HTML вместо Markdown, код разделён на bot.py / handlers.py / init.py. Переводы строк приведены к LF.
109 lines
No EOL
4.8 KiB
Python
109 lines
No EOL
4.8 KiB
Python
import vk_api
|
|
|
|
|
|
class VkParser:
|
|
def __init__(self, token):
|
|
"""Подключаемся к API VK"""
|
|
self.session = vk_api.VkApi(token=token)
|
|
self.vk = self.session.get_api()
|
|
|
|
def check(self):
|
|
return self.vk.account.getInfo(fields="lang")
|
|
|
|
def like(self, owner_id, post_id):
|
|
return self.vk.likes.add(type="post", owner_id=owner_id, item_id=post_id)
|
|
|
|
def login(self, domain):
|
|
posts = self.vk.wall.get(domain=domain, count=3)
|
|
post = max([[int(pt["id"]), int(pt["owner_id"])] for pt in posts["items"]])
|
|
|
|
[last_post, group_id] = post
|
|
typ = 0 if group_id > 0 else 1
|
|
return group_id, typ, last_post
|
|
|
|
def info(self, group_ids):
|
|
groups, peoples = [], []
|
|
|
|
for id in group_ids:
|
|
if id > 0: peoples.append(id)
|
|
else: groups.append(id)
|
|
|
|
if groups:
|
|
data = self.vk.groups.getById(group_ids=", ".join(list(map(lambda x: str(x)[1:], groups))))
|
|
groups = [[elem['screen_name'], elem['name']] for elem in data]
|
|
|
|
if peoples:
|
|
data = self.vk.users.get(user_ids=", ".join(list(map(str, peoples))), fields="domain, online")
|
|
peoples = [[elem['domain'], elem['first_name'] + ' ' + elem['last_name'], elem['online']] for elem in data]
|
|
|
|
return peoples, groups
|
|
|
|
def last_post(self, owner_id=None, domain=None, post=None):
|
|
if post: wall = self.vk.wall.getById(posts=post, extended=1)
|
|
else: wall = self.vk.wall.get(owner_id=owner_id, domain=domain, count=2, extended=1)
|
|
|
|
post = max([[int(pt["id"]), pt] for pt in wall["items"]])
|
|
owner_id = post[1]['owner_id']
|
|
|
|
if owner_id > 0:
|
|
[name, domain] = [[per['first_name'] + ' ' + per['last_name'], per['screen_name']] for per in wall["profiles"] if per["id"] == owner_id][0]
|
|
else:
|
|
[name, domain] = [[gro['name'], gro['screen_name']] for gro in wall["groups"] if gro["id"] == int(str(owner_id)[1:])][0]
|
|
|
|
if 'copy_history' in post[1]:
|
|
comment, orig_id = post[1]['text'], post[0]
|
|
|
|
post = [post[1]['copy_history'][0]['id'], post[1]['copy_history'][0]]
|
|
owner_id_r = post[1]['owner_id']
|
|
|
|
if owner_id_r > 0:
|
|
[name_r, domain_r] = [[per['first_name'] + ' ' + per['last_name'], per['screen_name']] for per in wall["profiles"] if per["id"] == owner_id_r][0]
|
|
else:
|
|
[name_r, domain_r] = [[gro['name'], gro['screen_name']] for gro in wall["groups"] if gro["id"] == int(str(owner_id_r)[1:])][0]
|
|
|
|
return get_output(post, name_r, domain_r, [orig_id, owner_id, name, domain, comment])
|
|
else:
|
|
return get_output(post, name, domain)
|
|
|
|
|
|
def get_output(post, name, domain, repost=None):
|
|
[post_id, post] = post
|
|
owner_id, output = post['owner_id'], []
|
|
|
|
attachments = post['attachments']
|
|
types = [[typ['type'], typ] for typ in attachments]
|
|
|
|
for typ in types:
|
|
if typ[0] == 'photo':
|
|
output.append([typ[0], typ[1]['photo']['sizes'][-1]['url']])
|
|
elif typ[0] == 'video':
|
|
video = typ[1]['video']
|
|
if 'photo_1280' in video:
|
|
frame = video['photo_1280']
|
|
else:
|
|
frame = video['photo_800']
|
|
|
|
output.append([typ[0], [video['title'], video['duration'], video['views'], frame]])
|
|
elif typ[0] == "doc":
|
|
output.append([typ[0], [typ[1]['doc']['title'], typ[1]['doc']['url'],
|
|
typ[1]['doc']['type'], typ[1]['doc']['size']]])
|
|
elif typ[0] == "audio":
|
|
output.append([typ[0], [typ[1]['audio']['title'], typ[1]['audio']['url'], typ[1]['audio']['artist']]])
|
|
elif typ[0] == "poll":
|
|
answer = [[que['text'], que['votes']] for que in typ[1]['poll']['answers']]
|
|
output.append([typ[0], [typ[1]['poll']['question'], typ[1]['poll']['votes'],
|
|
typ[1]['poll']['disable_unvote'], typ[1]['poll']['anonymous'], answer]])
|
|
elif typ[0] == "link":
|
|
output.append([typ[0], [typ[1]['link']['title'], typ[1]['link']['url']]])
|
|
|
|
if repost:
|
|
[author_id, author_owner_id, author_name, author_domain, author_comment] = repost
|
|
|
|
comment = f"<b>Автор репоста -</b> <a href='https://vk.com/{author_domain}?w=wall{author_owner_id}_{author_id}'>{author_name}</a>"
|
|
if author_comment != "": comment += f"\n<blockquote>{author_comment}</blockquote>\n"
|
|
comment += f"\n<b>Автор поста -</b> <a href='https://vk.com/{domain}?w=wall{owner_id}_{post_id}'>{name}</a>"
|
|
|
|
return [output, f"{post['text']}", comment, author_id]
|
|
else:
|
|
return [output, f"{post['text']}",
|
|
f"<b>Автор поста -</b> <a href='https://vk.com/{domain}?w=wall{owner_id}_{post_id}'>{name}</a>", post_id] |