Added insert methods

Author: wybiral

models/article.py

@@ -14,4 +14,19 @@ class Article(db.Model):
     date_published = db.Column(db.DateTime)
     __table_args__ = (
         db.UniqueConstraint('source_id', 'guid', name='uc_source_guid'),
-    )
+    )
+
+    @classmethod
+    def insert_from_feed(cls, source_id, feed_articles):
+        stmt = Article.__table__.insert().prefix_with('IGNORE')
+        articles = []
+        for article in feed_articles:
+            articles.append({
+                'title': article['title'],
+                'body': article['summary'],
+                'link': article['link'],
+                'guid': article['id'],
+                'source_id': source_id,
+                'date_published': article['published'],
+            })
+        db.engine.execute(stmt, articles)

models/source.py

@@ -8,3 +8,13 @@ class Source(db.Model):
     link = db.Column(db.Text, nullable=False)
     feed = db.Column(db.Text, nullable=False)
     date_added = db.Column(db.DateTime, default=datetime.datetime.utcnow)
+
+    @classmethod
+    def insert_from_feed(cls, feed, feed_source):
+        link = feed_source['link']
+        title = feed_source['title']
+        subtitle = feed_source['subtitle']
+        source = Source(feed=feed, link=link, title=title, subtitle=subtitle)
+        db.session.add(source)
+        db.session.commit()
+        return source