from flask import Flask
from flask import request
from flask import redirect
import subprocess
import traceback

app = Flask(__name__)

SITE_URL='/var/www/tsundere.me/'

@app.route('/')
def index():
    if len(request.query_string) > 0:
        try:
            if not request.query_string.startswith("http:") and not request.query_string.startswith("https:"):
                return app.send_static_file("invalid.htm")
            tmp_file = subprocess.check_output([SITE_URL + 'bin/tsundere.me', request.query_string]).strip()
            print "[upload] %s -> %s" % (request.query_string, tmp_file)
            if tmp_file == "bad image":
                return app.send_static_file("badimage.htm")
            if tmp_file == "no faces":
                return app.send_static_file("noface.htm")
            return redirect(tmp_file, code=302)
        except Exception as e:
            traceback.print_exc()
            return app.send_static_file("error.htm")
    else:
        return app.send_static_file("index.htm")