50 lines
1.3 KiB
Python
50 lines
1.3 KiB
Python
from json import JSONEncoder
|
|
import io
|
|
from random import randint
|
|
|
|
from aiohttp import ClientSession
|
|
|
|
|
|
class EnumEncoder(JSONEncoder):
|
|
def default(self, obj):
|
|
return obj.value
|
|
|
|
|
|
def get_self_id(api):
|
|
return api.groups.getById()[0]['id']
|
|
|
|
|
|
def generate_random_id():
|
|
return randint(-9 ** 99, 9 ** 99)
|
|
|
|
|
|
async def get_user_name(user_id, api, name_case='nom'):
|
|
user = (await api.users.get(user_ids=user_id, name_case=name_case))[0]
|
|
return f'{user["first_name"]} {user["last_name"]}'
|
|
|
|
|
|
def parse_attachments(attachments) -> tuple:
|
|
out = []
|
|
for attach in attachments:
|
|
t = attach['type']
|
|
if t == 'wall':
|
|
out.append(f'{t}{attach[t]["from_id"]}_{attach[t]["id"]}')
|
|
else:
|
|
out.append(f'{t}{attach[t]["owner_id"]}_{attach[t]["id"]}')
|
|
return tuple(out)
|
|
|
|
|
|
async def reupload_attachments(attachments, upload):
|
|
new_attachments = []
|
|
for a in attachments:
|
|
t = a['type']
|
|
if t != 'photo':
|
|
continue
|
|
url = a[t]['sizes'][-1]['url']
|
|
|
|
async with ClientSession() as session, session.get(url) as response:
|
|
file = io.BytesIO(await response.content.read())
|
|
attachment = upload.photo_messages(file)[0]
|
|
new_attachments.append(f'photo{attachment["owner_id"]}_{attachment["id"]}')
|
|
return new_attachments
|