From ef8c86710ba665b565f12805d1de654d8a6aac31 Mon Sep 17 00:00:00 2001
From: Corder Guy <corder@tamu.edu>
Date: Sat, 28 Apr 2018 20:26:48 -0500
Subject: [PATCH] Add avatar column and getter/setter

---
 db.sqlite3      | Bin 8192 -> 8192 bytes
 mocha_server.py |  39 ++++++++++++++++++++++++++++++++++-----
 2 files changed, 34 insertions(+), 5 deletions(-)

diff --git a/db.sqlite3 b/db.sqlite3
index 2323ffb322d4db12e88995bfde29a9dad99277ee..59f2cb0af603952e4e16cd8df3d7141acb3b38c5 100755
GIT binary patch
delta 129
zcmZp0XmFSyEvUf2z`z8=Fu*!d$CzJ%K~G^OFHne)KbC<%c4OgtKCVUuMs{&=amFV3
z$@TmV96Ab#Wr-z;MU&t2i?ft7aFuW75{Tnc<NwZ}1~Upsi!!n?h%)iYm!uY#Fn-BP
NEr)2@+$UAU3IG8r8&?1T

delta 101
zcmZp0XmFSyEhx>vz`z8=Fu*cV$CzK5LHFYhUZ4;oe<=fh>Bho0d=nd_xf(eb*~P`h
w85{E_r}H;V{?E_O(#XKoxS2~Jj%#9rBri83D}yK#Z%Jx#3FDW|eNsiN0QDsn<^TWy

diff --git a/mocha_server.py b/mocha_server.py
index 7e27ef6..f324350 100755
--- a/mocha_server.py
+++ b/mocha_server.py
@@ -161,6 +161,37 @@ def update(user_id, score):
     return '[]'
 
 
+def set_avatar(user_id, new_avatar):
+    conn = sqlite3.connect(DATABASE)
+    conn.row_factory = sqlite3.Row
+    cursor = conn.cursor()
+    cursor.execute('UPDATE users SET avatar = (?) WHERE user_id = (?)',
+                   (new_avatar, user_id))
+    conn.commit()
+    conn.close()
+
+    return '[]'
+
+
+def get_avatar(user_id):
+    conn = sqlite3.connect(DATABASE)
+    conn.row_factory = sqlite3.Row
+    cursor = conn.cursor()
+    cursor.execute(
+        'SELECT avatar FROM users WHERE user_id = (?)',
+        (user_id))
+
+    output = cursor.fetchall()
+    output = json.dumps([dict(row) for row in output])
+    conn.commit()
+    conn.close()
+
+    if output == '[]':
+        output = None
+
+    return output
+
+
 def process_request(uri):
     '''
     Handles the API endpoint.
@@ -215,10 +246,8 @@ def application(environ, start_response):
 
 
 if __name__ == '__main__':
-    new_score = random.randint(1, 100)
-    print(fetch_user(1))
-    print(new_score)
-    update(1, new_score)
-    print(fetch_user(1))
+    print(get_avatar('1'))
+    set_avatar('1', 'newavatar')
+    print(get_avatar('1'))
 
 # vim: tabstop=4 shiftwidth=4 softtabstop=4 expandtab