app.py

from flask import abort, Flask, render_template, request, redirect, url_for, session
import datetime, os, urllib, urllib2, json, string
from werkzeug.utils import secure_filename

import utils
import hashlib
import user

CLIENT_ID = "INSERT CLIENT ID HERE"
CLIENT_SECRET = "INSERT CLIENT SECRET HERE"
Access_token = ""

#path for upload folder
path = "static/images"
app = Flask(__name__)

def validate_form(form, required_keys):
    """ Check if a dictionary contains all the required keys """
    return set(required_keys) <= set(form)

#login route
@app.route("/", methods=["POST", "GET"])
def index():
    if Access_token == "":
        req = urllib2.Request("https://api.clarifai.com/v1/token/")
        data = {"client_id": CLIENT_ID, "client_secret": CLIENT_SECRET, "grant_type": "client_credentials"}
        req.data = urllib.urlencode(data)
        u = urllib2.urlopen(req)
        response = u.read()
        data = json.loads(response)
        global Access_token
        Access_token = data["access_token"]
    paths = user.get_pics("all")
    images = []
    tags = []
    tagStrings = []
    count = 0
    for i in paths:
        images.append([])
        temp = i[0][13:]
        images[count].append(temp)
        count = count + 1
    for e in images:
        e.append(user.get_name("static/images"+e[0])[0][0])
        e.append(user.get_username(user.match_UID(e[1])))
        e.append(user.get_lat("static/images"+e[0])[0][0])
        e.append(user.get_lon("static/images"+e[0])[0][0])
    #get tags from database
    for i in range(len(images)):
        tags.append(user.get_tags(path+images[i][0]))
    for tag in tags:
        tagStrings.append(generateTags(tag[0]))
    if "username" not in session:
        return render_template("index.html", images=images, tagStrings=tagStrings)
    return render_template("index.html",username=session['username'], images=images, tagStrings=tagStrings)

def generateTags(tags):
    t = ""
    t = "#"+ " #".join(tags.rsplit(","))
    return t

#create a new account app route
@app.route("/register", methods=["POST", "GET"])
def register():
    if request.method == "POST":
        # User has submitted a request to register an account
        required_keys = ["username", "pass", "passconfirm"]
        if not validate_form(request.form, required_keys):
            return render_template("register.html", message="Malformed request.", category="danger")

        username = request.form["username"]
        password = request.form["pass"]
        password_confirm = request.form["passconfirm"]
        if len(password) <= 5:
            return render_template("register.html",message="Password must contain more than 5 characters.",category="danger")
        if not username.isalnum():
            return render_template("register.html", message="Usernames must contain only alphanumeric characters.", category="danger")

        if password != password_confirm:
            return render_template("register.html", message="Passwords do not match.", category="danger")

        if user.get_user(username=username):
            return render_template("register.html", message="Username is already in use.", category="danger")

        user.add_user(username, password)

        return render_template("register.html", message="Account created!", category="success")
    return render_template("register.html")

@app.route("/login", methods=["GET", "POST"])
def login():
    if request.method == "POST":
        # User has submitted a request to login
        required_keys = ["username", "pass"]
        if not validate_form(request.form, required_keys):
            return render_template("login.html", message="Malformed request.", category="danger")

        username = request.form["username"]
        password = hashlib.sha1(request.form["pass"]).hexdigest()

        result = user.get_user(username=username)
        if result:
            if result[2] == password:
                session["username"] = username
                return redirect(url_for("index"))
            return render_template("login.html", message="Invalid password", category="danger")
        return render_template("login.html", message="Username does not exist...", add_mess="Register a new account?", category="danger")
    return render_template("login.html")

@app.route("/logout")
def logout():
    session.clear()
    return redirect(url_for("index"))

#to check if uploaded file is an image
extensions = set(['png','jpg','jpeg','PNG','JPG','JPEG'])
def checkFile(filename):
    return '.' in filename and filename.rsplit('.',1)[1] in extensions

#generate either html for user to upload image OR to save uploaded image
@app.route("/upload")
def upload_local():
    return render_template("upload.html",upload="True")

@app.route("/upload/local", methods=['POST'])
def local():
    file = request.files['file']
    if file and checkFile(file.filename):
        fn = request.form['filename']
        name = stripPunctuation(fn)
        ext = file.filename.rsplit('.',1)[1]
        filename = secure_filename(name+"."+ext)
        filename = repeatedName(filename,0,False)
        file.save(os.path.join(path,filename))
        uid = user.get_UID(session['username'])
        link = utils.uploadPic (os.path.join(path,filename))
        tags = utils.getTags(link, Access_token)
        stringTags=",".join(tags)
        user.add_pic(os.path.join(path,filename),uid,fn,stringTags)
        return render_template("index.html",username=session['username'],message="Image Uploaded!",category="success")
    return render_template("upload.html",upload="True",message="Invalid File",category="danger")

@app.route("/upload/web", methods=['POST'])
def web():
    if checkFile(request.form['link']):
        ext = request.form['link'].rsplit('.',1)[1]
        response = urllib.urlopen(request.form['link'])
        image = response.read()
        filename = stripPunctuation(request.form['filename'])+"."+ext
        filename = secure_filename(filename)
        filename = repeatedName(filename,0,False)
        uid = user.get_UID(session['username'])
        with open(path+"/"+filename,"wb") as out:
            out.write(image)
            link = utils.uploadPic (os.path.join(path,filename))
            tags = utils.getTags(link, Access_token)
            stringTags=",".join(tags)
            user.add_pic(os.path.join(path,filename),uid,request.form['filename'],stringTags)
            return render_template("index.html",username=session['username'],message="Image Uploaded!", category="succe\
ss")
    return render_template("upload.html",upload="True",message="Invalid File",category="danger")

def stripPunctuation(name):
    for p in string.punctuation:
        name=name.replace(p,"")
    return name

def repeatedName(name,num,looped):
    if name in os.listdir(path):
        if looped:
            name = name.rsplit('.',1)[0][:-(len(str(num)))]+"."+name.rsplit('.',1)[1]
        print name
        name = name.rsplit('.',1)[0]+str(num)+"."+name.rsplit('.',1)[1]
        num += 1
        print name + ", " + str(num)
        return repeatedName(name,num,True)
    return name

@app.route("/profile")
def profile():
    if 'username' in session:
        paths = user.get_pics(user.get_UID(session['username']))
        images = []
        tags = []
        tagStrings = []
        count = 0
        for i in paths:
            images.append([])
            temp = i[0][13:]
            images[count].append(temp)
            count = count + 1
        for e in images:
            e.append(user.get_name("static/images"+e[0])[0][0])
            e.append(user.get_username(user.match_UID(e[1])))
        for i in range(len(images)):
            tags.append(user.get_tags(path+images[i][0]))
        for tag in tags:
            tagStrings.append(generateTags(tag[0]))
        return render_template("profile.html",images=images,username=session['username'],tagStrings=tagStrings)
    return redirect(url_for("login"))

if __name__=="__main__":
    if not os.path.exists("data.db"):
        db_builder.create_tables()
    app.debug = True

    # Generate and store secret key if it doesn't exist
    with open(".secret_key", "a+b") as f:
        secret_key = f.read()
        if not secret_key:
            secret_key = os.urandom(64)
            f.write(secret_key)
            f.flush()
        app.secret_key = secret_key

    app.run()