From 804a667b19877450447de8e9185f3d22b50016a7 Mon Sep 17 00:00:00 2001
From: James Royal <jhr.atx@gmail.com>
Date: Fri, 14 Aug 2015 23:52:28 -0500
Subject: [PATCH] New profile URLs are saved to the database

---
 data/interfaces/default/edit_user.html |  2 ++
 data/interfaces/default/user.html      |  2 +-
 plexpy/datafactory.py                  | 26 ++++++++++++++++++++++++++
 plexpy/webserve.py                     |  4 ++++
 4 files changed, 33 insertions(+), 1 deletion(-)

diff --git a/data/interfaces/default/edit_user.html b/data/interfaces/default/edit_user.html
index a7eff993..717c67a0 100644
--- a/data/interfaces/default/edit_user.html
+++ b/data/interfaces/default/edit_user.html
@@ -93,6 +93,7 @@ DOCUMENTATION :: END
                     if ($.trim(friendly_name) !== '') {
                         $(".set-username").html(friendly_name);
                     }
+                    $("#user-profile-thumb").attr('src', thumb);
                 }
             });
         % else:
@@ -106,6 +107,7 @@ DOCUMENTATION :: END
                     if ($.trim(friendly_name) !== '') {
                         $(".set-username").html(friendly_name);
                     }
+                    $("#user-profile-thumb").attr('src', thumb);
                 }
             });
         % endif
diff --git a/data/interfaces/default/user.html b/data/interfaces/default/user.html
index e7af99e5..ba48c9ec 100644
--- a/data/interfaces/default/user.html
+++ b/data/interfaces/default/user.html
@@ -41,7 +41,7 @@ from plexpy import helpers
             <div class="table-card-back">
                 <div class="user-info-wrapper">
                     <div class="user-info-poster-face" id="user-gravatar">
-                        <img src="${data['thumb']}" height="80px" width="80px">
+                        <img id="user-profile-thumb" src="${data['thumb']}" height="80px" width="80px">
                     </div>
                     <div class="user-info-username">
                         <span class="set-username">${data['friendly_name']}</span> <span id="edit-user-tooltip" data-target="tooltip" title="Edit user details"><a href="#" data-toggle="modal" data-target="#edit-user-modal" id="toggle-edit-user-modal"><i class="fa fa-pencil"></i></a></span>
diff --git a/plexpy/datafactory.py b/plexpy/datafactory.py
index cda9bc6b..b525e2eb 100644
--- a/plexpy/datafactory.py
+++ b/plexpy/datafactory.py
@@ -248,6 +248,32 @@ class DataFactory(object):
             except Exception, e:
                 logger.debug(u"Uncaught exception %s" % e)
 
+    def set_user_profile_url(self, user=None, user_id=None, profile_url=None):
+        if user_id:
+            if profile_url.strip() == '':
+                profile_url = None
+
+            monitor_db = database.MonitorDatabase()
+
+            control_value_dict = {"user_id": user_id}
+            new_value_dict = {"thumb": profile_url}
+            try:
+                monitor_db.upsert('users', new_value_dict, control_value_dict)
+            except Exception, e:
+                logger.debug(u"Uncaught exception %s" % e)
+        if user:
+            if profile_url.strip() == '':
+                profile_url = None
+
+            monitor_db = database.MonitorDatabase()
+
+            control_value_dict = {"user_id": user_id}
+            new_value_dict = {"thumb": profile_url}
+            try:
+                monitor_db.upsert('users', new_value_dict, control_value_dict)
+            except Exception, e:
+                logger.debug(u"Uncaught exception %s" % e)
+
     def get_user_friendly_name(self, user=None, user_id=None):
         if user_id:
             monitor_db = database.MonitorDatabase()
diff --git a/plexpy/webserve.py b/plexpy/webserve.py
index 3657cec7..0df677b8 100644
--- a/plexpy/webserve.py
+++ b/plexpy/webserve.py
@@ -193,6 +193,8 @@ class WebInterface(object):
                                                     friendly_name=friendly_name,
                                                     do_notify=do_notify,
                                                     keep_history=keep_history)
+                data_factory.set_user_profile_url(user_id=user_id,
+                                                  profile_url=kwargs['thumb'])
 
                 status_message = "Successfully updated user."
                 return status_message
@@ -206,6 +208,8 @@ class WebInterface(object):
                                                     friendly_name=friendly_name,
                                                     do_notify=do_notify,
                                                     keep_history=keep_history)
+                data_factory.set_user_profile_url(user=user,
+                                                  profile_url=kwargs['thumb'])
 
                 status_message = "Successfully updated user."
                 return status_message