aboutsummaryrefslogtreecommitdiffstats
path: root/97suifangqa/apps/profile
diff options
context:
space:
mode:
Diffstat (limited to '97suifangqa/apps/profile')
-rw-r--r--97suifangqa/apps/profile/__init__.py0
-rw-r--r--97suifangqa/apps/profile/backends.py25
-rw-r--r--97suifangqa/apps/profile/forms.py4
-rw-r--r--97suifangqa/apps/profile/image.py97
-rw-r--r--97suifangqa/apps/profile/models.py144
-rw-r--r--97suifangqa/apps/profile/storage.py21
-rw-r--r--97suifangqa/apps/profile/urls.py9
-rw-r--r--97suifangqa/apps/profile/utils.py19
-rw-r--r--97suifangqa/apps/profile/views.py30
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())
+
Copyright © 2017-2018 Aaron LI. All Rights Reserved.