From e3916e6403b8933d9d8896b7289321c45b9990d2 Mon Sep 17 00:00:00 2001
From: Munyoki Kilyungi
Date: Thu, 1 Sep 2022 17:15:06 +0300
Subject: Refactor if clause to check for string in list

* wqflask/wqflask/docs.py (update_text): Check for the e-mail in a
list.
---
 wqflask/wqflask/docs.py | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/wqflask/wqflask/docs.py b/wqflask/wqflask/docs.py
index 06f83c7f..5065135b 100644
--- a/wqflask/wqflask/docs.py
+++ b/wqflask/wqflask/docs.py
@@ -31,9 +31,8 @@ def update_text(start_vars):
     content = start_vars['ckcontent']
     content = content.replace('%', '%%').replace(
         '"', '\\"').replace("'", "\\'")
-
     try:
-        if g.user_session.record['user_email_address'] == "zachary.a.sloan@gmail.com" or g.user_session.record['user_email_address'] == "labwilliams@gmail.com":
+        if g.user_session.record.get('user_email_address') in ["zachary.a.sloan@gmail.com", "labwilliams@gmail.com"]:
             with database_connection() as conn, conn.cursor() as cursor:
                 cursor.execute("UPDATE Docs SET content=%s WHERE entry=%s",
                                (content, start_vars.get("entry_type"),))
-- 
cgit 1.4.1