forked from Arachnid/bloggart
/
models.py
185 lines (149 loc) · 4.84 KB
/
models.py
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
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
import aetycoon
import datetime
import hashlib
import re
from google.appengine.api import memcache
from google.appengine.ext import db
import config
import markup
import utils
if config.default_markup in markup.MARKUP_MAP:
DEFAULT_MARKUP = config.default_markup
else:
DEFAULT_MARKUP = 'html'
class BlogDate(db.Model):
"""Contains a list of year-months for published blog posts."""
@classmethod
def get_key_name(cls, post):
return '%d/%02d' % (post.published.year, post.published.month)
@classmethod
def create_for_post(cls, post):
inst = BlogDate(key_name=BlogDate.get_key_name(post))
inst.put()
return inst
@classmethod
def datetime_from_key_name(cls, key_name):
year, month = key_name.split("/")
return datetime.datetime(int(year), int(month), 1)
@property
def date(self):
return BlogDate.datetime_from_key_name(self.key().name()).date()
class BlogPost(db.Model):
# Name of the handler that serves this Model
HANDLER = 'PostHandler'
# The URL path to the blog post. Posts have a path iff they are published.
path = db.StringProperty()
title = db.StringProperty(required=True, indexed=False)
body_markup = db.StringProperty(choices=set(markup.MARKUP_MAP),
default=DEFAULT_MARKUP)
body = db.TextProperty()
tags = aetycoon.SetProperty(basestring, indexed=False)
published = db.DateTimeProperty()
created = db.DateTimeProperty(auto_now_add=True)
updated = db.DateTimeProperty(auto_now=False)
deps = aetycoon.PickleProperty()
draft = db.TextProperty()
is_deleted = db.BooleanProperty(default=False)
@aetycoon.TransformProperty(tags)
def normalized_tags(tags):
return list(set(utils.slugify(x.lower()) for x in tags))
@property
def tag_pairs(self):
return [(x, utils.slugify(x.lower())) for x in self.tags]
@property
def rendered(self):
"""Returns the rendered body."""
return markup.render_body(self)
@property
def summary(self):
"""Returns a summary of the blog post."""
return markup.render_summary(self)
@property
def hash(self):
val = (self.title, self.body, self.published)
return hashlib.sha1(str(val)).hexdigest()
@property
def summary_hash(self):
val = (self.title, self.summary, self.tags, self.published)
return hashlib.sha1(str(val)).hexdigest()
def set_key_name(self, key_name):
post_properties = BlogPost.properties()
del post_properties['normalized_tags']
new_post = BlogPost(
key_name=key_name,
**dict([(prop, getattr(self, prop)) for prop in post_properties]))
return new_post
def update(self, body, is_draft=False):
if is_draft:
self.draft = body
else:
memcache.flush_all()
self.updated = datetime.datetime.now()
self.draft = None
self.body = body
if not self.path and not is_draft:
# Post is being published for the first time
self.published = self.updated = datetime.datetime.now()
same_path = True
count = 0
while same_path:
path = utils.format_post_path(self, count)
same_path = BlogPost.get_by_key_name(path)
count += 1
self.path = path
if self.is_saved() or not is_draft:
new_post = self.set_key_name(path)
new_post.put()
self.is_saved() and self.delete()
BlogDate.create_for_post(new_post)
return new_post
if not self.is_saved():
new_post = self.set_key_name('/draft:' + utils.slugify(self.title))
new_post.put()
return new_post
self.put()
return self
def remove(self):
if not self.is_saved():
return
# TODO: Delete BlogDate if this is the only entry for the date.
self.is_deleted = True
self.put()
memcache.flush_all()
@classmethod
def get_prev_next(cls, post):
"""Retrieves the chronologically previous and next post for this post"""
q = cls.all().order('-published')
q.filter('is_deleted =', False)
q.filter('published <', post.published)
prev = q.get()
q = cls.all().order('published')
q.filter('is_deleted =', False)
q.filter('published >', post.published)
next = q.get()
return prev,next
class Page(db.Model):
# The URL path to the page.
path = db.StringProperty(required=True)
title = db.TextProperty(required=True)
template = db.StringProperty(required=True)
body = db.TextProperty(required=True)
created = db.DateTimeProperty(required=True, auto_now_add=True)
updated = db.DateTimeProperty()
@property
def rendered(self):
# Returns the rendered body.
return markup.render_body(self)
@property
def hash(self):
val = (self.path, self.body, self.published)
return hashlib.sha1(str(val)).hexdigest()
def publish(self):
self._key_name = self.path
self.put()
memcache.flush_all()
def remove(self):
if not self.is_saved():
return
self.delete()
memcache.flush_all()