#controllers/category.py
import config, lib, datetime, uuid
from pytz import timezone
from bottle import route, template, request, redirect, response
from models import categorydb, settingdb, postdb
def getTimeZone():
khtz = timezone('Asia/Phnom_Penh')
date = datetime.datetime.now().astimezone(tz=khtz).strftime('%d-%m-%Y')
time = datetime.datetime.now().astimezone(tz=khtz).strftime('%H:%M:%S')
return (date, time)
@route('/category')
def post():
config.reset(settingdb.select())
config.kargs['blogTitle'] = "ទំព័រជំពូក"
config.kargs['posts'] = categorydb.select(config.kargs['dashboardPostLimit'])
config.kargs['thumbs'] = lib.getPostThumbs(config.kargs['posts'])
config.kargs['datetime'] = getTimeZone()
config.kargs['page'] = 1
author = request.get_cookie("logged-in", secret=config.kargs['secretKey'])
if author:
config.kargs['author'] = author
config.kargs['showEdit'] = True
return template('dashboard/category', data=config.kargs)
@route('/category/<id:int>')
def post(id):
config.reset(settingdb.select())
config.kargs['blogTitle'] = "ទំព័រការផ្សាយ"
config.kargs['post'] = categorydb.select(1, id)
config.kargs['posts'] = categorydb.select(config.kargs['frontPagePostLimit'])
print(len(config.kargs['posts']))
config.kargs['thumbs'] = lib.getPostThumbs(config.kargs['posts'])
config.kargs['page'] = 1
author = request.get_cookie("logged-in", secret=config.kargs['secretKey'])
if author:
config.kargs['showEdit'] = True
return template('category', data=config.kargs)
@route('/categories/<name>')
def category(name):
config.reset(settingdb.select())
config.kargs['blogTitle'] = "ទំព័រជំពូក"
config.kargs['category'] = name
config.kargs['posts'] = postdb.select(config.kargs['categoryPostLimit'], category=name)
config.kargs['thumbs'] = lib.getPostThumbs(config.kargs['posts'])
config.kargs['page'] = 1
author = request.get_cookie("logged-in", secret=config.kargs['secretKey'])
if author:
config.kargs['showEdit'] = True
return template('categories', data=config.kargs)
@route('/categorizing', method="POST")
def posting():
author = request.get_cookie("logged-in", secret=config.kargs['secretKey'])
if ((author != "Guest") and categorydb.check(author)):
title = request.forms.getunicode('fpost-title')
if title == "":
title = "untitled"
postdate = request.forms.getunicode('fpost-date')
posttime = request.forms.getunicode('fpost-time')
content = request.forms.getunicode('fcontent')
try:
postdate = datetime.datetime.strptime(postdate, "%d-%m-%Y")
except ValueError:
config.kargs['message'] = 'ទំរង់កាលបរិច្ឆេទមិនត្រឹមត្រូវ!'
return template('dashboard/category', data=config.kargs)
try:
posttime = datetime.datetime.strptime(posttime, "%H:%M:%S")
except ValueError:
config.kargs['message'] = 'ទំរង់ពេលវេលាមិនត្រឹមត្រូវ!'
return template('dashboard/category', data=config.kargs)
if 'postId' in config.kargs:
id = config.kargs['postId']
categorydb.update(title, postdate, posttime, content, id)
del config.kargs['postId']
else:
categorydb.insert(str(uuid.uuid4().int), title, author, postdate, posttime, content)
redirect('/category')
@route('/category/delete/<id:int>')
def delete(id):
author = request.get_cookie("logged-in", secret=config.kargs['secretKey'])
if ((author != "Guest") and categorydb.check(author)):
categorydb.delete(id)
redirect('/category')
@route('/category/edit/<id:int>')
def edit(id):
author = request.get_cookie("logged-in", secret=config.kargs['secretKey'])
if ((author != "Guest") and categorydb.check(author)):
config.reset(settingdb.select())
config.kargs['blogTitle'] = "ទំព័រកែតំរូវ"
config.kargs['posts'] = categorydb.select(config.kargs['dashboardPostLimit'])
config.kargs['thumbs'] = lib.getPostThumbs(config.kargs['posts'])
config.kargs['post'] = categorydb.select(1, id)
config.kargs['edit'] = True
config.kargs['postId'] = id
config.kargs['page'] = 1
return template('dashboard/category', data=config.kargs)
redirect('/category')
@route('/category/paginate/<place>')
def paginate(place):
if place == "backend":
postLimit = config.kargs['dashboardPostLimit']
else:
postLimit = config.kargs['frontPagePostLimit']
posts = categorydb.select(postLimit, page=config.kargs['page'])
def toString(post):
post[3] = post[3].strftime('%d-%m-%Y')
post[4] = post[4].strftime('%H:%M:%S')
if posts:
config.kargs['page'] += 1
posts = [list(obj) for obj in posts ]
[toString(obj) for obj in posts]
thumbs = lib.getPostThumbs(posts)
print(posts)
return {'json':posts, 'thumbs':thumbs}
else:
return {'json':0}
@route('/categories/paginate/<category>')
def paginate(category):
posts = postdb.select(config.kargs['categoryPostLimit'], category=category, page=config.kargs['page'])
def toString(post):
post[3] = post[3].strftime('%d-%m-%Y')
post[4] = post[4].strftime('%H:%M:%S')
if posts:
config.kargs['page'] += 1
posts = [list(obj) for obj in posts ]
[toString(obj) for obj in posts]
thumbs = lib.getPostThumbs(posts)
print(posts)
return {'json':posts, 'thumbs':thumbs}
else:
return {'json':0}
#models/postdb.py
import os
import psycopg2
def createTable():
if 'DYNO' in os.environ:
DATABASE_URL = os.environ['DATABASE_URL']
conn = psycopg2.connect(DATABASE_URL, sslmode='require')
cursor = conn.cursor()
else:
conn = psycopg2.connect(
database="postgres",
user="postgres",
password="sokhavuth",
host="localhost",
port="5432"
)
cursor = conn.cursor()
SQL = '''CREATE TABLE IF NOT EXISTS POST(
ID TEXT,
TITLE TEXT,
AUTHOR TEXT,
POSTDATE DATE,
POSTTIME TIME,
CATEGORY TEXT,
CONTENT TEXT
)'''
cursor.execute(SQL)
conn.commit()
conn.close()
def insert(*post):
createTable()
if 'DYNO' in os.environ:
DATABASE_URL = os.environ['DATABASE_URL']
conn = psycopg2.connect(DATABASE_URL, sslmode='require')
cursor = conn.cursor()
else:
conn = psycopg2.connect(
database="postgres",
user="postgres",
password="sokhavuth",
host="localhost",
port="5432"
)
cursor = conn.cursor()
cursor.execute("INSERT INTO POST (ID, TITLE, AUTHOR, POSTDATE, POSTTIME, CATEGORY, CONTENT) VALUES %s ", (post,))
conn.commit()
conn.close()
def select(amount, id=None, page=0, category=0):
createTable()
if 'DYNO' in os.environ:
DATABASE_URL = os.environ['DATABASE_URL']
conn = psycopg2.connect(DATABASE_URL, sslmode='require')
cursor = conn.cursor()
else:
conn = psycopg2.connect(
database="postgres",
user="postgres",
password="sokhavuth",
host="localhost",
port="5432"
)
cursor = conn.cursor()
if id and (amount == 1):
cursor.execute("SELECT * FROM POST WHERE ID = '" + str(id) +"'")
elif page:
SQL = "SELECT * FROM POST ORDER BY POSTDATE DESC, POSTTIME DESC OFFSET %s ROWS FETCH NEXT %s ROWS ONLY"
cursor.execute(SQL, (amount*page, amount))
elif category:
cursor.execute("SELECT * FROM POST WHERE CATEGORY = '" + category +"' ORDER BY POSTDATE DESC, POSTTIME DESC LIMIT " + str(amount))
elif category and page:
SQL = "SELECT * FROM POST WHERE CATEGORY = '%s' ORDER BY POSTDATE DESC, POSTTIME DESC OFFSET %s ROWS FETCH NEXT %s ROWS ONLY"
cursor.execute(SQL, (category, amount*page, amount))
else:
cursor.execute("SELECT * FROM POST ORDER BY POSTDATE DESC, POSTTIME DESC LIMIT " + str(amount))
result = cursor.fetchall()
return result
def check(username):
if 'DYNO' in os.environ:
DATABASE_URL = os.environ['DATABASE_URL']
conn = psycopg2.connect(DATABASE_URL, sslmode='require')
cursor = conn.cursor()
else:
conn = psycopg2.connect(
database="postgres",
user="postgres",
password="sokhavuth",
host="localhost",
port="5432"
)
cursor = conn.cursor()
cursor.execute("SELECT USERNAME FROM USERS WHERE USERNAME = '"+ username + "' LIMIT 1")
result = cursor.fetchone()
if result:
return True
else:
return False
def delete(id):
if 'DYNO' in os.environ:
DATABASE_URL = os.environ['DATABASE_URL']
conn = psycopg2.connect(DATABASE_URL, sslmode='require')
cursor = conn.cursor()
else:
conn = psycopg2.connect(
database="postgres",
user="postgres",
password="sokhavuth",
host="localhost",
port="5432"
)
cursor = conn.cursor()
cursor.execute("DELETE FROM POST WHERE ID = '" + str(id) + "'")
conn.commit()
conn.close()
def update(*args):
if 'DYNO' in os.environ:
DATABASE_URL = os.environ['DATABASE_URL']
conn = psycopg2.connect(DATABASE_URL, sslmode='require')
cursor = conn.cursor()
else:
conn = psycopg2.connect(
database="postgres",
user="postgres",
password="sokhavuth",
host="localhost",
port="5432"
)
cursor = conn.cursor()
sql = "UPDATE POST SET TITLE = %s, POSTDATE = %s, POSTTIME = %s, CATEGORY = %s, CONTENT = %s WHERE ID = '%s' "
cursor.execute(sql, args)
conn.commit()
conn.close()
def search(query):
createTable()
if 'DYNO' in os.environ:
DATABASE_URL = os.environ['DATABASE_URL']
conn = psycopg2.connect(DATABASE_URL, sslmode='require')
cursor = conn.cursor()
else:
conn = psycopg2.connect(
database="postgres",
user="postgres",
password="sokhavuth",
host="localhost",
port="5432"
)
cursor = conn.cursor()
sql = "SELECT * from POST WHERE"
sql += " TITLE LIKE '%"+query+"%'"
sql += " OR CONTENT LIKE '%"+query+"%'"
sql += " ORDER BY POSTDATE DESC, POSTTIME DESC LIMIT 20"
cursor.execute(sql)
result = cursor.fetchall()
return result
GitHub: https://github.com/Sokhavuth/kwblog
Heroku: https://khmerweb-kwblog.herokuapp.com/
