From 99b747d4ba835f2a4e80d4cc715c67669334e6c8 Mon Sep 17 00:00:00 2001
From: sqozz <sqozz@geekify.de>
Date: Thu, 11 Feb 2016 01:35:07 +0100
Subject: [PATCH] Auto prepend the default scheme (http://) before redirect

---
 schort.py | 12 ++++++++----
 1 file changed, 8 insertions(+), 4 deletions(-)

diff --git a/schort.py b/schort.py
index 9e5b0a2..66607c6 100755
--- a/schort.py
+++ b/schort.py
@@ -1,6 +1,8 @@
 #!/usr/bin/env python3
 from flask import Flask, render_template, url_for, request, redirect, abort
 import sqlite3, random, string, time, hashlib, base64
+from urllib.parse import urlparse
+
 app = Flask(__name__)
 
 @app.route('/', methods=['GET', 'POST'])
@@ -11,7 +13,11 @@ def short(shortLink=""):
 		c = conn.cursor()
 		result = c.execute('SELECT * FROM links WHERE shortLink=?', (shortLink, )).fetchone()
 		if result:
-			return redirect(result[1], code=301) # Redirect to long URL saved in the database
+			url = result[1]
+			parsedUrl = urlparse(url)
+			if parsedUrl.scheme == "":
+				url = "http://" + url
+			return redirect(url, code=301) # Redirect to long URL saved in the database
 		else:
 			message = ""
 			if len(shortLink) > 0:
@@ -65,9 +71,7 @@ def initDB():
 	c.execute('''CREATE TABLE IF NOT EXISTS links (shortLink UNIQUE NOT NULL, longLink, timestamp, ip, redirectMethod);''')
 	conn.commit()
 	conn.close()
-	print("DB init")
 
 if __name__ == '__main__':
 	initDB()
-#	app.run(debug=True)
-	app.run(debug=False)
+	app.run(debug=True) # If you call this file directly it will always run in debug mode. THIS IS VERY DANGEROUS!