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)
|
