-
Notifications
You must be signed in to change notification settings - Fork 0
/
main.py
88 lines (66 loc) · 2.35 KB
/
main.py
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
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
from recentBooks import get_new_books
import requests
import telegramConsts as tc
import os
from dotenv import load_dotenv
import time
debug = False
def main():
bot_token = get_env_value('BOT_TOKEN')
channel_id = get_env_value('CHANNEL_ID')
while True:
try:
# get the new books
new_books = get_new_books(ignore_warnings=True)
# reverse it, so that we will send the newest books last
new_books.reverse()
for book in new_books:
title = book[0]
author = book[1]
url = book[2]
# this url gives small image.
# some changes on the url can bring better image
url = get_better_image(url)
description = book[3]
message = create_message(title, author, description)
if debug:
print('sending photo... (message: ' + message + ')')
send_photo(url, message, bot_token, channel_id)
# sleep for 10 minutes
time.sleep(60 * 10)
except Exception as e:
print(e)
def send_message(message, bot_token, channel_id):
requests.get(tc.SEND_MSG.format(bot_token, channel_id, message))
def send_photo(photo_url, caption, bot_token, channel_id):
requests.get(tc.SEND_PHOTO.format(bot_token, channel_id, photo_url, caption))
def get_env_value(name):
load_dotenv()
return os.getenv(name)
def get_better_image(url):
# there are 2 links to images, the one we have is smaller.
# so we take the url and ,ake it the url of the bigger image
try:
new_url = ''
url = url.strip('/')
for i in [0, 1, 2, 6]:
new_url += url.split('/')[i] + '/'
new_url = new_url[:new_url.find('150')] + '100' + new_url[new_url.find('150')+3:]
new_url = new_url[:new_url.rfind('150')] + '100' + new_url[new_url.rfind('150')+3:]
new_url = new_url.strip('/')
test = requests.get('http://' + new_url)
if test.status_code == 200:
return new_url
return url
except Exception:
return url
def create_message(title, author, description):
msg = ''
msg += '<b>' + title + '</b>'
msg += ' / '
msg += author
msg += '\n'
msg += description
return msg
if __name__ == '__main__':
main()