From ccafc4f462414befc77db0eedbcadc45344f2bca Mon Sep 17 00:00:00 2001
From: xiaohuanshu <xiaohuanshu@gmail.com>
Date: Tue, 20 Sep 2016 13:28:40 +0800
Subject: [PATCH] add avatar for user

---
 checkinsystem/settings_base.py                  |   2 ++
 checkinsystem/urls.py                           |   9 ++++++++-
 media/README.MD                                 |   1 +
 .../img/avatar5.png => media/avatar/default.png | Bin
 templates/header.html                           |   8 ++++----
 user/models.py                                  |  14 ++++++++++++++
 .../commands/updateavatarforalluser.py          |  16 ++++++++++++++++
 7 files changed, 45 insertions(+), 5 deletions(-)
 create mode 100644 media/README.MD
 rename static/img/avatar5.png => media/avatar/default.png (100%)
 create mode 100644 wechat/management/commands/updateavatarforalluser.py

diff --git a/checkinsystem/settings_base.py b/checkinsystem/settings_base.py
index 1d1ae33..bf77639 100644
--- a/checkinsystem/settings_base.py
+++ b/checkinsystem/settings_base.py
@@ -96,6 +96,8 @@
 STATICFILES_DIRS = (
     os.path.join(BASE_DIR, "static"),
 )
+MEDIA_URL = '/media/'
+MEDIA_ROOT = os.path.join(BASE_DIR, "media")
 
 SESSION_ENGINE = 'django.contrib.sessions.backends.cache'
 SESSION_EXPIRE_AT_BROWSER_CLOSE = True
diff --git a/checkinsystem/urls.py b/checkinsystem/urls.py
index 1d7702d..20f9855 100644
--- a/checkinsystem/urls.py
+++ b/checkinsystem/urls.py
@@ -13,7 +13,8 @@
     1. Import the include() function: from django.conf.urls import url, include
     2. Add a URL to urlpatterns:  url(r'^blog/', include('blog.urls'))
 """
-from django.conf.urls import url,include
+from django.conf.urls import url, include, patterns
+from django.conf import settings
 
 urlpatterns = [
     url(r'^$', 'center.views.home', name='home'),
@@ -26,3 +27,9 @@
     url(r'^rbac/', include('rbac.urls', namespace="rbac")),
     url(r'^seat$', 'center.views.seat', name='seat'),
 ]
+
+if settings.DEBUG:
+    urlpatterns += patterns('',
+                            url(r'^media/(?P<path>.*)$', 'django.views.static.serve',
+                                {'document_root': settings.MEDIA_ROOT}),
+                            )
diff --git a/media/README.MD b/media/README.MD
new file mode 100644
index 0000000..c9a33ff
--- /dev/null
+++ b/media/README.MD
@@ -0,0 +1 @@
+upload dir
diff --git a/static/img/avatar5.png b/media/avatar/default.png
similarity index 100%
rename from static/img/avatar5.png
rename to media/avatar/default.png
diff --git a/templates/header.html b/templates/header.html
index cc54a7b..774a9b5 100755
--- a/templates/header.html
+++ b/templates/header.html
@@ -59,18 +59,18 @@ <h4>
                     <!-- Menu Toggle Button -->
                     <a href="#" class="dropdown-toggle" data-toggle="dropdown">
                         <!-- The user image in the navbar-->
-                        <img src="{% static 'img/avatar5.png' %}" class="user-image" alt="User Image">
+                        <img src="{{ request.user.avatar.url }}" class="user-image" alt="User Image">
                         <!-- hidden-xs hides the username on small devices so only the image appears. -->
                         <span class="hidden-xs">{{ request.user.username }}</span>
                     </a>
                     <ul class="dropdown-menu">
                         <!-- The user image in the menu -->
                         <li class="user-header">
-                            <img src="{% static 'img/avatar5.png' %}" class="img-circle" alt="User Image">
+                            <img src="{{ request.user.avatar.url }}" class="img-circle" alt="User Image">
 
                             <p>
-                                1234213421342134123
-                                <small>1234213412412342134</small>
+                                {{ request.user.username }}
+                                <small>{{ request.user.email }}</small>
                             </p>
                         </li>
                         <!-- Menu Body -->
diff --git a/user/models.py b/user/models.py
index 4e2e643..c91490f 100644
--- a/user/models.py
+++ b/user/models.py
@@ -30,6 +30,7 @@ class User(models.Model):
     email = models.CharField(max_length=40, blank=True, null=True)
     verify = models.NullBooleanField()
     role = models.ManyToManyField(Role, through='Usertorole', through_fields=('user', 'role'))
+    avatar = models.ImageField(upload_to='avatar', default='avatar/default.png')
 
     @classmethod_cache
     def isteacher(self):
@@ -43,6 +44,19 @@ def hasresourcejurisdiction(self, jurisdiction):
         from rbac.auth import is_user_has_resourcejurisdiction
         return is_user_has_resourcejurisdiction(self, jurisdiction)
 
+    def updateavatarfromwechat(self):
+        if self.openid is not None:
+            from wechat.api import wechat
+            userinfo = wechat.get_user_info(self.openid, lang='zh_CN')
+            avatar_url = userinfo['headimgurl']
+            from django.core.files import File
+            from django.core.files.temp import NamedTemporaryFile
+            import urllib2
+            img_temp = NamedTemporaryFile(delete=True)
+            img_temp.write(urllib2.urlopen(avatar_url).read())
+            img_temp.flush()
+            self.avatar.save('%s.jpeg' % self.openid, File(img_temp))
+
     def __unicode__(self):
         return u"%s" % (self.username)
 
diff --git a/wechat/management/commands/updateavatarforalluser.py b/wechat/management/commands/updateavatarforalluser.py
new file mode 100644
index 0000000..f27be6c
--- /dev/null
+++ b/wechat/management/commands/updateavatarforalluser.py
@@ -0,0 +1,16 @@
+# -*- coding: utf-8 -*-
+from django.core.management.base import BaseCommand, CommandError
+from user.models import User
+from wechat_sdk.exceptions import OfficialAPIError
+
+class Command(BaseCommand):
+    def handle(self, *args, **options):
+        counter=0
+        users=User.objects.exclude(openid=None).all()
+        for u in users:
+            try:
+                u.updateavatarfromwechat()
+            except OfficialAPIError:
+                pass
+            counter+=1
+        print "update %d users"%counter
\ No newline at end of file