193 lines
8.4 KiB
Python
193 lines
8.4 KiB
Python
# -*- coding: utf-8 -*-
|
|
# vim: autoindent shiftwidth=4 expandtab textwidth=80 tabstop=4 softtabstop=4
|
|
|
|
###############################################################################
|
|
# ScribeEngine - Open Source Blog Software #
|
|
# --------------------------------------------------------------------------- #
|
|
# Copyright (c) 2010 Raoul Snyman #
|
|
# --------------------------------------------------------------------------- #
|
|
# This program is free software; you can redistribute it and/or modify it #
|
|
# under the terms of the GNU General Public License as published by the Free #
|
|
# Software Foundation; version 2 of the License. #
|
|
# #
|
|
# This program is distributed in the hope that it will be useful, but WITHOUT #
|
|
# ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or #
|
|
# FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for #
|
|
# more details. #
|
|
# #
|
|
# You should have received a copy of the GNU General Public License along #
|
|
# with this program; if not, write to the Free Software Foundation, Inc., 59 #
|
|
# Temple Place, Suite 330, Boston, MA 02111-1307 USA #
|
|
###############################################################################
|
|
|
|
import logging
|
|
from datetime import datetime
|
|
from pprint import pformat
|
|
|
|
from sqlalchemy.sql import or_
|
|
from pytz import timezone
|
|
|
|
from scribeengine.lib.base import *
|
|
from scribeengine.lib import utils
|
|
from scribeengine.model import Post, Comment, Tag
|
|
from scribeengine.model.meta import Session
|
|
|
|
log = logging.getLogger(__name__)
|
|
|
|
class BlogController(BaseController):
|
|
|
|
def __before__(self):
|
|
BaseController.__before__(self)
|
|
self._add_javascript(u'jquery.elastic.js')
|
|
self._add_javascript(u'ScribeEngine.Blog.js')
|
|
|
|
def index(self):
|
|
posts = Session.query(Post)\
|
|
.filter_by(status=u'published')\
|
|
.order_by(Post.created.desc())
|
|
pagination = utils.paginate(posts, 10,
|
|
int(request.GET.get(u'page', 1)), '/')
|
|
c.posts = pagination[u'records']
|
|
if pagination[u'prev'] != pagination[u'page']:
|
|
c.first_page = pagination[u'first']
|
|
c.prev_page = pagination[u'prev']
|
|
if pagination[u'next'] != pagination[u'page']:
|
|
c.next_page = pagination[u'next']
|
|
c.last_page = pagination[u'last']
|
|
c.list_start = pagination[u'start']
|
|
c.list_total = pagination[u'total']
|
|
c.list_end = pagination[u'end']
|
|
return render(u'/blog/index.mako')
|
|
|
|
def archive(self, year=None, month=None, day=None):
|
|
if day and month and year:
|
|
start_date = datetime(int(year), int(month), int(day), 0, 0, 0, 0)
|
|
end_date = datetime(int(year), int(month), int(day), 23, 59, 59, 99999)
|
|
c.datestring = start_date.strftime('%d %B %Y')
|
|
if c.datestring[0] == u'0':
|
|
c.datestring = c.datestring[1:]
|
|
elif month and year and not day:
|
|
start_date = utils.month_first_day(datetime(int(year), int(month), 1))
|
|
end_date = utils.month_last_day(datetime(int(year), int(month), 1))
|
|
c.datestring = start_date.strftime('%B %Y')
|
|
elif year and not month:
|
|
start_date = datetime(int(year), 1, 1, 0, 0, 0, 0)
|
|
end_date = datetime(int(year), 12, 31, 23, 59, 59, 99999)
|
|
c.datestring = start_date.strftime('%Y')
|
|
else:
|
|
start_date = None
|
|
end_date = None
|
|
c.datestring = u'all time'
|
|
c.page_title = u'Archive for %s.' % c.datestring
|
|
posts = Session.query(Post)
|
|
if start_date and end_date:
|
|
posts = posts\
|
|
.filter(Post.created >= start_date)\
|
|
.filter(Post.created <= end_date)
|
|
posts = posts.order_by(Post.created.desc())
|
|
pagination = utils.paginate(posts, 10,
|
|
int(request.GET.get(u'page', 1)), '/')
|
|
c.posts = pagination[u'records']
|
|
if pagination[u'prev'] != pagination[u'page']:
|
|
c.first_page = pagination[u'first']
|
|
c.prev_page = pagination[u'prev']
|
|
if pagination[u'next'] != pagination[u'page']:
|
|
c.next_page = pagination[u'next']
|
|
c.last_page = pagination[u'last']
|
|
c.list_start = pagination[u'start']
|
|
c.list_total = pagination[u'total']
|
|
c.list_end = pagination[u'end']
|
|
return render(u'/blog/archive.mako')
|
|
|
|
def view(self, url):
|
|
c.post = Session.query(Post)\
|
|
.filter_by(url=url)\
|
|
.filter_by(status=u'published')\
|
|
.first()
|
|
c.page_title = c.post.title
|
|
return render(u'/blog/view.mako')
|
|
|
|
def tag(self, id=None):
|
|
if not id:
|
|
h.redirect_to('/')
|
|
c.tag = Session.query(Tag).filter_by(url=id).first()
|
|
if not c.tag:
|
|
h.redirect_to('/')
|
|
c.page_title = u'Blog posts with tag: %s' % c.tag.name
|
|
return render('/blog/tag.mako')
|
|
|
|
@authenticate()
|
|
def comment_POST(self, id):
|
|
if not id:
|
|
h.flash.set_message(u'There was a problem submitting your comment.', u'error')
|
|
h.redirect_to('/')
|
|
post = Session.query(Post).get(id)
|
|
if not post or post.comment_status != u'open':
|
|
h.flash.set_message(u'There was a problem submitting your comment.', u'error')
|
|
h.redirect_to('/')
|
|
comment = Comment(
|
|
user = c.current_user,
|
|
title = c.form_values[u'title'],
|
|
body = c.form_values[u'body']
|
|
)
|
|
post.comments.append(comment)
|
|
Session.add(post)
|
|
Session.commit()
|
|
h.flash.set_message(u'Successfully submitted your comment.', u'success')
|
|
h.redirect_to(h.url_for_post(post))
|
|
|
|
def search(self):
|
|
c.querystring = request.GET.get(u'q')
|
|
if not c.querystring:
|
|
h.flash.set_message(u'You didn\'t supply anything to search for.', u'error')
|
|
h.redirect_to('/')
|
|
kwprocessor = utils.KeywordProcessor(
|
|
groups=[None, '+', '-'],
|
|
group=tuple,
|
|
normalize=lambda s: s.strip(' \"\'')
|
|
)
|
|
keywords, ands, nots = kwprocessor.split(c.querystring)
|
|
or_clauses = []
|
|
for kw in keywords:
|
|
or_clauses.append(Post.body.contains(kw.strip()))
|
|
or_clauses.append(Post.title.contains(kw.strip()))
|
|
and_clauses = [or_(Post.body.contains(aw.strip()),
|
|
Post.title.contains(aw.strip())) for aw in ands]
|
|
not_clauses = [or_(Post.body.contains(nw.strip()),
|
|
Post.title.contains(nw.strip())) for nw in nots]
|
|
c.posts = Session.query(Post)
|
|
if len(or_clauses) > 0:
|
|
c.posts = c.posts.filter(or_(*or_clauses))
|
|
if len(and_clauses) > 0:
|
|
for and_clause in and_clauses:
|
|
c.posts = c.posts.filter(and_clause)
|
|
if len(not_clauses) > 0:
|
|
for not_clause in not_clauses:
|
|
c.posts = c.posts.filter(~not_clause)
|
|
c.posts = c.posts.order_by(Post.created.desc()).all()
|
|
c.page_title = u'Search'
|
|
return render(u'/blog/search.mako')
|
|
|
|
def calendar(self, year, month):
|
|
#c.calendar = Calendar(6)
|
|
#c.today = datetime.today()
|
|
server_tz = timezone(config.get(u'server.timezone', u'UTC'))
|
|
user_tz = c.current_user.timezone if c.current_user and c.current_user.timezone else u'UTC'
|
|
now = datetime.now(server_tz).astimezone(timezone(user_tz))
|
|
c.thismonth = now.replace(int(year), int(month))
|
|
c.prev_month = c.thismonth - monthdelta(1)
|
|
c.next_month = c.thismonth + monthdelta(1)
|
|
month_start = datetime(c.thismonth.year, c.thismonth.month, 1, 0, 0, 0, 0)
|
|
month_end = c.next_month.replace(day=1, hour=23, minute=59, second=59, microsecond=9999) - timedelta(seconds=1)
|
|
posts = Session.query(Post)\
|
|
.filter(Post.created >= month_start)\
|
|
.filter(Post.created <= month_end)\
|
|
.all()
|
|
c.month_posts = {}
|
|
for post in posts:
|
|
if post.created.day not in c.month_posts:
|
|
c.month_posts[post.created.day] = []
|
|
c.month_posts[post.created.day].append(post)
|
|
return render(u'/calendar.mako')
|
|
|