1 files changed, 31 insertions, 0 deletions
diff --git a/pgcommitfest/userprofile/util.py b/pgcommitfest/userprofile/util.py
new file mode 100644
index 0000000..af50caf
--- /dev/null
+++ b/pgcommitfest/userprofile/util.py
@@ -0,0 +1,31 @@
+from Crypto.Hash import SHA256
+from Crypto import Random
+
+from models import UserProfile
+
+def generate_random_token():
+	"""
+	Generate a random token of 64 characters. This token will be
+	generated using a strong random number, and then hex encoded to make
+	sure all characters are safe to put in emails and URLs.
+	"""
+	s = SHA256.new()
+	r = Random.new()
+	s.update(r.read(250))
+	return s.hexdigest()
+
+
+class UserWrapper(object):
+	def __init__(self, user):
+		self.user = user
+
+	@property
+	def email(self):
+		try:
+			up = UserProfile.objects.get(user=self.user)
+			if up.selectedemail and up.selectedemail.confirmed:
+				return up.selectedemail.email
+			else:
+				return self.user.email
+		except UserProfile.DoesNotExist:
+			return self.user.email