用 Python 構建你自己的 RSS 提示系統 | Linux 中國

這幾行程式碼建立一個名為 magazine_rss.sqlite 檔案的新 sqlite 資料庫，然後在資料庫建立一個名為 magazine 的新表。這個表有兩個列 —— title 和 date —— 它們能存諸 TEXT 型別的資料，也就是說每個列的值都是文字字元。

檢查資料庫中的舊文章

由於我們僅希望增加新的文章到我們的資料庫中，因此我們需要一個功能去檢查 RSS 源中的文章在資料庫中是否存在。我們將根據它來判斷是否傳送（有新文章的）郵件提示。Ok，現在我們來寫這個功能的程式碼。

def article_is_not_db(article_title, article_date):
    """ Check if a given pair of article title and date
    is in the database.
    Args:
        article_title (str): The title of an article
        article_date  (str): The publication date of an article
    Return:
        True if the article is not in the database
        False if the article is already present in the database
    """
    db.execute("SELECT * from magazine WHERE title=? AND date=?", (article_title, article_date))
    if not db.fetchall():
        return True
    else:
        return False

這個功能的主要部分是一個 SQL 查詢，我們執行它去搜索資料庫。我們使用一個 SELECT 命令去定義我們將要在哪個列上執行這個查詢。我們使用 * 符號去選取所有列（title 和 date）。然後，我們使用查詢的 WHERE 條件 article_title 和 article_date 去匹配標題和日期列中的值，以檢索出我們需要的內容。

最後，我們使用一個簡單的傳回 True 或者 False 的邏輯來表示是否在資料庫中找到匹配的文章。

在資料庫中新增新文章

現在我們可以寫一些程式碼去新增新文章到資料庫中。

def add_article_to_db(article_title, article_date):
    """ Add a new article title and date to the database
    Args:
        article_title (str): The title of an article
        article_date (str): The publication date of an article
    """
    db.execute("INSERT INTO magazine VALUES (?,?)", (article_title, article_date))
    db_connection.commit()

這個功能很簡單，我們使用了一個 SQL 查詢去插入一個新行到 magazine 表的 article_title 和 article_date 列中。然後提交它到資料庫中永久儲存。

這些就是在資料庫中所需要的東西，接下來我們看一下，如何使用 Python 實現提示系統和傳送電子郵件。

傳送電子郵件提示

我們使用 Python 標準庫模組 smtplib 來建立一個傳送電子郵件的功能。我們也可以使用標準庫中的 email 模組去格式化我們的電子郵件資訊。

def send_notification(article_title, article_url):
    """ Add a new article title and date to the database
    Args:
        article_title (str): The title of an article
        article_url (str): The url to access the article
    """
    smtp_server = smtplib.SMTP('smtp.gmail.com', 587)
    smtp_server.ehlo()
    smtp_server.starttls()
    smtp_server.login('your_email@gmail.com', '123your_password')
    msg = MIMEText(f'\nHi there is a new Fedora Magazine article : {article_title}. \nYou can read it here {article_url}')
    msg['Subject'] = 'New Fedora Magazine Article Available'
    msg['From'] = 'your_email@gmail.com'
    msg['To'] = 'destination_email@gmail.com'
    smtp_server.send_message(msg)
    smtp_server.quit()

def read_article_feed():
    """ Get articles from RSS feed """
    feed = feedparser.parse('https://fedoramagazine.org/feed/')
    for article in feed['entries']:
        if article_is_not_db(article['title'], article['published']):
            send_notification(article['title'], article['link'])
            add_article_to_db(article['title'], article['published'])
if __name__ == '__main__':
    read_article_feed()
    db_connection.close()

$ chmod a+x my_rss_notifier.py
$ sudo cp my_rss_notifier.py /etc/cron.hourly