aboutsummaryrefslogtreecommitdiff
path: root/pyaggr3g470r/models.py
blob: f1cac323c42a0710fcf683c0186148ecf445455c (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
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)
bgstack15