Total Complexity | 6 |
Total Lines | 50 |
Duplicated Lines | 0 % |
Changes | 0 |
1 | # -*- encoding: utf-8 -*- |
||
2 | """ |
||
3 | cron: */15 6-22 * * * |
||
4 | new Env('RSS 订阅推送'); |
||
5 | """ |
||
6 | |||
7 | from datetime import datetime, timedelta |
||
8 | |||
9 | import feedparser |
||
10 | |||
11 | from notify_mtr import send |
||
12 | from utils_models import History, Rss, db |
||
13 | |||
14 | |||
15 | class RssRobot: |
||
16 | def main(self): |
||
17 | self.remove_old_history() |
||
18 | rss_list = Rss.select() |
||
19 | msg = "" |
||
20 | post_url_list = [ |
||
21 | rss_history.url |
||
22 | for rss_history in History.select().where( |
||
23 | History.publish_at == datetime.today().strftime("%Y-%m-%d") |
||
24 | ) |
||
25 | ] |
||
26 | for rss in rss_list: |
||
27 | rss_history_list = [] |
||
28 | feed = feedparser.parse(rss.feed) |
||
29 | for entry in feed.entries: |
||
30 | if entry.link not in post_url_list: |
||
31 | msg = msg + f"{entry.title}\n{entry.link}\n\n" |
||
32 | rss_history_list.append(History(url=entry.link)) |
||
33 | |||
34 | with db.atomic(): |
||
35 | History.bulk_create(rss_history_list, batch_size=10) |
||
36 | |||
37 | return msg |
||
38 | |||
39 | def remove_old_history(self): |
||
40 | # 只保留最近一周的记录 |
||
41 | week_date_range = datetime.now() + timedelta(days=-7) |
||
42 | History.delete().where( |
||
43 | History.publish_at < week_date_range.strftime("%Y-%m-%d") |
||
44 | ).execute() |
||
45 | |||
46 | |||
47 | if __name__ == "__main__": |
||
48 | res = RssRobot().main() |
||
49 | send("RSS 订阅", res) |
||
50 |