Автоматизация на питоне, как настроить?

Question

[JRBRO]

Всем привет, у меня комплексный вопрос. Имеется следующий код

driver = webdriver.Chrome(executable_path=r"C:\Users\...\chromedriver")
driver.get("www.google.com/1")
print(driver.title)
driver.maximize_window()
driver.find_element_by_css_selector("Search").click()
P = subprocess.Popen("ffmpeg -f gdigrab -framerate ntsc -video_size 1920x1080 -i desktop  -f dshow -i audio=""@device_cm_{33D9A762-90C8-11D0-BD43-00A0C911CE86}\wave_{9C18FEC6-11C8-4EA9-B1F1-0BD9A491C5F5}"" -vcodec libx264 -pix_fmt yuv420p -preset ultrafast -t 00:00:05 Google1.mp4", text=True, shell=True)
P.wait()
P = subprocess.Popen("ffmpeg -i Google1.mp4 -vn -acodec copy Google1.aac", text=True, shell=True)
P.wait()
P = subprocess.Popen("ffmpeg -i Google1.mp4 -vf fps=1 Google1%d.png", text=True, shell=True)
P.wait()

driver.quit()

filename = os.listdir(os.getcwd()) 
frames=['.jpeg','.jpg','.svg','.png','.PNG']

framesLocation= "Frames"


for file in filename:
    if os.path.splitext(file)[1] in frames:
        if(path.exists(framesLocation)):
            shutil.move(file,framesLocation)
        else:
            os.mkdir(framesLocation)
            shutil.move(file,framesLocation)

Цель кода такая-
1. Зайти на google1
2. Снять видео при помощи ffmpeg
3. Отделить аудио
4. Разбить на кадры
5. Кадры закинуть в отдельную папку

Суть вопроса, как не перекопируя заставить код:

Научить скрипт выбирать google1, google2, google3 откуда либо, именно сам линк (их будет очень много)
Переименовывать каждый следующий файл в соответствии с линком, для google1 такой, для google2 такой и т.д.
Раскидывать фреймы в соответствующую папку, не просто Frames, а Google1-> Frames, Google2-> Frames

Для единого кейса это работает, как сделать большое кол-во кейсов без переписывания кода 100 раз (очень много кейсов будет) и чтобы имена генерировались в соответствием с кейсом.

Answer 1

[Алан Гибизов]

В общем случае, код, который надо исполнить много раз с подстановкой разных значений, надо вынести в функцию, принимающую на вход нужное значение;
потом создать или получить последовательность (например, кортеж или список) с набором значений для подстановки, либо сделать генератор значений;
потом начать в цикле перебирать список значений и на каждое вызывать функцию, передавая в неё очередное значение.
В общем, всё довольно просто.

Comments

[JRBRO]

В теории, я это все плюс минус понял, здесь проблема перенести все эти идеи в код :(

[Алан Гибизов]

JRBRO, вы ожидаете, что вам это перенесут в код добрые самаритяне? Это вам на freelance.habr.com

[JRBRO]

Алан Гибизов, не обязательно переносить, достаточно было бы примера. Но я невкоем случае не требую такой помощи, рад и разжеваной теории и работаю "с тем, что есть", помощь она в любом виде сильно ценится

[Алан Гибизов]

JRBRO, ну вот:

Пример

from pprint import pprint


def make_me_beautiful(url):  # создаем функцию, внутрь которой кладем обработку того, что попадает на вход функции (url)
    result = None
    if 'http://' in url:
        beauty = f'URL {url} is beautiful!'
        result = url
    else:
        beauty = f'URL {url} is not beautiful.'
    print(beauty)
    return result


if __name__ == '__main__':
    url_list = ['http://google.ru', 'http://yandex.ru', 'google.com', 'http://rambler.ru']  # список данных
    # для обработки в цикле
    result_list = []
    for item in url_list:  # перебираем список
        my_result = make_me_beautiful(url=item)  # вызываем функцию
        if my_result is not None:
            result_list.append(my_result)  # помещаем результат выполнения функции в итоговый список

    pprint(result_list)

[Алан Гибизов]

JRBRO, а так-то надо бы книжку "Изучаем Python" Марк Лутц проработать - очень способствует...

[JRBRO]

Алан Гибизов, спасибо большое за пример и конечно же за советы. Обязательно буду продолжать обучение и стремиться к верхам. Спасибо.

[JRBRO]

Осталось решить как заставить

P = subprocess.Popen("ffmpeg -f gdigrab -framerate ntsc -video_size 1920x1080 -i desktop -f dshow -i audio=""@device_cm_{33D9A762-90C8-11D0-BD43-00A0C911CE86}\wave_{9C18FEC6-11C8-4EA9-B1F1-0BD9A491C5F5}"" -vcodec libx264 -pix_fmt yuv420p -preset ultrafast -t 00:00:5 google1.mp4", text=True, shell=True)

Google1.mp4 менятся на Google2.mp4 и тогда полная победа. С URL все сработало, научил их вытаскивать из txt файла

[Алан Гибизов]

JRBRO, так в чём проблема? Я этот нюанс в примере отобразил, там подстановка в строку значений в f-формате.

Answer 2

[MrYaz9]

Используя asycio. Закинь сайты в main, а работу с ними в другую функцию(с входными данными названия сайта) и потом просто вместо google вставляй эти данные.

Comments

[JRBRO]

Посмотрю данное решение