add docstrings

This commit is contained in:
mitsuha_s 2022-07-08 20:57:41 +00:00
parent d25ca99da1
commit 2caa1e0618
3 changed files with 9 additions and 4 deletions

View File

@ -60,6 +60,7 @@ class Database:
self.conn.commit()
def unsubscribe_user_by_url(self, user_id: int, url: str) -> None:
"""Subscribe a user to the feed by url."""
feed_id = self.find_feed_by_url(url)
if feed_id is None:
raise DisplayableException('Feed does not exist')
@ -128,7 +129,7 @@ class Database:
def update_feed_items(self, feed_id: int, new_items: list[FeedItem]) -> None:
"""Replace last feed items with a list items that receive."""
for i in range(len(new_items)):
for i, _ in enumerate(new_items):
new_items[i] = [feed_id] + list(new_items[i].__dict__.values())[:-1]
self.cur.execute('DELETE FROM feeds_last_items WHERE feed_id = ?', [feed_id])

View File

@ -59,7 +59,7 @@ class CommandProcessor:
feed_list = ''
for index, feed in enumerate(feeds, start=1):
feed_list += '* ' + str(index) + ': ' + f'''<a href="{feed['url']}">{feed['title']}</a''' + '\n'
feed_list += '* ' + str(index) + ': ' + f'''<a href="{feed['url']}">{feed['title']}</a>''' + '\n'
self.bot.reply_to(message, 'Your feeds:\n' + feed_list)
@ -105,7 +105,7 @@ class Notifier:
text=f'Updates from the {feed_title} feed:'
)
self.sent_counter += 1
for update in updates:
self.__send_update(chat_id, update)
self.sent_counter += 1

View File

@ -4,12 +4,15 @@ from telegram import Notifier
class UpdateManager:
"""Implement the feed update."""
def __init__(self, database: Database, notifier: Notifier, rss_reader: RssReader) -> None:
self.database: Database = database
self.notifier: Notifier = notifier
self.rss_reader: RssReader = rss_reader
def update(self):
"""Send new feed items to the user."""
feeds = self.database.find_feeds()
for feed_id, feed_url in feeds:
@ -27,9 +30,10 @@ class UpdateManager:
self.database.update_feed_items(feed_id, new_items)
def __calculate_difference(self, new_items: list[FeedItem], old_items_urls: list[str]) -> list[FeedItem]:
"""Calculate new feed items."""
if not old_items_urls:
return new_items
diff = []
for item in new_items: