from __future__ import unicode_literals

from flask import (
    Flask, render_template, url_for, redirect, request, session, flash
)
from flask.ext.script import Manager, Server
from flask.ext.bootstrap import Bootstrap

import forms

app = Flask(__name__)
server = Server(host='0.0.0.0', port='5000')
app.secret_key = 'Some mega ultra hiper secret key'
app.debug = True
manager = Manager(app)
manager.add_command('runserver', server)
bootstrap = Bootstrap(app)

IMAGE_EXTENSIONS = ['.jpg', '.jpeg', '.gif', '.png']


@app.route('/')
def index():

    if 'USERNAME' in session:
        return render_template('index.html', name = session['USERNAME'])
    else:
        return render_template('index.html')
    """
        if user is logged:
            render template index.html with user name in "name" parameter
        else:
            just render template index.html
    """


@app.route('/browse')
def browse():
    """
        render template with get_images function in "images" parameter
    """


@app.route('/register', methods=['GET', 'POST'])
def register():
    """
        function to register users.
        Checks if user already exists in users.txt using user_exists function.
        if user exists:
            flash message: "Sorry but user with given username already exists"
            and redirect to register.html
        Rendered based on RegistrationForm.
    """


@app.route('/login', methods=['GET', 'POST'])
def login():
    form = forms.LoginForm(request.form)

    if request.method == 'POST':
        username = request.form.get('username','')
        password = request.form.get('password','')
        if (username, password) in ['TEST','TEST']:
            session['USERNAME'] = 'TEST'
            return redirect(url_for('index'))
    else:
        return render_template('login.html', form=form)







    """
        if user provide corrected login and password combination:
            flash message: "You have been logged successfully"
            and redirect to index.html
        else:
            render login.html template with LoginForm as "form" parameter
    """


@app.route('/logout')
def logout():
    """
        Pop user from session, send flash message:
        "You have been logged out succesfully"
        and redirect to index page
    """


def validate_password(username, password):
    """
        Helper function for login().
        Checks if username and password combination match to data in users.txt
        @param username: user name
        @param password: user password
        @return: True or False
    """


def user_exists(username):
    """
        Helper function for register().
        Checks if user exists in users.txt
        @param username: user name
        @return: True or False
    """


def get_images():
    """
        Helper function for browse()
        Creates reversed list with urls to images taken from images.txt file
    """


@app.errorhandler(404)
def page_not_found(e):
    """
        render template 404.html and send 404 http status
    """


@app.errorhandler(500)
def internal_server_error(e):
    """
        render template 500.html and send 500 http status
    """


@app.route('/add', methods=['GET', 'POST'])
def add():
    """
        Render add.html with forms.AdditionForm(request.form)as "form" param
        if form is correct open images.txt file and add url to next line, then
        flash message: 'Photo added succesfully' and redirect to browse.html
    """


if __name__ == '__main__':
    manager.run()