main.py

from vk import vk_wrapper
import json
import os
from uuid import uuid1 as generate_id
import requests
from PIL import Image
import zipfile
from multiprocessing import Pool
import shutil

vkHandler = None

def init():
    global vkHandler

    if not os.path.isdir('userData'):
        os.mkdir("userData")

    config = {
        'access_token': None,
        'group_id': None,
        'api_version': '5.103'
    }

    if os.path.isfile('config.json'):
        with open('config.json') as config_file:
            config = json.load(config_file)
    else:
        with open('config.json', 'w') as config_file:
            json.dump(config, config_file)
            print('Please configurate \'config.json\' and run script again.')
            exit()
    vkHandler = vk_wrapper(**config)

def sequence_generator():
    number = 0
    while True:
        yield number
        number += 1

def get_img_request_handler(command):

    landscape_dict = set(['landscape', 'land', 'l', 'wide', 'w'])
    portrait_dict = set(['portrait', 'port', 'p', 'high'])

    def handle_command(obj, command):
        pic = Image.open(obj)
        width, height = pic.size
        if width > height and command in portrait_dict:
            pic = pic.transpose(Image.ROTATE_270)
        elif width < height and command in landscape_dict:
            pic = pic.transpose(Image.ROTATE_90)
        pic.save(obj)

    def handler(dir):
        imgs = [file for file in os.listdir(dir) if os.path.isfile(dir+"/"+file)]
        arch_path = dir+"/arch.zip"
        print("handling photo req | dir: ", dir)
        with zipfile.ZipFile(arch_path, "w") as archive:
            for img in imgs:
                handle_command(dir+"/"+img, command)
                archive.write(dir+"/"+img, "imgs/"+img)
        return arch_path

    return handler

def img_names_generator():
    for name in sequence_generator():
        yield str(name)+".jpg"

def handle_request(event, links, request_handler, *args):
    request_id = str(generate_id())
    request_dir = "userData/" + request_id
    os.mkdir(request_dir)

    vkHandler.send_message('downloading attachments...')
    download_files(links, request_dir, *args)
    vkHandler.send_message('processing files...')
    arch_path = request_handler(request_dir)
    vkHandler.send_message('uploading archive...')
    vkHandler.send_message(attachment = arch_path)
    shutil.rmtree(request_dir)

def download_file(kwargs):
    resp = requests.get(kwargs['url'])
    with open(kwargs['path'] + "/" + str(kwargs['name']), "wb") as writer:
        writer.write(resp.content)


def download_files(links, path = ".", name_generator = sequence_generator()):
    pool = Pool(max(min(len(links), os.cpu_count()), 1))
    links = [{'url': link, 'path': path, 'name': next(name_generator)} for link in links]
    print(links)
    pool.map(download_file, links)
    pool.close()
    pool.join()
        

def bot_loop():
    print("Bot loop started.")
    for event in vkHandler.listen():
        if event.type == vkHandler.longpoll.VkBotEventType.MESSAGE_NEW:
            photos, command = vkHandler.get_photos_links(), vkHandler.get_command()
            try:
                handle_request(event, photos, get_img_request_handler(command), img_names_generator())
            except BaseException as e:
                vkHandler.send_message("An error occured while processing your request.\n"+str(e))


def main():
    init()
    print("Bot successfuly inited. Starting bot loop.")
    while True:
        try:
            bot_loop()
        except BaseException as e:
            print(e)
            print("An error occured while bot worked. Restarting...")

if __name__ == '__main__':
    main()