from flask import Flask, request, session, g, redirect, url_for, \
    abort, render_template, flash, Blueprint
from flask.ext.mysqldb import MySQL
import datetime

admin = Blueprint('admin', __name__,
                        template_folder='templates')
import flask_site

@admin.route('/')
def home():
    if not session.get('logged_in'):
        return redirect(url_for('login'))
    else:
        return render_template('admin/home.html')

@admin.route('/posts')
def posts():
    if not session.get('logged_in'):
        return redirect(url_for('login'))
    else:
        g.db.execute('SELECT * FROM blog_posts ORDER BY updated_on DESC')
        entries = [dict(title=row[1],  category=row[3], tags=row[4], created=row[6].strftime("%d-%m-%Y"), updated=row[7].strftime("%d-%m-%Y")) for row in g.db.fetchall()]
        return render_template('admin/posts.html', entries=entries)

@admin.route('/new-post', methods=['GET', 'POST'])
def new_post():
    def format_post(text):
        code = False
        text = text.split("\n")
        new_text = []
        for line in text:
            if line[:4] == "<pre":
                code = True
                break
            if line[-6:] == "</pre>":
                code = False
                break
            if code == True:
                break
            new_text.append("<p>" + line.replace("'", "\\'") + "</p>")
        return "".join(new_text)
    if not session.get('logged_in'):
        return redirect(url_for('login'))
    else:
        if request.method == 'POST':
            formatted_text = format_post(request.form['text'])
            g.db.execute("insert into blog_posts (title, text, category, tags, url, created_on, updated_on) values ('{0}', '{1}', '{2}', '{3}', '{4}', '{5}', '{6}')" .format(request.form['title'], formatted_text, request.form['category'], request.form['tags'], request.form['title'].lower().replace(' ', '-').replace(',', '-').replace('--', '-'), str(datetime.datetime.now()), str(datetime.datetime.now())))
            flask_site.mysql.connection.commit()
            flash('New post added successfully')
            return redirect(url_for('blog'))
        else:
            return render_template('admin/new-post.html')

@admin.route('/edit-post', methods=['GET', 'POST'])
def edit_post():
    if not session.get('logged_in'):
        return redirect(url_for('login'))
    else:
        if request.method == 'POST':
            g.db.execute('UPDATE blog_posts SET (title, text, category, tags, updated_on) values (?, ?, ?, ?,  str(datetime.datetime.now()))',
            [request.form['title'], request.form['text'], request.form['category'], request.form['tags']])
            flask_site.mysql.connection.commit()
            flash('New post added successfully')
            return redirect(url_for('blog'))
        else:
            return render_template('admin/new-post.html')