main_workers.py

from uploader import Image_Upload
from config import BASE_PATH
import requests
import json
import os
import re


def Sort_Files(onlyFiles):
    """
    Сортировка файлов по числам в названии,
    если есть хоть один непронумированный файл,
    сортировка будет пропущена
    """
    try:
        onlyFiles.sort(key=lambda file: int(
            ''.join(re.findall('\d+', file.partition('.')[:1][0]))))
    except:
        pass


def Send_Title(title_name, Postmen_data):
    Files_Names = Get_Files_Names(title_name)
    Files_urls = []
    senders = []
    for file in Files_Names:
        senders.append(Image_Upload.delay(title=title_name, file=file))
    for sender in senders:
        sender.wait(timeout=None, interval=0.5)
        if sender.get() == 'None':
            print('Fail')
        Files_urls.append({'tag': 'img', 'attrs': {'src': sender.get()}})
    return Send_Post(title_name, Files_urls, postmen_data=Postmen_data)


def Get_Files_Names(title, subtitle=None):
    '''
    Возвращает отсортированный список названий файлов
    (subtitle - недопиленная система мульти-пакетного обработчика)
    '''
    if subtitle == None:
        osDir = f"{BASE_PATH}{title}/"
    else:
        osDir = f"{BASE_PATH}{title}/{subtitle}/"
    onlyFiles = [f for f in os.listdir(
        osDir) if os.path.isfile(os.path.join(osDir, f))]
    try:
        onlyFiles.remove('.DS_Store')
    except:
        pass
    Sort_Files(onlyFiles)
    return onlyFiles


def Send_Post(title_name, content, postmen_data):
    '''
    Создание поста
    на вход получает:
    postmen_data - первичные данные с токеном
    content и title_name - набор ссылок на файлы и название поста(тайтла)
    '''
    postmen_data['title'], postmen_data['content'] = title_name, content
    create_page = requests.post(
        'https://api.telegra.ph/createPage?', json=postmen_data)
    return json.loads(create_page.content)