server.py

from tornado.ncss import Server
from re_template_renderer import render_template
#uncomment later when DB is fixed
from db.plutonium import User,Post,Comment,database_connect

#to create street.db, double click create_db.py

###DECORATORS###
def loginRequired(fn):
    def inner(response, *args, **kwargs):
        user = response.get_secure_cookie('userCookie')
        if user is None:
            response.redirect('/login')
        else:
            return fn(response, *args, **kwargs)
    return inner

def notLoginRequired(fn):
    def inner(response, *args, **kwargs):
        user = response.get_secure_cookie('userCookie')
        if user is None:
            return fn(response, *args, **kwargs)
        else:
            response.redirect('/home')
    return inner

###REGULAR REFERENCE FUNCTIONS###

def home(response):
    user = get_current_user(response)
    posts = Post.get_by_recent(10)
    print(posts)
    html = render_template('main.html', {'user': user, 'posts': posts})
    response.write(html)

def login_handler(response):
    email = response.get_field("email")
    password = response.get_field("password")
    try:
        user = User.login(email, password)
        response.set_secure_cookie('userCookie', email)
        response.redirect('/home')
    except ValueError:
        user = get_current_user(response)
        html = render_template('login.html', {'user': user, 'invalidUser': "Invalid login." })
        response.write(html)

def signup_handler(response):
    name = response.get_field('name', '')
    email = response.get_field('email', '')
    password = response.get_field('password', '')
    confpassword = response.get_field('confpassword', '')
    usr_description = response.get_field('usr_describe', '')
    
    #when done with mvp keep entered fields
    if (not name) or (not email) or (not password) or (not confpassword):
        user = get_current_user(response)
        html = render_template('signup.html', {'user': user, 'errorMessage': "You must fill in all fields. Please try again." })
        response.write(html)
    elif password != confpassword:
        user = get_current_user(response)
        html = render_template('signup.html', {'user': user, 'errorMessage': "Password did not match. Please try again." })
        response.write(html)
    elif len(password) < 6:
        user = get_current_user(response)
        html = render_template('signup.html', {'user': user, 'errorMessage': "Password must be longer than 6 characters. Please try again." })
        response.write(html)
    elif password.lower() == 'password':
        user = get_current_user(response)
        html = render_template('signup.html', {'user': user, 'errorMessage': "'Password' is far to weak. Please submit a different password." })
        response.write(html)
    else:
        try:
            user = User.register(email,password,name,usr_description)
            response.set_secure_cookie('userCookie', email)
            response.redirect('/home')
        except ValueError as e:
            user = get_current_user(response)
            html = render_template('signup.html', {'user': user, 'errorMessage': str(e) })
            response.write(html)

def profile(response,name):
    user = get_current_user(response)
    print(user)
    print(user.profile_picture)
    html = render_template('profile.html', {'user': user})
    response.write(html)

def get_current_user(response):
    email = response.get_secure_cookie("userCookie")    #change back to User(), later when DB is fixed
    if email is not None:
        email = email.decode()
        user = User.get(email)
        return user
    return None

@loginRequired
def view_post(response, post_id):
    try:
        post = Post.get(post_id)
        user = get_current_user(response)
        comments = Post.comments(post_id)
        for c in comments:
            c.author = User.get_by_id(c.author)

        poster = User.get_by_id(post.author_id)
        html = render_template('content.html', {'user': user,'post':post,'poster':poster, 'comments': comments})
        response.write(html)
    except Exception as e:
        print(e)
        user = get_current_user(response)
        html = render_template('404errorpage.html', {'user': user})
        response.write(html)

def demo(response):
    user = get_current_user(response)
    html = render_template('demo.html', {'user': user, 'comments':['great', 'meh']})
    response.write(html)

def notfound(response):
    user = get_current_user(response)
    html = render_template('404errorpage.html', {'user': user})
    response.write(html)

###NOT LOGGED IN EXCLUSIVE PAGES###
@notLoginRequired
def login(response):
    user = get_current_user(response)
    html = render_template('login.html', {'user': user})
    response.write(html)

@notLoginRequired
def signup(response):
    user = get_current_user(response)
    html = render_template('signup.html', {'user': user})
    response.write(html)

###LOGIN EXCLUSIVE PAGES###
@loginRequired
def submit(response):
    user = get_current_user(response)
    html = render_template('new_post.html', {'user': user})
    response.write(html)

@loginRequired
def logout(response):
    response.clear_cookie("userCookie")
    response.redirect('/home')

@loginRequired
def profile(response,name):
    user = get_current_user(response)
    posts = User.get_posts(user.user_id)
    html = render_template('profile.html', {'user': user, 'posts': posts})
    response.write(html)

@loginRequired
def submit_handler(response):
    user = get_current_user(response)
    title = response.get_field("title")
    location = response.get_field("location")
    image = response.get_file("postImage")
    description = response.get_field("description")

    if (not title) or (not location) or (not image) or (not description) or (image[1] is None):
        html = render_template('new_post.html', {'user': user, 'invalidPost': "Please fill in all fields." })
        response.write(html)
    else:
        pictureName = 'static/postimages/'+title+'.'+image[1].split('/')[1]
        with open(pictureName,'wb') as pictureFile:
            pictureFile.write(image[2])
        createPost = Post.create(user.user_id,title,description,pictureName,location)
        response.redirect("/post/"+str(createPost.id))

@loginRequired
def comment_post(response, post_id):
    user = get_current_user(response)
    comment = response.get_field("comment")

    if comment:
        comment = Comment.create(user.user_id, post_id, comment)
    response.redirect('/post/' + post_id)

database_connect('db/street.db')

server = Server()
server.register(r'/?(?:home)?', home)
server.register(r'/profile(?:(?:/([\w\.\-]+)?)|/?)', profile)
server.register(r'/login', login, post=login_handler)
server.register(r'/signup',signup, post=signup_handler)
server.register(r'/post/(\d+)', view_post, post=comment_post)
server.register(r'/submit',submit, post=submit_handler)
server.register(r'/demo',demo)
server.register(r'/logout',logout)
server.register(r'.+',notfound)

server.run()