--- /dev/null
+# -*- coding: utf-8 -*-
+import datetime
+from south.db import db
+from south.v2 import SchemaMigration
+from django.db import models
+
+
+class Migration(SchemaMigration):
+
+ def forwards(self, orm):
+ # Adding model 'CourseKarmaEvent'
+ db.create_table(u'users_coursekarmaevent', (
+ (u'id', self.gf('django.db.models.fields.AutoField')(primary_key=True)),
+ ('points', self.gf('django.db.models.fields.IntegerField')()),
+ ('user', self.gf('django.db.models.fields.related.ForeignKey')(to=orm['auth.User'])),
+ ('timestamp', self.gf('django.db.models.fields.DateTimeField')(default=datetime.datetime.utcnow)),
+ ('course', self.gf('django.db.models.fields.related.ForeignKey')(to=orm['courses.Course'])),
+ ('event_type', self.gf('django.db.models.fields.CharField')(max_length=15)),
+ ))
+ db.send_create_signal(u'users', ['CourseKarmaEvent'])
+
+ # Adding model 'NoteKarmaEvent'
+ db.create_table(u'users_notekarmaevent', (
+ (u'id', self.gf('django.db.models.fields.AutoField')(primary_key=True)),
+ ('points', self.gf('django.db.models.fields.IntegerField')()),
+ ('user', self.gf('django.db.models.fields.related.ForeignKey')(to=orm['auth.User'])),
+ ('timestamp', self.gf('django.db.models.fields.DateTimeField')(default=datetime.datetime.utcnow)),
+ ('note', self.gf('django.db.models.fields.related.ForeignKey')(to=orm['notes.Note'])),
+ ('event_type', self.gf('django.db.models.fields.CharField')(max_length=15)),
+ ))
+ db.send_create_signal(u'users', ['NoteKarmaEvent'])
+
+ # Adding model 'GenericKarmaEvent'
+ db.create_table(u'users_generickarmaevent', (
+ (u'id', self.gf('django.db.models.fields.AutoField')(primary_key=True)),
+ ('points', self.gf('django.db.models.fields.IntegerField')()),
+ ('user', self.gf('django.db.models.fields.related.ForeignKey')(to=orm['auth.User'])),
+ ('timestamp', self.gf('django.db.models.fields.DateTimeField')(default=datetime.datetime.utcnow)),
+ ('message', self.gf('django.db.models.fields.CharField')(max_length=255)),
+ ))
+ db.send_create_signal(u'users', ['GenericKarmaEvent'])
+
+ # Deleting field 'UserProfile.karma'
+ db.delete_column(u'users_userprofile', 'karma')
+
+
+ def backwards(self, orm):
+ # Deleting model 'CourseKarmaEvent'
+ db.delete_table(u'users_coursekarmaevent')
+
+ # Deleting model 'NoteKarmaEvent'
+ db.delete_table(u'users_notekarmaevent')
+
+ # Deleting model 'GenericKarmaEvent'
+ db.delete_table(u'users_generickarmaevent')
+
+ # Adding field 'UserProfile.karma'
+ db.add_column(u'users_userprofile', 'karma',
+ self.gf('django.db.models.fields.IntegerField')(default=0),
+ keep_default=False)
+
+
+ models = {
+ u'auth.group': {
+ 'Meta': {'object_name': 'Group'},
+ u'id': ('django.db.models.fields.AutoField', [], {'primary_key': 'True'}),
+ 'name': ('django.db.models.fields.CharField', [], {'unique': 'True', 'max_length': '80'}),
+ 'permissions': ('django.db.models.fields.related.ManyToManyField', [], {'to': u"orm['auth.Permission']", 'symmetrical': 'False', 'blank': 'True'})
+ },
+ u'auth.permission': {
+ 'Meta': {'ordering': "(u'content_type__app_label', u'content_type__model', u'codename')", 'unique_together': "((u'content_type', u'codename'),)", 'object_name': 'Permission'},
+ 'codename': ('django.db.models.fields.CharField', [], {'max_length': '100'}),
+ 'content_type': ('django.db.models.fields.related.ForeignKey', [], {'to': u"orm['contenttypes.ContentType']"}),
+ u'id': ('django.db.models.fields.AutoField', [], {'primary_key': 'True'}),
+ 'name': ('django.db.models.fields.CharField', [], {'max_length': '50'})
+ },
+ u'auth.user': {
+ 'Meta': {'object_name': 'User'},
+ 'date_joined': ('django.db.models.fields.DateTimeField', [], {'default': 'datetime.datetime.now'}),
+ 'email': ('django.db.models.fields.EmailField', [], {'max_length': '75', 'blank': 'True'}),
+ 'first_name': ('django.db.models.fields.CharField', [], {'max_length': '30', 'blank': 'True'}),
+ 'groups': ('django.db.models.fields.related.ManyToManyField', [], {'to': u"orm['auth.Group']", 'symmetrical': 'False', 'blank': 'True'}),
+ u'id': ('django.db.models.fields.AutoField', [], {'primary_key': 'True'}),
+ 'is_active': ('django.db.models.fields.BooleanField', [], {'default': 'True'}),
+ 'is_staff': ('django.db.models.fields.BooleanField', [], {'default': 'False'}),
+ 'is_superuser': ('django.db.models.fields.BooleanField', [], {'default': 'False'}),
+ 'last_login': ('django.db.models.fields.DateTimeField', [], {'default': 'datetime.datetime.now'}),
+ 'last_name': ('django.db.models.fields.CharField', [], {'max_length': '30', 'blank': 'True'}),
+ 'password': ('django.db.models.fields.CharField', [], {'max_length': '128'}),
+ 'user_permissions': ('django.db.models.fields.related.ManyToManyField', [], {'to': u"orm['auth.Permission']", 'symmetrical': 'False', 'blank': 'True'}),
+ 'username': ('django.db.models.fields.CharField', [], {'unique': 'True', 'max_length': '30'})
+ },
+ u'contenttypes.contenttype': {
+ 'Meta': {'ordering': "('name',)", 'unique_together': "(('app_label', 'model'),)", 'object_name': 'ContentType', 'db_table': "'django_content_type'"},
+ 'app_label': ('django.db.models.fields.CharField', [], {'max_length': '100'}),
+ u'id': ('django.db.models.fields.AutoField', [], {'primary_key': 'True'}),
+ 'model': ('django.db.models.fields.CharField', [], {'max_length': '100'}),
+ 'name': ('django.db.models.fields.CharField', [], {'max_length': '100'})
+ },
+ u'courses.course': {
+ 'Meta': {'ordering': "['-file_count', 'school', 'name']", 'unique_together': "(('name', 'department'),)", 'object_name': 'Course'},
+ 'created_at': ('django.db.models.fields.DateTimeField', [], {'auto_now_add': 'True', 'blank': 'True'}),
+ 'department': ('django.db.models.fields.related.ForeignKey', [], {'to': u"orm['courses.Department']", 'null': 'True', 'blank': 'True'}),
+ 'desc': ('django.db.models.fields.TextField', [], {'max_length': '511', 'null': 'True', 'blank': 'True'}),
+ 'file_count': ('django.db.models.fields.IntegerField', [], {'default': '0'}),
+ 'flags': ('django.db.models.fields.IntegerField', [], {'default': '0'}),
+ u'id': ('django.db.models.fields.AutoField', [], {'primary_key': 'True'}),
+ 'instructor_email': ('django.db.models.fields.EmailField', [], {'max_length': '75', 'null': 'True', 'blank': 'True'}),
+ 'instructor_name': ('django.db.models.fields.CharField', [], {'max_length': '255', 'null': 'True', 'blank': 'True'}),
+ 'name': ('django.db.models.fields.CharField', [], {'max_length': '255'}),
+ 'school': ('django.db.models.fields.related.ForeignKey', [], {'to': u"orm['courses.School']", 'null': 'True', 'blank': 'True'}),
+ 'slug': ('django.db.models.fields.SlugField', [], {'max_length': '150', 'null': 'True'}),
+ 'updated_at': ('django.db.models.fields.DateTimeField', [], {'default': 'datetime.datetime.utcnow'}),
+ 'url': ('django.db.models.fields.URLField', [], {'max_length': '511', 'null': 'True', 'blank': 'True'})
+ },
+ u'courses.department': {
+ 'Meta': {'unique_together': "(('name', 'school'),)", 'object_name': 'Department'},
+ u'id': ('django.db.models.fields.AutoField', [], {'primary_key': 'True'}),
+ 'name': ('django.db.models.fields.CharField', [], {'max_length': '255'}),
+ 'school': ('django.db.models.fields.related.ForeignKey', [], {'to': u"orm['courses.School']"}),
+ 'slug': ('django.db.models.fields.SlugField', [], {'max_length': '150', 'null': 'True'}),
+ 'url': ('django.db.models.fields.URLField', [], {'max_length': '511', 'null': 'True', 'blank': 'True'})
+ },
+ u'courses.school': {
+ 'Meta': {'ordering': "['-file_count', '-priority', 'name']", 'object_name': 'School'},
+ 'alias': ('django.db.models.fields.CharField', [], {'max_length': '255', 'null': 'True', 'blank': 'True'}),
+ 'facebook_id': ('django.db.models.fields.BigIntegerField', [], {'null': 'True', 'blank': 'True'}),
+ 'file_count': ('django.db.models.fields.IntegerField', [], {'default': '0'}),
+ 'hashtag': ('django.db.models.fields.CharField', [], {'max_length': '16', 'unique': 'True', 'null': 'True', 'blank': 'True'}),
+ u'id': ('django.db.models.fields.AutoField', [], {'primary_key': 'True'}),
+ 'location': ('django.db.models.fields.CharField', [], {'max_length': '255', 'null': 'True', 'blank': 'True'}),
+ 'name': ('django.db.models.fields.CharField', [], {'max_length': '255'}),
+ 'priority': ('django.db.models.fields.BooleanField', [], {'default': 'False'}),
+ 'slug': ('django.db.models.fields.SlugField', [], {'max_length': '150', 'null': 'True'}),
+ 'url': ('django.db.models.fields.URLField', [], {'max_length': '511', 'blank': 'True'}),
+ 'usde_id': ('django.db.models.fields.BigIntegerField', [], {'unique': 'True', 'null': 'True', 'blank': 'True'})
+ },
+ u'licenses.license': {
+ 'Meta': {'object_name': 'License'},
+ 'html': ('django.db.models.fields.TextField', [], {}),
+ u'id': ('django.db.models.fields.AutoField', [], {'primary_key': 'True'}),
+ 'name': ('django.db.models.fields.CharField', [], {'unique': 'True', 'max_length': '80'})
+ },
+ u'notes.note': {
+ 'Meta': {'ordering': "['-uploaded_at']", 'unique_together': "(('fp_file', 'upstream_link'),)", 'object_name': 'Note'},
+ 'course': ('django.db.models.fields.related.ForeignKey', [], {'to': u"orm['courses.Course']"}),
+ 'file_type': ('django.db.models.fields.CharField', [], {'default': "'???'", 'max_length': '15', 'null': 'True', 'blank': 'True'}),
+ 'flags': ('django.db.models.fields.IntegerField', [], {'default': '0'}),
+ 'fp_file': ('django_filepicker.models.FPFileField', [], {'max_length': '100', 'null': 'True', 'blank': 'True'}),
+ 'gdrive_url': ('django.db.models.fields.URLField', [], {'max_length': '1024', 'unique': 'True', 'null': 'True', 'blank': 'True'}),
+ 'html': ('django.db.models.fields.TextField', [], {'null': 'True', 'blank': 'True'}),
+ u'id': ('django.db.models.fields.AutoField', [], {'primary_key': 'True'}),
+ 'ip': ('django.db.models.fields.GenericIPAddressField', [], {'max_length': '39', 'null': 'True', 'blank': 'True'}),
+ 'is_hidden': ('django.db.models.fields.BooleanField', [], {'default': 'False'}),
+ 'license': ('django.db.models.fields.related.ForeignKey', [], {'to': u"orm['licenses.License']", 'null': 'True', 'blank': 'True'}),
+ 'mimetype': ('django.db.models.fields.CharField', [], {'max_length': '255', 'null': 'True', 'blank': 'True'}),
+ 'name': ('django.db.models.fields.CharField', [], {'max_length': '255', 'null': 'True', 'blank': 'True'}),
+ 'pdf_file': ('django.db.models.fields.files.FileField', [], {'max_length': '100', 'null': 'True', 'blank': 'True'}),
+ 'slug': ('django.db.models.fields.SlugField', [], {'unique': 'True', 'max_length': '255'}),
+ 'static_html': ('django.db.models.fields.BooleanField', [], {'default': 'False'}),
+ 'text': ('django.db.models.fields.TextField', [], {'null': 'True', 'blank': 'True'}),
+ 'thanks': ('django.db.models.fields.PositiveIntegerField', [], {'default': '0'}),
+ 'tweeted': ('django.db.models.fields.BooleanField', [], {'default': 'False'}),
+ 'uploaded_at': ('django.db.models.fields.DateTimeField', [], {'default': 'datetime.datetime.utcnow', 'null': 'True'}),
+ 'upstream_link': ('django.db.models.fields.URLField', [], {'max_length': '1024', 'unique': 'True', 'null': 'True', 'blank': 'True'}),
+ 'user': ('django.db.models.fields.related.ForeignKey', [], {'to': u"orm['auth.User']", 'null': 'True', 'on_delete': 'models.SET_NULL', 'blank': 'True'}),
+ 'year': ('django.db.models.fields.IntegerField', [], {'default': '2014', 'null': 'True', 'blank': 'True'})
+ },
+ u'taggit.tag': {
+ 'Meta': {'ordering': "['namespace', 'name']", 'object_name': 'Tag'},
+ u'id': ('django.db.models.fields.AutoField', [], {'primary_key': 'True'}),
+ 'name': ('django.db.models.fields.CharField', [], {'unique': 'True', 'max_length': '100'}),
+ 'namespace': ('django.db.models.fields.CharField', [], {'max_length': '100', 'null': 'True', 'blank': 'True'}),
+ 'slug': ('django.db.models.fields.SlugField', [], {'unique': 'True', 'max_length': '100'})
+ },
+ u'taggit.taggeditem': {
+ 'Meta': {'object_name': 'TaggedItem'},
+ 'content_type': ('django.db.models.fields.related.ForeignKey', [], {'related_name': "u'taggit_taggeditem_tagged_items'", 'to': u"orm['contenttypes.ContentType']"}),
+ u'id': ('django.db.models.fields.AutoField', [], {'primary_key': 'True'}),
+ 'object_id': ('django.db.models.fields.IntegerField', [], {'db_index': 'True'}),
+ 'tag': ('django.db.models.fields.related.ForeignKey', [], {'related_name': "u'taggit_taggeditem_items'", 'to': u"orm['taggit.Tag']"})
+ },
+ u'users.coursekarmaevent': {
+ 'Meta': {'object_name': 'CourseKarmaEvent'},
+ 'course': ('django.db.models.fields.related.ForeignKey', [], {'to': u"orm['courses.Course']"}),
+ 'event_type': ('django.db.models.fields.CharField', [], {'max_length': '15'}),
+ u'id': ('django.db.models.fields.AutoField', [], {'primary_key': 'True'}),
+ 'points': ('django.db.models.fields.IntegerField', [], {}),
+ 'timestamp': ('django.db.models.fields.DateTimeField', [], {'default': 'datetime.datetime.utcnow'}),
+ 'user': ('django.db.models.fields.related.ForeignKey', [], {'to': u"orm['auth.User']"})
+ },
+ u'users.generickarmaevent': {
+ 'Meta': {'object_name': 'GenericKarmaEvent'},
+ u'id': ('django.db.models.fields.AutoField', [], {'primary_key': 'True'}),
+ 'message': ('django.db.models.fields.CharField', [], {'max_length': '255'}),
+ 'points': ('django.db.models.fields.IntegerField', [], {}),
+ 'timestamp': ('django.db.models.fields.DateTimeField', [], {'default': 'datetime.datetime.utcnow'}),
+ 'user': ('django.db.models.fields.related.ForeignKey', [], {'to': u"orm['auth.User']"})
+ },
+ u'users.notekarmaevent': {
+ 'Meta': {'object_name': 'NoteKarmaEvent'},
+ 'event_type': ('django.db.models.fields.CharField', [], {'max_length': '15'}),
+ u'id': ('django.db.models.fields.AutoField', [], {'primary_key': 'True'}),
+ 'note': ('django.db.models.fields.related.ForeignKey', [], {'to': u"orm['notes.Note']"}),
+ 'points': ('django.db.models.fields.IntegerField', [], {}),
+ 'timestamp': ('django.db.models.fields.DateTimeField', [], {'default': 'datetime.datetime.utcnow'}),
+ 'user': ('django.db.models.fields.related.ForeignKey', [], {'to': u"orm['auth.User']"})
+ },
+ u'users.userprofile': {
+ 'Meta': {'object_name': 'UserProfile'},
+ u'id': ('django.db.models.fields.AutoField', [], {'primary_key': 'True'}),
+ 'school': ('django.db.models.fields.related.ForeignKey', [], {'to': u"orm['courses.School']", 'null': 'True', 'blank': 'True'}),
+ 'user': ('django.db.models.fields.related.OneToOneField', [], {'to': u"orm['auth.User']", 'unique': 'True'})
+ }
+ }
+
+ complete_apps = ['users']
\ No newline at end of file
#!/usr/bin/env python
# -*- coding:utf8 -*-
# Copyright (C) 2013 FinalsClub Foundation
-import random
import logging
-from allauth.account.signals import user_logged_in, user_signed_up, email_confirmed, email_changed, email_added
-from allauth.socialaccount.signals import pre_social_login
+import datetime
from django.contrib.auth.models import User
-from django.core.exceptions import ObjectDoesNotExist
-from django.db.models.signals import post_save, pre_save
+from django.db.models import Sum
+from django.db.models.signals import post_save
from django.dispatch import receiver
from django.db import models, DatabaseError
from django.middleware.transaction import transaction
logger = logging.getLogger(__name__)
+
class UserProfile(models.Model):
user = models.OneToOneField(User)
school = models.ForeignKey(School, blank=True, null=True)
- karma = models.IntegerField(default=0)
+ def get_points(self):
+ sum = 0
+ for cls in ALL_KARMA_EVENT_CLASSES:
+ points = cls.objects.filter(user=self.user).aggregate(Sum('points'))['points__sum']
+ if points:
+ sum += points
+
+ return sum
+
+ NO_BADGE = 0
+ PROSPECT = 1
+ BEGINNER = 2
+ TRAINEE = 3
+ APPRENTICE = 4
+ SCHOLAR = 5
+
+ BADGES = (
+ PROSPECT,
+ BEGINNER,
+ TRAINEE,
+ APPRENTICE,
+ SCHOLAR
+ )
+
+ BADGE_NAMES = {
+ PROSPECT: 'Prospect',
+ BEGINNER: 'Beginner',
+ TRAINEE: 'Trainee',
+ APPRENTICE: 'Apprentice',
+ SCHOLAR: 'Scholar'
+ }
+
+ BADGE_THRESHOLDS = {
+ PROSPECT: 10,
+ BEGINNER: 100,
+ TRAINEE: 200,
+ APPRENTICE: 500,
+ SCHOLAR: 1000
+ }
+
+ def get_badge(self):
+ points = self.get_points()
+ highest_badge = self.NO_BADGE
+ for badge in self.BADGES:
+ if points >= self.BADGE_THRESHOLDS[badge]:
+ highest_badge = badge
+ return highest_badge
def __unicode__(self):
return self.user.__unicode__()
+
+class BaseKarmaEvent(models.Model):
+ points = models.IntegerField()
+ user = models.ForeignKey(User)
+ timestamp = models.DateTimeField(default=datetime.datetime.utcnow)
+
+ class Meta:
+ abstract = True
+
+ def get_message(self):
+ raise NotImplemented()
+
+
+class GenericKarmaEvent(BaseKarmaEvent):
+ message = models.CharField(max_length=255)
+
+ @staticmethod
+ def create_event(user, message, points):
+ event = GenericKarmaEvent.objects.create(user=user,
+ points=points,
+ message=message)
+ event.save()
+
+ def get_message(self):
+ return self.message
+
+
+class NoteKarmaEvent(BaseKarmaEvent):
+ UPLOAD = 'upload'
+ THANKS = 'thanks'
+ NOTE_DELETED = 'deleted'
+ GIVE_FLAG = 'give_flag'
+ GET_FLAGGED = 'get_flagged'
+ EVENT_TYPE_CHOICES = (
+ (UPLOAD, "You uploaded a note"),
+ (THANKS, "You received a thanks for your note"),
+ (NOTE_DELETED, "Your note was deleted"),
+ (GIVE_FLAG, "You flagged a note"),
+ (GET_FLAGGED, "Your note was flagged as spam"),
+ )
+ note = models.ForeignKey('notes.Note')
+ event_type = models.CharField(max_length=15, choices=EVENT_TYPE_CHOICES)
+
+ POINTS = {
+ UPLOAD: 5,
+ THANKS: 1,
+ NOTE_DELETED: -5,
+ GIVE_FLAG: -1,
+ GET_FLAGGED: -100
+ }
+
+ def get_message(self):
+ return self.get_event_type_display()
+
+ def __unicode__(self):
+ return unicode(self.user) + ' -- ' + self.get_event_type_display() + ' -- ' + unicode(self.note)
+
+ @staticmethod
+ def create_event(user, note, type):
+ event = NoteKarmaEvent.objects.create(user=user,
+ note=note,
+ points=NoteKarmaEvent.POINTS[type],
+ event_type=type)
+ event.save()
+
+
+class CourseKarmaEvent(BaseKarmaEvent):
+ GIVE_FLAG = 'give_flag'
+ GET_FLAGGED = 'get_flagged'
+ EVENT_TYPE_CHOICES = (
+ (GIVE_FLAG, "You flagged a course"),
+ (GET_FLAGGED, "Your course was flagged as spam"),
+ )
+ course = models.ForeignKey('courses.Course')
+ event_type = models.CharField(max_length=15, choices=EVENT_TYPE_CHOICES)
+
+ POINTS = {
+ GIVE_FLAG: -1,
+ GET_FLAGGED: -100
+ }
+
+ def get_message(self):
+ return self.get_event_type_display()
+
+ def __unicode__(self):
+ return unicode(self.user) + ' -- ' + self.get_event_type_display() + ' -- ' + unicode(self.course)
+
+ @staticmethod
+ def create_event(user, course, type):
+ event = CourseKarmaEvent.objects.create(user=user,
+ course=course,
+ points=CourseKarmaEvent.POINTS[type],
+ event_type=type)
+ event.save()
+
+
+ALL_KARMA_EVENT_CLASSES = (GenericKarmaEvent, NoteKarmaEvent, CourseKarmaEvent)
+
+
def user_display_name(user):
"""Return the best way to display a user's
name to them on the site."""
else:
return user.username
+
@receiver(post_save, sender=User, weak=True)
def create_user_profile(sender, instance, created, **kwargs):
if created: