import logging import os import urllib.parse from bitfield.models import BitField from django.contrib.auth.models import User from django.core.exceptions import SuspiciousOperation from django.db import models from django.db.models import Count import uuid from django_gravatar.helpers import has_gravatar, get_gravatar_url from sorl import thumbnail from core.utils.file import generate_save_file_name from core.utils.url import unique_slugify from dss import settings from spa.models.basemodel import BaseModel logger = logging.getLogger(__name__) def avatar_name(instance, filename): return generate_save_file_name(str(instance.id), 'avatars', filename) class UserProfileManager(models.Manager): def get_query_set(self): return super(UserProfileManager, self).get_query_set().annotate(mix_count=Count('mixes')) class UserUpdateException(Exception): pass class UserProfile(BaseModel): class Meta: app_label = 'spa' objects = UserProfileManager() ACTIVITY_SHARE_NETWORK_FACEBOOK = 1 ACTIVITY_SHARE_NETWORK_TWITTER = 2 user = models.OneToOneField(User, unique=True, related_name='userprofile', on_delete=models.CASCADE) uid = models.UUIDField(primary_key=False, editable=False, null=True) avatar_type = models.CharField(max_length=15, default='social') avatar_image = models.ImageField(max_length=1024, blank=True, upload_to=avatar_name) display_name = models.CharField(blank=True, max_length=35) description = models.CharField(blank=True, max_length=2048) slug = models.SlugField(max_length=50, blank=True, null=True, default=None) activity_sharing_networks = models.IntegerField(default=0) NOTIFICATION_CHOICES = ( ('plays', 'Plays'), ('likes', 'Likes'), ('favourites', 'Favourites'), ('follows', 'Follows'), ('comments', 'Comments'), ) activity_sharing_facebook = BitField(flags=NOTIFICATION_CHOICES, default=0) activity_sharing_twitter = BitField(flags=NOTIFICATION_CHOICES, default=0) email_notifications = BitField(flags=NOTIFICATION_CHOICES, default=0) following = models.ManyToManyField('self', blank=True, symmetrical=False, related_name='followers') # location properties city = models.CharField(max_length=100, blank=True, null=True) country = models.CharField(max_length=100, blank=True, null=True) last_known_session = models.CharField(max_length=250, blank=True, null=True) def __str__(self): return self.__unicode__() def __unicode__(self): return "%s - %s" % (self.user.get_full_name(), self.slug) def save(self, force_insert=False, force_update=False, using=None, update_fields=None): """ Save Photo after ensuring it is not blank. Resize as needed. """ if self.slug is None or self.slug == '': self.slug = unique_slugify(self, self.user.get_username()) print("Slugified: %s" % self.slug) return super(UserProfile, self).save(force_insert, force_update, using, update_fields) def get_roles(self): """ return [ '*', 'admin', 'staff', 'homepage', 'user', 'guest' ] """ roles = ['guest'] if self.user.is_superuser: roles.append('admin') if self.user.is_staff: roles.append('staff') if self.user.is_authenticated: roles.append('user') return roles def get_first_name(self): return self.user.first_name def get_last_name(self): return self.user.last_name def get_username(self): return self.user.username def get_email(self): return self.user.email email = property(get_email) def get_first_name(self): return self.user.first_name first_name = property(get_first_name) def get_last_name(self): return self.user.last_name last_name = property(get_last_name) def __create_slug(self): try: unique_slugify(self, self.get_username() or self.user.get_full_name(), slug_separator='_') self.save() except Exception as e: self.logger.error("Unable to create profile slug: %s", e) def get_session_id(self): return str(self.id) def toggle_favourite(self, mix, value): try: if value: if self.activity.filter(mix=mix).count() == 0: self.activity.model.add(mix=mix, user=self) self.favourites.model.save() else: self.favourites.model.delete(mix=mix) except Exception as ex: self.logger.error("Exception updating favourite: %s" % ex) def is_follower(self, user): try: return user in self.followers.all() except Exception as ex: logger.error(ex) return False def get_absolute_url(self): if self.slug is None or len(self.slug) == 0: self.__create_slug() return "user/%s" % self.slug def get_nice_name(self): return self.display_name or self.first_name + ' ' + self.last_name def get_sized_avatar_image(self, width, height): try: #import ipdb; ipdb.set_trace() image = self.get_avatar_image() logger.debug("get_sized_avatar_image: %s".format(image)) sized = thumbnail.get_thumbnail(image, "%sx%s" % (width, height), crop="center") return urllib.parse.urljoin(settings.MEDIA_URL, sized.name) except SuspiciousOperation: return UserProfile.get_default_avatar_image() except Exception as ex: return UserProfile.get_default_avatar_image() def get_avatar_image(self): avatar_type = self.avatar_type if avatar_type == 'gravatar': gravatar_exists = has_gravatar(self.email) if gravatar_exists: return get_gravatar_url(self.email) else: if os.path.exists(self.avatar_image.file.name): return self.avatar_image else: return self.get_default_avatar_image() return UserProfile.get_default_avatar_image() def get_profile_url(self): return '/user/%s' % (self.slug) def get_profile_description(self): try: return self.description except Exception as ex: pass return settings.DEFAULT_USER_TITLE @classmethod def get_default_avatar_image(cls): return settings.DEFAULT_USER_IMAGE @classmethod def get_default_display_name(cls): return settings.DEFAULT_USER_NAME @classmethod def get_user(cls, user): if user is not None: return user else: username = 'anonymous' u = User(username=username, first_name='Anonymous', last_name='User') u.set_unusable_password() u.username = u.id # comment out the next two lines if you aren't using profiles p = UserProfile(user=u) return p def add_following(self, user): from spa.models.activity import ActivityFollow try: if user is None: return if user.user.is_authenticated: v = ActivityFollow(user=self, to_user=user) v.save() self.following.add(user) self.save() except Exception as ex: self.logger.error("Exception updating like: %s" % ex) raise UserUpdateException(ex) def remove_following(self, user): try: if user is None: return if user.user.is_authenticated: self.following.remove(user) self.save() except Exception as ex: self.logger.error("Exception updating like: %s" % ex) raise UserUpdateException(ex)