import flask
from .. import auth
import urllib.parse

def checkForSession():
	if "uid" in flask.session:
		session = auth.validateSession(flask.session["uid"])
		if session != None:
			return session
	return None

def redirectAndSave(path):	
	flask.session["donePage"] = urllib.parse.urlparse(flask.request.url).path
	return response

def markRedirectAsRecovered():
	if "donePage" in flask.session:
		del flask.session["donePage"]
	else:
		raise KeyError("No redirect to be recovered from.")
		
def canRecoverFromRedirect():
	if "donePage" in flask.session:
		return flask.session["donePage"]
	return None


import flask
from .. import auth
import urllib.parse

def checkForSession():
	if "uid" in flask.session:
		session = auth.validateSession(flask.session["uid"])
		if session != None:
			return session
	return None

def redirectAndSave(path):	
	flask.session["donePage"] = urllib.parse.urlparse(flask.request.url).path
	return response

def recoverFromRedirect():
	donePage = flask.request.cookies["donePage"]
	response = flask.make_response(flask.redirect(donePage))	
	response.set_cookie("donePage", "", expires=0)
	return response

def canRecoverFromRedirect():
	if "donePage" in flask.session:
		return flask.session["donePage"]
	return None


--- 
+++ 
@@ -13,12 +13,12 @@
 	flask.session["donePage"] = urllib.parse.urlparse(flask.request.url).path
 	return response
 
-def recoverFromRedirect():
-	donePage = flask.request.cookies["donePage"]
-	response = flask.make_response(flask.redirect(donePage))	
-	response.set_cookie("donePage", "", expires=0)
-	return response
-
+def markRedirectAsRecovered():
+	if "donePage" in flask.session:
+		del flask.session["donePage"]
+	else:
+		raise KeyError("No redirect to be recovered from.")
+		
 def canRecoverFromRedirect():
 	if "donePage" in flask.session:
 		return flask.session["donePage"]
Remove unneeded recoverFromRedirect and add markRedirectAsRecovered
