Add option for sending message to Discord when a new feed is added

This commit is contained in:
2021-12-13 17:30:18 +01:00
parent 6d6432a8dc
commit 6458efd28b

View File

@ -1,3 +1,5 @@
from typing import Optional
import typer import typer
from dhooks import Webhook from dhooks import Webhook
from reader import FeedExistsError, make_reader from reader import FeedExistsError, make_reader
@ -10,7 +12,10 @@ reader = make_reader(Settings.db_file) # For RSS (https://github.com/lemon24/re
@app.command() @app.command()
def add(feed_url: str) -> None: def add(
feed_url: str = typer.Argument(..., help="RSS or Atom feed URL."),
notify_discord: bool = typer.Option(False, help="Send message to Discord."),
) -> None:
"""Add a feed to the database """Add a feed to the database
Args: Args:
@ -32,6 +37,13 @@ def add(feed_url: str) -> None:
for entry in entries: for entry in entries:
reader.mark_entry_as_read(entry) reader.mark_entry_as_read(entry)
if notify_discord:
# Send a message to Discord
hook.send(
f"discord-rss-bot: {feed_url} added to the database.\nYou now have "
f"{reader.get_feed_counts()} feeds."
)
typer.echo(f"{feed_url} added") typer.echo(f"{feed_url} added")