#!/usr/local/bin/python2.7

from flask import Flask, render_template, url_for, request, redirect, flash

import math
import random

app = Flask(__name__)
# necessary for flashing
app.secret_key = 'now is the winter of our discontent'

numRequests = 0

@app.route('/')
def welcome():
    global numRequests
    numRequests += 1
    exes = [ 'capitals/Paris',
                'sqrt/16.0',
                'sqrt/17.1',
                'print_stmt',
                'console_error',
                'no_view',
                'urls',
                'canonical',
                'canonical/',
                'fragile',
                'fragile/',
                'form/',
                'render/',
                'render_styled/',
                'login/',
                'logged_in/',
                'shoo/',
                'login_again/'
                ]
    links = [ '<li><a href="{url}">{url}</a></li>'.format(url=ex)
              for ex in exes ]
    return '''<h1>Welcome</h1>
<p>Hello World! These are the examples for lecture 2.
<p>Try <ul>{examples}</ul>
<p>There have been {n} requests since this app
started'''.format(n=numRequests,examples=''.join(links))

@app.route('/capitals/<city>')
def capitals(city):
    known = {'Paris':'France',
             'London':'England',
             'Madrid':'Spain',
             'Beijing':'China'}
    if city in known:
        return '<p>{city} is the capital of {country}'.format(city=city,country=known[city])
    else:
        return '''<p>I don't know the country whose capital is {city}'''.format(city=city)

@app.route('/sqrt/<float:n>')
def sqrt(n):
    return '<p>{x} is the square root of {y}'.format(x=math.sqrt(abs(n)),y=n)

@app.route('/print_stmt/')
def print_stmt():
    n = random.randint(1,10)
    print 'n is ',n
    return '<p>Can you guess your lucky number?'

@app.route('/console_error/')
def console_error():
    # this doesn't work
    roman = ['x','v','i','c','m','d','l']
    s = random.choice(roman)
    n = int(s)
    print '{roman} converts to {arabic}'.format(roman=s,arabic=n)

@app.route('/no_view/')
def no_view():
    n = random.randint(1,10)
    if n < 9:
        print '<p>Keep trying, you only scored a {n}'.format(n=n)
    else:
        return '<p>Wow, you scored a 10!!'
        
@app.route('/urls/')
def urls():
    hrefs = [ url_for('print_stmt'),
              url_for('capitals', city='London'),
              url_for('sqrt', n=18)
              ]
    links = [ '<p><a href="{h}">{h}</a>'.format(h=h)
              for h in hrefs ]
    return ''.join(links)

@app.route('/canonical/')
def canonical():
    return '<p>Well done!'

@app.route('/fragile')
def fragile():
    return '<p>Maybe?'

@app.route('/form/')
def form():
    f = '''<p><form method="{meth}" action="{script}">
<input type="text" name="comment">
<input type="submit" value="{meth}">
</form>'''
    f1 = f.format(meth='post',script=url_for('form_data'))
    f2 = f.format(meth='get', script=url_for('form_data'))
    return (f1+f2)

@app.route('/form_data/', methods=['POST','GET'])
def form_data():
    meth = request.method
    if meth == 'POST':
        stuff = request.form['comment']
    else:
        stuff = request.args.get('comment')
    return '''<p>Your form was submitted via {meth} and contained
<div style="border:2px solid green">{stuff}</div>'''.format(meth=meth,stuff=stuff)

@app.route('/render/')
def render():
    return render_template('trio.html',a='Harry',b='Ron',c='Hermione')

@app.route('/render_styled/')
def render_styled():
    return render_template('trio-styled.html',a='Harry',b='Ron',c='Hermione')

@app.route('/login/')
def login():
    return render_template('login.html', script=url_for('process_login'))

@app.route('/process_login/', methods=['POST'])
def process_login():
    if request.form['user'] == 'admin' and request.form['pass'] == 'secret':
        return redirect( url_for('logged_in') )
    else:
        return redirect( url_for('shoo') )

@app.route('/logged_in/')
def logged_in():
    return '<p>Welcome Admin!'

@app.route('/shoo/')
def shoo():
    return '<p>Wrong password; go away!'

@app.route('/login_again/', methods=['GET','POST'])
def login_again():
    if request.method == 'GET':
        return render_template('login_flashing.html',script=url_for('login_again'))
    else:
        if request.form['user'] == 'admin' and request.form['pass'] == 'secret':
            return redirect( url_for('logged_in') )
        else:
            flash("No, sorry; please try again")
            return redirect( url_for('login_again') )

if __name__ == '__main__':
    app.debug = True
    app.run('0.0.0.0',2716)