diff options
author | Cédric Bonhomme <kimble.mandel@gmail.com> | 2013-07-24 14:24:54 +0200 |
---|---|---|
committer | Cédric Bonhomme <kimble.mandel@gmail.com> | 2013-07-24 14:24:54 +0200 |
commit | 7044f4676a083fc85c1092cf31eeda3dd279915d (patch) | |
tree | f36892898e2c06e1f4496f9c890d155252c1a48e /source/search.py | |
parent | Fixed an URL in the README. (diff) | |
download | newspipe-7044f4676a083fc85c1092cf31eeda3dd279915d.tar.gz newspipe-7044f4676a083fc85c1092cf31eeda3dd279915d.tar.bz2 newspipe-7044f4676a083fc85c1092cf31eeda3dd279915d.zip |
added add_to_index function, to add an article to the Whoosh index.
Diffstat (limited to 'source/search.py')
-rw-r--r-- | source/search.py | 18 |
1 files changed, 17 insertions, 1 deletions
diff --git a/source/search.py b/source/search.py index e9e4c801..63abddc4 100644 --- a/source/search.py +++ b/source/search.py @@ -63,6 +63,22 @@ def create_index(): feed_id=feed["feed_id"]) writer.commit() +def add_to_index(articles, feed): + """ + Add a list of articles to the index. + """ + try: + ix = open_dir(indexdir) + except (EmptyIndexError, OSError) as e: + raise EmptyIndexError + writer = ix.writer() + for article in articles: + writer.add_document(title=article["article_title"], \ + content=utils.clear_string(article["article_content"]), \ + article_id=article["article_id"] , \ + feed_id=feed["feed_id"]) + writer.commit() + def search(term): """ Search for `term` in the index. @@ -82,4 +98,4 @@ if __name__ == "__main__": #create_index() results = search("Nothomb") for article in results: - print(article)
\ No newline at end of file + print(article) |