aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--pyaggr3g470r/crawler.py208
-rw-r--r--pyaggr3g470r/feedgetter.py2
-rw-r--r--pyaggr3g470r/views.py7
-rw-r--r--requirements.txt1
4 files changed, 214 insertions, 4 deletions
diff --git a/pyaggr3g470r/crawler.py b/pyaggr3g470r/crawler.py
new file mode 100644
index 00000000..15849b9d
--- /dev/null
+++ b/pyaggr3g470r/crawler.py
@@ -0,0 +1,208 @@
+#! /usr/bin/env python
+# -*- coding: utf-8 -
+
+import feedparser
+import urllib2
+import requests
+from requests.exceptions import *
+#from requests.packages.urllib3.exceptions import DecodeError
+from urlparse import urlparse
+from datetime import datetime
+
+import gevent.monkey
+#gevent.monkey.patch_socket()
+gevent.monkey.patch_all()
+from gevent import Timeout
+
+from gevent.pool import Pool
+
+"""
+import logging
+logging.basicConfig()
+logging.getLogger().setLevel(logging.CRITICAL)
+requests_log = logging.getLogger("requests.packages.urllib3")
+requests_log.setLevel(logging.CRITICAL)
+requests_log.propagate = True
+"""
+
+
+import models
+import conf
+if not conf.ON_HEROKU:
+ import search
+import utils
+
+if not conf.ON_HEROKU:
+ from flask.ext.mail import Message
+ from pyaggr3g470r import mail
+
+from pyaggr3g470r import app, db
+from pyaggr3g470r.models import User, Feed, Article
+
+import log
+pyaggr3g470r_log = log.Log("feedgetter")
+
+
+
+
+
+HEADERS = {'User-Agent': conf.USER_AGENT}
+
+class TooLong(Exception):
+ def __init__(self):
+ """
+ Log a when greenlet took to long to fetch a resource.
+ """
+ pass #logger.warning("Greenlet took to long")
+
+
+class FeedGetter(object):
+ """
+ This class is in charge of retrieving feeds listed in ./var/feed.lst.
+ This class uses feedparser module from Mark Pilgrim.
+ For each feed a new thread is launched.
+ """
+ def __init__(self, email):
+ """
+ Initializes the database connection.
+ """
+ feedparser.USER_AGENT = conf.USER_AGENT
+ if conf.HTTP_PROXY == "":
+ self.proxy = urllib2.ProxyHandler({})
+ self.proxies = {}
+ else:
+ self.proxy = urllib2.ProxyHandler({"http" : conf.HTTP_PROXY, \
+ "https": conf.HTTP_PROXY})
+ self.proxies = {
+ "http": "http://" + conf.HTTP_PROXY,
+ "https": "http://" + conf.HTTP_PROXY
+ }
+ feedparser.USER_AGENT = conf.USER_AGENT
+ self.user = User.query.filter(User.email == email).first()
+
+ def retrieve_async(self, feeds):
+ """
+ Spawn different jobs in order to retrieve a list of distant resources.
+ Returns a list of models.Item objects.
+ """
+ def fetch(feed):
+ """
+ Fetch the content located at 'wsw_item.url'.
+ """
+ pyaggr3g470r_log.info("Fetching " + feed.title)
+ a_feed = feedparser.parse(feed.link, handlers = [self.proxy])
+ if a_feed['entries'] == []:
+ return
+
+ # Feed informations
+ if feed.title == "":
+ try:
+ feed.title = a_feed.feed.title
+ except:
+ feed.title = ""
+ if feed.description == "":
+ try:
+ feed.description = a_feed.feed.subtitle
+ except:
+ feed.description = ""
+
+ articles = []
+ for article in a_feed['entries']:
+
+ nice_url = article.link.encode("utf-8")
+ if conf.RESOLVE_ARTICLE_URL:
+ try:
+ # resolves URL behind proxies (like feedproxy.google.com)
+ r = requests.get(article.link, timeout=5.0, proxies=self.proxies)
+ nice_url = r.url.encode("utf-8")
+ except Timeout:
+ pyaggr3g470r_log.warning("Timeout when getting the real URL of %s." % (article.link,))
+ continue
+ except Exception as e:
+ pyaggr3g470r_log.warning("Unable to get the real URL of %s. Error: %s" % (article.link, str(e)))
+ continue
+ # remove utm_* parameters
+ nice_url = utils.clean_url(nice_url)
+
+ description = ""
+ article_title = ""
+ try:
+ # article content
+ description = article.content[0].value
+ except AttributeError:
+ try:
+ # article description
+ description = article.description
+ except Exception:
+ description = ""
+ try:
+ description = BeautifulSoup(description, "html.parser").decode()
+ article_title = BeautifulSoup(article.title, "html.parser").decode()
+ except Exception as E:
+ #pyaggr3g470r_log.error("Problem when sanitizing the content of the article %s (%s)" % (article_title, nice_url))
+ article_title = article.title
+
+ try:
+ post_date = datetime(*article.published_parsed[:6])
+ except:
+ post_date = datetime(*article.updated_parsed[:6])
+
+ # save the article
+ article = Article(link=nice_url, title=article_title, \
+ content=description, readed=False, like=False, date=post_date, \
+ user_id=self.user.id, feed_id=feed.id)
+ articles.append(article)
+
+
+
+ return feed, articles
+
+ jobs = []
+ pool = Pool(20)
+ jobs = [pool.spawn(fetch, feed) for feed in feeds]
+ pool.join()
+
+ return jobs
+
+ def insert_database(self, elements):
+ """
+ Insert articles in the database.
+ """
+ pyaggr3g470r_log.info("Database insertion...")
+ for feed, articles in elements:
+
+ for article in articles:
+
+ exist = Article.query.filter(Article.user_id == self.user.id, Article.feed_id == feed.id, Article.link == article.link).first()
+ if exist != None:
+ pyaggr3g470r_log.error("Article %s (%s) already in the database." % (article.title, article.link))
+ continue
+
+ try:
+ feed.articles.append(article)
+ #db.session.merge(article)
+ db.session.commit()
+ pyaggr3g470r_log.info("New article %s (%s) added." % (article.title, article.link))
+ except IntegrityError:
+ pyaggr3g470r_log.error("Article %s (%s) already in the database." % (article.title, article.link))
+ db.session.rollback()
+ continue
+ except Exception as e:
+ pyaggr3g470r_log.error("Error when inserting article in database: " + str(e))
+ continue
+ db.session.close()
+ return True
+
+
+ def retrieve_feed(self, feed_id=None):
+ """
+ Launch
+ """
+ user = User.query.filter(User.email == self.user.email).first()
+ feeds = [feed for feed in user.feeds if feed.enabled]
+ if feed_id != None:
+ feeds = [feed for feed in feeds if feed.id == feed_id]
+
+ responses = self.retrieve_async(feeds)
+
+ self.insert_database([item.value for item in responses if item.value is not None]) \ No newline at end of file
diff --git a/pyaggr3g470r/feedgetter.py b/pyaggr3g470r/feedgetter.py
index ccdc60f2..b06c8eff 100644
--- a/pyaggr3g470r/feedgetter.py
+++ b/pyaggr3g470r/feedgetter.py
@@ -85,7 +85,7 @@ class FeedGetter(object):
"""
feeds = [feed for feed in self.user.feeds if feed.enabled]
if feed_id != None:
- feeds = [feed for feed in feeds if str(feed.id) == feed_id]
+ feeds = [feed for feed in feeds if feed.id == feed_id]
for current_feed in feeds:
try:
# launch a new thread for the RSS feed
diff --git a/pyaggr3g470r/views.py b/pyaggr3g470r/views.py
index a8d7e19c..ddc0ddf0 100644
--- a/pyaggr3g470r/views.py
+++ b/pyaggr3g470r/views.py
@@ -37,7 +37,7 @@ from werkzeug import generate_password_hash
import conf
import utils
import export
-import feedgetter
+import feedgetter, crawler
import models
if not conf.ON_HEROKU:
import search as fastsearch
@@ -181,13 +181,14 @@ def home():
return render_template('home.html', result=result, head_title=unread_articles)
@app.route('/fetch/', methods=['GET'])
-@app.route('/fetch/<feed_id>', methods=['GET'])
+@app.route('/fetch/<int:feed_id>', methods=['GET'])
@login_required
def fetch(feed_id=None):
"""
Triggers the download of news.
"""
- feed_getter = feedgetter.FeedGetter(g.user.email)
+ #feed_getter = feedgetter.FeedGetter(g.user.email)
+ feed_getter = crawler.FeedGetter(g.user.email)
feed_getter.retrieve_feed(feed_id)
flash("New articles retrieved.", 'success')
return redirect(redirect_url())
diff --git a/requirements.txt b/requirements.txt
index 9919f2f0..5b516396 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -11,3 +11,4 @@ Flask-Principal
Flask-WTF
Flask-Mail
WTForms
+gevent \ No newline at end of file
bgstack15