cloned from 'bitbucket', 2013/08/13

9 files changed, 349 insertions, 0 deletions

diff --git a/97suifangqa/apps/profile/__init__.py b/97suifangqa/apps/profile/__init__.py
new file mode 100644
index 0000000..e69de29
--- /dev/null
+++ b/97suifangqa/apps/profile/__init__.py
diff --git a/97suifangqa/apps/profile/backends.py b/97suifangqa/apps/profile/backends.py
new file mode 100644
index 0000000..8903805
--- /dev/null
+++ b/97suifangqa/apps/profile/backends.py
@@ -0,0 +1,25 @@
+#-*- coding: utf-8 -*-
+
+from django.contrib.auth.models import User
+
+class EmailOrUsernameModelBackend(object):
+
+	def authenticate(self, username=None, password=None):
+		if '@' in username:
+			kwargs = {'email': username}
+		else:
+			kwargs = {'username': username}
+		try:
+			user = User.objects.get(**kwargs)
+			if user.check_password(password):
+				return user
+		except User.DoesNotExist:
+			pass
+		return None
+
+	def get_user(self, user_id):
+		try:
+			return User.objects.get(pk=user_id)
+		except User.DoesNotExist:
+			pass
+		return None
\ No newline at end of file
diff --git a/97suifangqa/apps/profile/forms.py b/97suifangqa/apps/profile/forms.py
new file mode 100644
index 0000000..1d3aed5
--- /dev/null
+++ b/97suifangqa/apps/profile/forms.py
@@ -0,0 +1,4 @@
+# -*- coding: utf-8 -*-
+
+from django import forms
+from django.contrib.auth.forms import UserCreationForm
diff --git a/97suifangqa/apps/profile/image.py b/97suifangqa/apps/profile/image.py
new file mode 100644
index 0000000..b5dd550
--- /dev/null
+++ b/97suifangqa/apps/profile/image.py
@@ -0,0 +1,97 @@
+# -*- coding: utf-8 -*-
+"""
+	图片处理函数
+"""
+import os
+import time
+
+from hashlib import md5
+from django.conf import settings
+from os import mkdir, path as _path
+
+try:
+	from PIL import Image, ImageOps, ImageDraw
+except:
+	import Image, ImageOps, ImageDraw
+
+
+def rm_avatar_thumb(object_id, avatar_link):
+	name, ext = _path.splitext(avatar_link)
+	try:
+		fn = generat_location(object_id, ext, 200)
+		os.remove(fn)
+	except:
+		pass
+	
+
+def _mkdir(dirname):
+	"""
+		遍历创建文件夹
+	"""
+	index = []
+	initial = 0
+	for i in range(dirname.count('/')):
+		initial = dirname.find('/', initial, len(dirname))
+		index.insert(len(index), initial)
+		initial += 1
+	for j in index[1:]:
+		_dir = dirname[:j]
+		if _dir and not _path.exists(_dir):
+			mkdir(_dir)
+
+
+def generat_dir_and_fn(uid, flag):
+	root = settings.MEDIA_ROOT
+	fn = md5('%s' % uid).hexdigest()
+	d1 = fn[-2:]
+	d2 = fn[-4:-2]
+	dirname = "%s%s/%s/%s/" % (root, flag, d1, d2)
+	return dirname, fn
+
+
+def generat_location(uid, ext, size, flag):
+	"""
+		获取文件路径
+	"""
+	fmt = "%s%s_%s%s"
+	dirname, fn = generat_dir_and_fn(uid, flag)
+	_mkdir(dirname)
+	if not ext or \
+		ext == '.':
+		ext = '.jpg'
+	return fmt % (dirname, fn, size, ext)
+
+
+def crop(image, object_id, size=100, mask_img=None, flag="cache", **kwargs):
+	"""
+		截取图片
+	"""
+	_size = (size, size)
+	path = image.path
+	root, filename = _path.split(path)
+	name, ext = _path.splitext(filename)
+
+	original = Image.open(open(path, 'rb'))
+
+	# 保存为圆形缩略图
+	"""
+	mask = Image.new('L', _size, 0)
+	draw = ImageDraw.Draw(mask) 
+	draw.ellipse((0, 0) + _size, fill=255)
+	thumb = ImageOps.fit(original, mask.size, centering=(0.5, 0.5))
+	thumb.putalpha(mask)
+	"""
+	try:
+		if mask_img:
+			mask = Image.open(mask_img).convert('L')
+			thumb = ImageOps.fit(original, _size, centering=(0.5, 0.5))
+			thumb.putalpha(mask)
+		else: thumb = ImageOps.fit(original, _size, Image.ANTIALIAS)
+		
+		thumb_path = generat_location(object_id, ext, size, flag)
+		if not _path.exists(thumb_path):
+			thumb.save(thumb_path)
+	except:
+		thumb_path = path
+		
+	return thumb_path.replace(settings.MEDIA_ROOT, settings.MEDIA_URL)
\ No newline at end of file
diff --git a/97suifangqa/apps/profile/models.py b/97suifangqa/apps/profile/models.py
new file mode 100644
index 0000000..2168f39
--- /dev/null
+++ b/97suifangqa/apps/profile/models.py
@@ -0,0 +1,144 @@
+# -*- coding: utf-8 -*-
+
+from django.db import models
+from django.contrib import admin
+from django.contrib.auth.models import User
+from django.contrib.contenttypes.models import ContentType
+from django.contrib.contenttypes import generic
+
+from .storage import OverwriteStorage
+from .utils import avatar_by_user
+from .image import crop
+
+
+class Profile(models.Model):
+
+    gender_choices = (
+        (0, u"男"),
+        (1, u"女")
+    )
+
+    education_choices = (
+        (0, u"高中"),
+        (1, u"大专"),
+        (2, u"本科"),
+        (3, u"硕士"),
+        (4, u"博士"))
+
+    user      = models.OneToOneField(User, null=True, blank=True)
+    name      = models.CharField(u"用户名", max_length=20, null=True, blank=True)
+    avatar    = models.ImageField(u"头像", upload_to="uploads/avatar/", storage=OverwriteStorage())
+    education = models.IntegerField(u"学历", choices=education_choices)
+    email     = models.EmailField(u"邮箱", primary_key=True)
+    gender    = models.IntegerField(u"性别", choices=gender_choices, default=0)
+    user_level= models.IntegerField(u"等级", default=0)
+    medicines = models.ManyToManyField("medicine.Medicine", related_name="users", verbose_name= u"药物", null=True, blank=True)
+    hospital  = models.ForeignKey("location.Hospital", related_name="patients", verbose_name= u"医院", null=True, blank=True)
+    
+    class Meta:
+        verbose_name_plural = u"用户信息"
+        
+    def __unicode__(self):
+        return "< Profile : %s >" % self.id
+
+    def save(self, exist=False, *args, **kwargs):
+        # 保存
+
+        if self.id:
+            exist = True
+        if not exist and not self.avatar:
+            # 自动生成头像
+            self.avatar = avatar_by_user(self.user)
+        super(Profile, self).save(*args, **kwargs)
+
+    def thumbnail(self, size=100):
+		# 头像截图
+        return crop(self.avatar, self.user.id, size, flag="user")
+
+class UserHospitalItem(models.Model):
+    
+    created_at = models.DateTimeField(u"创建时间", auto_now_add=True)
+    user       = models.ForeignKey(User, verbose_name=u"用户", null=True, blank=True)
+    hospital   = models.ManyToManyField("location.Hospital", related_name='userhospitalitems', verbose_name=u"就诊医院", null=True, blank=True)
+
+    class Meta:
+        verbose_name_plural=u"就诊医院记录"
+
+    def __unicode__(self):
+        return "< UserHospitalItem: %s>" % self.id
+
+
+class UserReadingLog(models.Model):
+
+    created_at = models.DateTimeField(u"创建时间", auto_now_add=True)
+    user       = models.ForeignKey(User, verbose_name=u"用户", null=True, blank=True)
+    #sciblog   = models.ForeignKey('blog.SciBlog', related_name="readinglogs", verbose_name=u"文章")
+    sciblog    = models.TextField(u"文章")              #TODO:修改成真正的sciblog外键
+    readevent  = models.ManyToManyField('ReadEvent', related_name="readinglogs", verbose_name=u"阅读事件", null=True, blank=True)
+
+    class Meta:
+        verbose_name_plural=u"用户医学文章阅读记录"
+
+    def __unicode__(self):
+        return "< UserReadingLog: %s >" % self.id
+    
+
+class ReadEvent(models.Model):
+
+    type_choices = (
+        (0, "isUnd"),
+        (1, "fav"),
+        (2, "ask"),
+        (3, "note"),)
+        
+    type           = models.IntegerField(u"类型", choices=type_choices, default=0)
+
+    content_type   = models.ForeignKey(ContentType, null=True, blank=True)                   # 将ReadEvent作为GenericForeignKey
+    object_id      = models.PositiveIntegerField(null=True, blank=True)
+    content_object = generic.GenericForeignKey("content_type", "object_id")
+
+    class Meta:
+        verbose_name_plural=u"文章阅读事件"
+
+    def __unicode__(self):
+        return "< ReadEvent: %s >" % self.id
+
+
+class Noting(models.Model):
+
+    content        = models.TextField(u"内容")                               # TODO: 完善代码，确保能输入中文
+    user           = models.ForeignKey(User, verbose_name=u"用户", null=True, blank=True)
+
+    content_type   = models.ForeignKey(ContentType, null=True, blank=True)                         # 将ReadEvent作为GenericForeignKey
+    object_id      = models.PositiveIntegerField(null=True, blank=True)
+    content_object = generic.GenericForeignKey("content_type", "object_id")
+    
+    class Meta:
+        verbose_name_plural=u"记笔记"
+
+    def __unicode__(self):
+        return "< Noting: %s >" % self.id
+
+
+class UserLevelLog(models.Model):
+
+    value      = models.FloatField(u"获得经验值")
+    created_at = models.DateTimeField(u"创建时间", auto_now_add=True)
+    user       = models.ForeignKey(User, verbose_name=u"用户", null=True, blank=True)
+
+    class Meta:
+        verbose_name_plural=u"用户等级记录"
+
+    def __unicode__(self):
+        return "< UserLevelLog: %s >" % self.id
+
+
+
+admin.site.register([
+                     Profile,
+                     UserHospitalItem,
+                     UserReadingLog,
+                     ReadEvent,
+                     Noting,
+                     UserLevelLog,
+                    ])
diff --git a/97suifangqa/apps/profile/storage.py b/97suifangqa/apps/profile/storage.py
new file mode 100644
index 0000000..f5069fc
--- /dev/null
+++ b/97suifangqa/apps/profile/storage.py
@@ -0,0 +1,21 @@
+#-*- coding: utf-8 -*-
+"""
+	文件存储
+"""
+import os
+import time 
+import random
+
+from django.core.files.storage import FileSystemStorage
+
+class OverwriteStorage(FileSystemStorage):
+    
+    def _save(self, name, content):
+        if self.exists(name):
+            self.delete(name)
+        ext = os.path.splitext(name)[1]
+        d = os.path.dirname(name)
+        fn = time.strftime("%Y%m%d%H%M%S")
+        fn = fn + "_%d" % random.randint(0,1000)
+        name = os.path.join(d, fn + ext)
+        return super(OverwriteStorage, self)._save(name, content)
\ No newline at end of file
diff --git a/97suifangqa/apps/profile/urls.py b/97suifangqa/apps/profile/urls.py
new file mode 100644
index 0000000..cbc453d
--- /dev/null
+++ b/97suifangqa/apps/profile/urls.py
@@ -0,0 +1,9 @@
+from django.conf.urls import patterns, url
+
+from .views import  *
+
+urlpatterns = patterns('',
+        url(r'^login/?$', login, name = "login"),
+        url(r'^logout/?$', logout, name = "logout"),
+        url(r'^signup/?$', signup, name = "signup"),
+                      )
diff --git a/97suifangqa/apps/profile/utils.py b/97suifangqa/apps/profile/utils.py
new file mode 100644
index 0000000..27067d5
--- /dev/null
+++ b/97suifangqa/apps/profile/utils.py
@@ -0,0 +1,19 @@
+#-*- coding: utf-8 -*-
+
+import random
+
+from django.conf import settings
+
+
+def avatar_by_user(user):
+	"""
+		生成头像
+	"""
+	from django.core.files.images import ImageFile
+
+	_avatar_path = "%s/avatars/%s.png" % (settings.MEDIA_ROOT, random.randint(1,22))
+
+	_avatar = ImageFile(open(_avatar_path))
+	user.profile.avatar = _avatar
+	user.profile.save()
+	return user.profile
\ No newline at end of file
diff --git a/97suifangqa/apps/profile/views.py b/97suifangqa/apps/profile/views.py
new file mode 100644
index 0000000..c41e62b
--- /dev/null
+++ b/97suifangqa/apps/profile/views.py
@@ -0,0 +1,30 @@
+# -*- coding: utf-8 -*-
+
+from django.http import HttpResponse, HttpResponseRedirect
+from django.conf import settings 
+from django.shortcuts import render
+from django.contrib.auth.views import login, logout
+from django.contrib.auth import login as auth_login
+
+from .forms import UserCreationForm
+
+
+
+def signup(request):
+    u'''
+    用户注册
+    '''
+    if request.user.is_authenticated():
+        return HttpResponseRedirect(settings.LOGIN_REDIRECT_URL)
+
+    if request.method == 'POST':
+        form = UserCreationForm(request.POST)
+        if form.is_valid():
+            user = form.save()
+            return HttpResponseRedirect(request.REQUEST.get('next'))
+    else:
+        form = UserCreationForm()
+
+    return render(request, 'registration/signup.html',
+                  locals())
+