144 lines
3.8 KiB
Python
144 lines
3.8 KiB
Python
# -*- coding: utf-8 -*-
|
|
from __future__ import unicode_literals, absolute_import
|
|
|
|
from django.conf import settings
|
|
from django.contrib.auth.models import AbstractUser
|
|
from django.urls import reverse
|
|
from django.db import models
|
|
from django.utils.translation import ugettext_lazy as _
|
|
from stdimage.models import StdImageField
|
|
from django.db.models.signals import post_save
|
|
from mhackspace.base.tasks import matrix_message
|
|
|
|
|
|
class User(AbstractUser):
|
|
name = models.CharField(_("Name of User"), blank=True, max_length=255)
|
|
public = models.BooleanField(
|
|
default=False, help_text="If the users email is public on post feeds"
|
|
)
|
|
_image = StdImageField(
|
|
upload_to="avatars/",
|
|
blank=True,
|
|
null=True,
|
|
db_column="image",
|
|
variations={"profile": {"width": 256, "height": 256, "crop": True}},
|
|
)
|
|
|
|
# https://github.com/pennersr/django-allauth/issues/520
|
|
@property
|
|
def image(self):
|
|
return self._image
|
|
|
|
@image.setter
|
|
def image(self, value):
|
|
self._image = value
|
|
|
|
def __str__(self):
|
|
return self.username
|
|
|
|
def get_absolute_url(self):
|
|
return reverse("users:detail", kwargs={"username": self.username})
|
|
|
|
|
|
class Blurb(models.Model):
|
|
user = models.OneToOneField(
|
|
settings.AUTH_USER_MODEL, on_delete=models.CASCADE, related_name="+"
|
|
)
|
|
skills = models.CharField(max_length=255)
|
|
description = models.TextField()
|
|
|
|
|
|
MEMBERSHIP_ACTIVE = 1
|
|
MEMBERSHIP_CANCELLED = 4
|
|
|
|
MEMBERSHIP_STATUS_CHOICES = (
|
|
(0, "Guest user"),
|
|
(MEMBERSHIP_ACTIVE, "Active membership"),
|
|
(3, "Membership Expired"),
|
|
(MEMBERSHIP_CANCELLED, "Membership Cancelled"),
|
|
)
|
|
|
|
MEMBERSHIP_STRING = {
|
|
0: "Guest user",
|
|
MEMBERSHIP_ACTIVE: "Active membership",
|
|
3: "Membership Expired",
|
|
MEMBERSHIP_CANCELLED: "Membership Cancelled",
|
|
}
|
|
|
|
MEMBERSHIP_STATUS = {
|
|
"signup": 0, # This means the user has not completed signup
|
|
"active": MEMBERSHIP_ACTIVE,
|
|
"expired": 3,
|
|
"cancelled": MEMBERSHIP_CANCELLED,
|
|
}
|
|
|
|
|
|
class Membership(models.Model):
|
|
user = models.OneToOneField(
|
|
settings.AUTH_USER_MODEL,
|
|
null=True,
|
|
blank=True,
|
|
default=None,
|
|
related_name="user",
|
|
unique=True,
|
|
on_delete=models.CASCADE,
|
|
)
|
|
payment = models.DecimalField(max_digits=6, decimal_places=2, default=0.0)
|
|
date = models.DateTimeField()
|
|
reference = models.CharField(max_length=255)
|
|
status = models.PositiveSmallIntegerField(
|
|
default=0, choices=MEMBERSHIP_STATUS_CHOICES
|
|
)
|
|
email = models.CharField(max_length=255, unique=True)
|
|
|
|
@property
|
|
def get_status(self):
|
|
return MEMBERSHIP_STRING[self.status]
|
|
|
|
def is_active(self):
|
|
if self.status is MEMBERSHIP_ACTIVE:
|
|
return True
|
|
return False
|
|
|
|
def lookup_status(name):
|
|
if not name:
|
|
return 0
|
|
return MEMBERSHIP_STATUS.get(name.lower(), 0)
|
|
|
|
def __str__(self):
|
|
return self.reference
|
|
|
|
|
|
# users rfid card to user mapping, user can have more than one card
|
|
class Rfid(models.Model):
|
|
code = models.CharField(max_length=7)
|
|
description = models.CharField(
|
|
_("Short rfid description"), blank=True, max_length=255
|
|
)
|
|
user = models.ForeignKey(
|
|
settings.AUTH_USER_MODEL,
|
|
null=True,
|
|
blank=True,
|
|
on_delete=models.CASCADE
|
|
# related_name='rfid_user'
|
|
)
|
|
|
|
def __str__(self):
|
|
return self.description
|
|
|
|
def name(self):
|
|
return self.user.name
|
|
|
|
|
|
def send_subscription_update_message(sender, instance, **kwargs):
|
|
matrix_message.delay(
|
|
room='admin',
|
|
prefix=' - MEMBERSHIP',
|
|
message='Changed to %s for user %s' % (
|
|
instance.get_status,
|
|
instance.user.username))
|
|
|
|
|
|
# Needs to be change to seend to admin room, and not triger on scheduled job
|
|
# post_save.connect(send_subscription_update_message, sender=Membership)
|