blob: 052efd68e77d3cbdfff4e8301519ee40766ac5c5 (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
|
from django.contrib.auth.models import AbstractUser
from django.db import models
from django.urls import reverse
from django.utils.translation import ugettext_lazy as _
from ctrack.organisations.models import Stakeholder
class User(AbstractUser):
name = models.CharField(_("Name of User"), blank=True, max_length=255)
stakeholder = models.OneToOneField(
Stakeholder, on_delete=models.CASCADE, null=True, blank=True
)
def get_absolute_url(self):
return reverse("users:detail", kwargs={"username": self.username})
@property
def is_stakeholder(self):
if self.stakeholder is not None:
return True
else:
return False
def get_organisation_name(self):
if self.is_stakeholder:
return self.stakeholder.person.organisation.name
|
