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
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
|
#! /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
import bson.objectid
class User(Document, UserMixin):
firstname = StringField(required=True)
lastname = StringField(required = True)
email = EmailField(required=True, unique=True)
pwdhash = StringField(required=True)
feeds = ListField(EmbeddedDocumentField('Feed'))
created_at = DateTimeField(required=True, default=datetime.now)
def get_id(self):
return self.email
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.email
class Feed(EmbeddedDocument):
oid = ObjectIdField(default=bson.objectid.ObjectId , primary_key=True)
title = StringField(required=True)
link = StringField(required=True, unique=True)
site_link = StringField(required=True)
mail = BooleanField(default=False)
articles = ListField(ReferenceField('Article', dbref = False))
created_date = DateTimeField(required=True, default=datetime.now)
meta = {
'ordering': ['+title']
}
def __eq__(self, other):
return self.title == other.title
def __str__(self):
return 'Feed: %s' % self.title
class Article(Document):
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)
meta = {
'ordering': ['-date'],
'indexes': [
{'fields': ['-date'],
'sparse': True, 'types': False },
]
}
def __str__(self):
return 'Article: %s' % self.title
if __name__ == "__main__":
# Point of entry in execution mode
db = connect('pyaggr3g470r1')
db.drop_database('pyaggr3g470r1')
from werkzeug import generate_password_hash
password = generate_password_hash("admin")
user1 = User(firstname="Cédric", lastname="Bonhomme", \
email="kimble.mandel@gmail.com", pwdhash=generate_password_hash("admin"))
user1.save()
import mongodb
mongo = mongodb.Articles("127.0.0.1", 27017, \
"pyaggr3g470r", "***", "")
feeds = mongo.get_all_feeds()
for feed in feeds:
articles = []
feed_articles = mongo.get_articles(feed["feed_id"])
feed_articles = sorted(feed_articles, key=lambda t: t['article_date'], reverse=True)
for article in feed_articles:
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"], \
retrieved_date=article["article_date"])
articles.append(article1)
article1.save()
sorted(articles, key=lambda t: t.date, reverse=True)
feed1 = Feed(title=feed["feed_title"], link=feed["feed_link"],
site_link=feed["site_link"], mail=feed["mail"],
articles=articles)
#feed1.save()
user1.feeds.append(feed1)
user1.save()
|
