move avatar from thirdpart to seahub

avatar/forms.py

@@ -0,0 +1,69 @@
+import os
+
+from django import forms
+from django.forms import widgets
+from django.utils.safestring import mark_safe
+from django.utils.translation import ugettext_lazy as _
+from django.template.defaultfilters import filesizeformat
+
+from avatar.models import Avatar
+from avatar.settings import (AVATAR_MAX_AVATARS_PER_USER, AVATAR_MAX_SIZE,
+                             AVATAR_ALLOWED_FILE_EXTS, AVATAR_DEFAULT_SIZE)
+
+
+def avatar_img(avatar, size):
+    if not avatar.thumbnail_exists(size):
+        avatar.create_thumbnail(size)
+    return mark_safe("""<img src="%s" alt="%s" width="%s" height="%s" />""" % 
+        (avatar.avatar_url(size), unicode(avatar), size, size))
+
+class UploadAvatarForm(forms.Form):
+
+    avatar = forms.ImageField()
+    
+    def __init__(self, *args, **kwargs):
+        self.user = kwargs.pop('user')
+        super(UploadAvatarForm, self).__init__(*args, **kwargs)
+        
+    def clean_avatar(self):
+        data = self.cleaned_data['avatar']
+        if AVATAR_ALLOWED_FILE_EXTS:
+            (root, ext) = os.path.splitext(data.name.lower())
+            if ext not in AVATAR_ALLOWED_FILE_EXTS:
+               raise forms.ValidationError(
+                _(u"%(ext)s is an invalid file extension. Authorized extensions are : %(valid_exts_list)s") % 
+                { 'ext' : ext, 'valid_exts_list' : ", ".join(AVATAR_ALLOWED_FILE_EXTS) }) 
+        if data.size > AVATAR_MAX_SIZE:
+            raise forms.ValidationError(
+                _(u"Your file is too big (%(size)s), the maximum allowed size is %(max_valid_size)s") %
+                { 'size' : filesizeformat(data.size), 'max_valid_size' : filesizeformat(AVATAR_MAX_SIZE)} )
+        count = Avatar.objects.filter(user=self.user).count()
+        if AVATAR_MAX_AVATARS_PER_USER > 1 and \
+           count >= AVATAR_MAX_AVATARS_PER_USER: 
+            raise forms.ValidationError(
+                _(u"You already have %(nb_avatars)d avatars, and the maximum allowed is %(nb_max_avatars)d.") %
+                { 'nb_avatars' : count, 'nb_max_avatars' : AVATAR_MAX_AVATARS_PER_USER})
+        return        
+        
+
+class PrimaryAvatarForm(forms.Form):
+    
+    def __init__(self, *args, **kwargs):
+        user = kwargs.pop('user')
+        size = kwargs.pop('size', AVATAR_DEFAULT_SIZE)
+        avatars = kwargs.pop('avatars')
+        super(PrimaryAvatarForm, self).__init__(*args, **kwargs)
+        self.fields['choice'] = forms.ChoiceField(
+            choices=[(c.id, avatar_img(c, size)) for c in avatars],
+            widget=widgets.RadioSelect)
+
+class DeleteAvatarForm(forms.Form):
+
+    def __init__(self, *args, **kwargs):
+        user = kwargs.pop('user')
+        size = kwargs.pop('size', AVATAR_DEFAULT_SIZE)
+        avatars = kwargs.pop('avatars')
+        super(DeleteAvatarForm, self).__init__(*args, **kwargs)
+        self.fields['choices'] = forms.MultipleChoiceField(
+            choices=[(c.id, avatar_img(c, size)) for c in avatars],
+            widget=widgets.CheckboxSelectMultiple)