-
Notifications
You must be signed in to change notification settings - Fork 3
/
apipoll.py
55 lines (46 loc) · 1.55 KB
/
apipoll.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
import json
import requests
import time
import os
class Checker:
def __init__(self, urls, wait_time):
self.wait_time = wait_time
self.urls = urls
self.videos = self.get_videos()
self.main_loop()
@staticmethod
def get_data(url):
set_url = os.environ['api_url']
# urlfile = open("api_url.txt",'r')
# set_url = urlfile.read()
# urlfile.close()
url_input = requests.get(set_url).json()['data']
data = json.dumps(url_input)
data_dict = json.loads(data)
videos = [videos['id'] for videos in data_dict]
return videos
def get_videos(self):
videos = set()
for url in self.urls:
videos.update(Checker.get_data(url))
return videos
def check_new_videos(self):
new_videos = self.get_videos()
videos_diff = list(new_videos.difference(self.videos))
current_time = time.strftime("%H:%M:%S", time.localtime())
if len(videos_diff) > 0:
print(current_time, videos_diff)
new_id = str(list(new_videos.difference(self.videos)))
print(new_id)
with open("new_id.txt",'w') as idfile:
pass
idfile = open("new_id.txt", "w+")
idfile.write(new_id)
idfile.close()
else:
print(current_time, "No new videos")
self.videos = new_videos
def main_loop(self):
while True:
time.sleep(self.wait_time)
self.check_new_videos()