-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathmain_script.py
More file actions
50 lines (45 loc) · 1.72 KB
/
main_script.py
File metadata and controls
50 lines (45 loc) · 1.72 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
import argparse
import os
import random
import time
from dotenv import load_dotenv
from telegram.error import NetworkError
from image_bot import send_photo_to_chat
if __name__ == "__main__":
load_dotenv()
bot_token = os.environ["TELEGRAM_TOKEN"]
chat_id = os.environ["TELEGRAM_CHAT_ID"]
parser = argparse.ArgumentParser(
description="""Бот последовательно загружает фотографии в Telegram Chat
из выбранной директории, с задержкой публикации на указанное время.
Бот должен быть админом чата.
Требует TELEGRAM_TOKEN и TELEGRAM_CHAT_ID, которые должны быть указаны в переменной .env"""
)
parser.add_argument(
"-i",
"--images_dir_path",
help="путь к директории с изображениями",
default="images",
)
parser.add_argument(
"-s",
"--sleep_time",
help="время задержки публикации фото, в часах",
default=4,
type=float,
)
args = parser.parse_args()
sleep_time = args.sleep_time * 3600
images_paths = []
for address, dirs, files in os.walk(args.images_dir_path):
for name in files:
images_paths.append(os.path.join(address, name))
while True:
try:
for image_path in images_paths:
send_photo_to_chat(image_path, chat_id, bot_token)
time.sleep(sleep_time)
random.shuffle(images_paths)
except NetworkError:
print("Connection lost. Trying to reconnecting")
time.sleep(2)