diff options
author | Cédric Bonhomme <cedric@cedricbonhomme.org> | 2014-04-08 23:06:27 +0200 |
---|---|---|
committer | Cédric Bonhomme <cedric@cedricbonhomme.org> | 2014-04-08 23:06:27 +0200 |
commit | ef4dd7682b7f9e8716391e6ea8859b3db8b77561 (patch) | |
tree | 4972d50a63da01e962f4e1f526eaa71a3cf1d485 /pyaggr3g470r/utils.py | |
parent | Update requirements.txt file. (diff) | |
download | newspipe-ef4dd7682b7f9e8716391e6ea8859b3db8b77561.tar.gz newspipe-ef4dd7682b7f9e8716391e6ea8859b3db8b77561.tar.bz2 newspipe-ef4dd7682b7f9e8716391e6ea8859b3db8b77561.zip |
Import/export of feeds with OPML is working.
Diffstat (limited to 'pyaggr3g470r/utils.py')
-rwxr-xr-x | pyaggr3g470r/utils.py | 11 |
1 files changed, 6 insertions, 5 deletions
diff --git a/pyaggr3g470r/utils.py b/pyaggr3g470r/utils.py index bb02c4df..16777278 100755 --- a/pyaggr3g470r/utils.py +++ b/pyaggr3g470r/utils.py @@ -47,7 +47,8 @@ from collections import Counter from contextlib import contextmanager import conf -import models +from pyaggr3g470r import db +from pyaggr3g470r.models import User, Feed # regular expression to check URL url_finders = [ \ @@ -76,7 +77,7 @@ def import_opml(email, opml_file): """ Import new feeds from an OPML file. """ - user = models.User.objects(email=email).first() + user = User.query.filter(User.email == email).first() try: subscriptions = opml.parse(opml_file) except Exception as e: @@ -108,12 +109,12 @@ def import_opml(email, opml_file): except: site_link = "" - new_feed = models.Feed(title=title, description=description, link=link, site_link=site_link, email=False, enabled=True) + new_feed = Feed(title=title, description=description, link=link, site_link=site_link, email_notification=False, enabled=True) + user.feeds.append(new_feed) - nb += 1 - user.save() + db.session.commit() return nb def open_url(url): |