Miroslavsckaya/tg_rss_bot
1
1
Fork 0
Code Issues 7 Pull requests Projects 1 Releases Activity

refactor rss.py
Some checks failed
continuous-integration/drone/push Build is passing
Details
continuous-integration/drone/pr Build is failing
Details

Browse source
This commit is contained in:
mitsuha_s 2022-06-12 13:18:55 +00:00
parent 18f680a9e9
commit b2c49a583a
1 changed files with 14 additions and 24 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

38
rss.py
View file

@ -2,33 +2,23 @@ import feedparser
class FeedItem: class FeedItem:
def __init__(self, url: str, title: str, description: str, date: str) -> None: def __init__(self, item: feedparser.util.FeedParserDict) -> None:
self.url = url self.url = item.get('link', '')
self.title = title self.title = item.get('title', '')
self.description = description self.description = item.get('summary', '')
self.date = date self.date = item.get('published', '')
class Feed: class Feed:
def __init__(self, url: str, items: list[FeedItem]) -> None: def __init__(self, url: str, feed: feedparser.util.FeedParserDict) -> None:
self.url = url self.url = url
self.items = items self.items = self.__get_items(feed.entries)
def __get_items(self, items: list[feedparser.util.FeedParserDict]) -> list[FeedItem]:
list_items = []
for item in items:
list_items.append(FeedItem(item))
return list_items
class RssReader: class RssReader:
def get_feed(self, url: str) -> Feed: def get_feed(self, url: str) -> Feed:
f = feedparser.parse(url) return Feed(url, feedparser.parse(url))
items = self.__get_items(f.entries)
return Feed(url, items)
def __convert_to_feed_item(self, item: feedparser.util.FeedParserDict) -> FeedItem:
title = item.get('title', '')
url = item.get('link', '')
description = item.get('summary', '')
date = item.get('published', '')
return FeedItem(url, title, description, date)
def __get_items(self, items: list) -> list:
list_items = []
for item in items:
list_items.append(self.__convert_to_feed_item(item))
return list_items