1 files changed, 91 insertions, 0 deletions
diff --git a/pyaggr3g470r/models.py b/pyaggr3g470r/models.py
new file mode 100644
index 00000000..f1cac323
--- /dev/null
+++ b/pyaggr3g470r/models.py
@@ -0,0 +1,91 @@
+#! /usr/bin/env python
+# -*- coding: utf-8 -*-
+
+from mongoengine import *
+from datetime import datetime
+
+from werkzeug import generate_password_hash, check_password_hash
+from flask.ext.login import UserMixin
+
+class User(Document, UserMixin):
+    firstname  = StringField(required=True)
+    lastname = StringField(required = True)
+    email = EmailField(required=True, unique=True)
+    pwdhash = StringField(required=True)
+    created_at = DateTimeField(required=True, default=datetime.now)
+
+    def set_password(self, password):
+        self.pwdhash = generate_password_hash(password)
+
+    def check_password(self, password):
+        return check_password_hash(self.pwdhash, password)
+
+    #required for administrative interface
+    def __unicode__(self):
+        return self.nickname
+
+class Feed(DynamicDocument):
+    title = StringField(required=True)
+    link = StringField(required=True)
+    image = StringField(required=True)
+    site_link = StringField(required=True)
+    mail = BooleanField()
+    #articles = ListField(EmbeddedDocumentField('Article'))
+    articles = ListField(ReferenceField('Article', dbref = False))
+    created_date = DateTimeField(required=True, default=datetime.now)
+
+    def __unicode__(self):
+        return 'Feed: %s' % self.title
+
+#class Article(EmbeddedDocument):
+class Article(DynamicDocument):
+    date = DateTimeField(required=True)
+    link = StringField(required=True)
+    title = StringField(required=True)
+    content = StringField(required=True)
+    readed = BooleanField()
+    like = BooleanField()
+    retrieved_date = DateTimeField(required=True, default=datetime.now)
+
+    def __unicode__(self):
+        return 'Article: %s' % self.title
+
+if __name__ == "__main__":
+    # Point of entry in execution mode
+    connect('pyaggr3g470r1')
+
+    Feed.drop_collection()
+    try:
+        Article.drop_collection()
+        # pas de méthode save() pour un objet EmbeddedDocument.
+    except:
+        pass
+
+    import conf
+    import mongodb
+    mongo = mongodb.Articles(conf.MONGODB_ADDRESS, conf.MONGODB_PORT, \
+                        conf.MONGODB_DBNAME, conf.MONGODB_USER, conf.MONGODB_PASSWORD)
+    feeds = mongo.get_all_feeds()
+    for feed in feeds:
+        articles = []
+
+        for article in mongo.get_articles(feed["feed_id"]):
+            article1 = Article(date=article["article_date"], link=article["article_link"], \
+                       title=article["article_title"], content=article["article_content"], \
+                        readed=article["article_readed"], like=article["article_like"])
+
+            articles.append(article1)
+            try:
+                article1.save()
+            except:
+                # pas de méthode save() pour un objet EmbeddedDocument.
+                pass
+
+        feed1 = Feed(title=feed["feed_title"], link=feed["feed_link"],
+                 image=feed["feed_image"], site_link=feed["site_link"], mail=feed["mail"],
+                 articles=articles)
+        feed1.save()
+
+
+    for feed in Feed.objects:
+        print(feed.articles[0].title)
+\ No newline at end of file