108 lines
3.3 KiB
Python
108 lines
3.3 KiB
Python
from django.db import models
|
|
import jsonfield
|
|
from django.contrib.auth.models import User, AbstractBaseUser
|
|
from django.conf import settings
|
|
from PIL import Image
|
|
|
|
# MAKE EMAIL UNIQUE
|
|
from django.contrib.auth.models import AbstractUser, User
|
|
from django.contrib.auth.models import Permission
|
|
|
|
# UNIQUE and NO BLANK fields while user-registration
|
|
User._meta.get_field('email')._unique = True
|
|
User._meta.get_field('email').blank = False
|
|
User._meta.get_field('first_name').blank = False
|
|
User._meta.get_field('last_name').blank = False
|
|
|
|
'''
|
|
Class Agency
|
|
|
|
Speichert alle Infos für die Agency, darunter später auch die Aufgaben und Bereiche
|
|
Bezug normale User: Erhalten bei Erstellung durch den Administrator automatisch die aktuelle agency:
|
|
request.user.profile.agency
|
|
--> Dahinter ist das Agency-Objekt!
|
|
|
|
'''
|
|
|
|
class Agency(models.Model):
|
|
name = models.CharField(default="", max_length=200)
|
|
inhaber = models.CharField(default="", max_length=200, blank=True)
|
|
street = models.CharField(default="", max_length=200, blank=True)
|
|
city = models.CharField(default="", max_length=200, blank=True)
|
|
email = models.EmailField(default="", blank=True)
|
|
phone = models.CharField(default="", max_length=20, blank=True)
|
|
agencypic = models.ImageField(default='default.jpg', upload_to='agencymain', blank=True)
|
|
|
|
def __str__(self):
|
|
return f'{self.name}'
|
|
|
|
# Hier wird definiert, welche Parameter in der URL übertragen werden inkl. Bezeichner
|
|
def get_absolute_url(self):
|
|
return reverse('agency-update', kwargs={'pk':self.pk})
|
|
|
|
'''
|
|
Class Profile
|
|
|
|
Klasse für die Zusatzinfos eines Nutzers.
|
|
- Vorname
|
|
- Nachname
|
|
- Email
|
|
- Bild
|
|
- Telefonnummer
|
|
- Fest
|
|
- Mobil
|
|
- Funktion
|
|
- Auswahlfelder
|
|
- Außendienst/Innendienst/Azubi
|
|
- Rollen
|
|
- Bereiche
|
|
- Aufgaben
|
|
|
|
'''
|
|
class Profile(models.Model):
|
|
|
|
# AGENCY TASKS
|
|
agency_task = [
|
|
("lead", "Agenturleitung"),
|
|
("external", "Außendienst"),
|
|
("indoor", "Innendienst"),
|
|
("trainee", "Azubi"),
|
|
]
|
|
|
|
# Wenn der User gelöscht wird, wird auch das Profil gelöscht
|
|
user = models.OneToOneField(User, on_delete=models.CASCADE)
|
|
phoneland = models.CharField(max_length=60, blank=True)
|
|
phonemobile = models.CharField(max_length=60, blank=True)
|
|
func = models.CharField(choices=agency_task, default="external", max_length=50)
|
|
# Wenn dieses Profil gelöscht wird, wird NICHT die Agency geslöscht
|
|
agency = models.ForeignKey(Agency, on_delete=models.PROTECT)
|
|
image = models.ImageField(default='default.jpg', upload_to='userprofilepics', blank=True)
|
|
compfunc = models.CharField(max_length=60, blank=True)
|
|
|
|
def __str__(self):
|
|
return f'{self.user.last_name}'
|
|
|
|
# Hier Path für Templates des Models mit Parametern
|
|
def get_absolute_url(self):
|
|
return reverse('users-update', kwargs={'pk':self.pk})
|
|
|
|
# Erst Oberklasse speichern, dann Bild verkleinern
|
|
def save(self, **kwargs):
|
|
super().save()
|
|
img = Image.open(self.image.path)
|
|
|
|
# Bildspeichergröße
|
|
if(img.height > 300 or img.width > 300):
|
|
output_size = (300,300)
|
|
img.thumbnail(output_size)
|
|
img.save(self.image.path)
|
|
|
|
# PERMISSIONS - Über alle Modelle hinweg, in der url.py wird dann die route verhindert!
|
|
# Im template: if perms.users.PERMISSION
|
|
class Meta:
|
|
permissions = [
|
|
('agency_change', 'Agenturinformationen verändern'),
|
|
('users_usermanagement', 'Benutzer bearbeiten'),
|
|
('tasks_management', 'Aufgabenbereiche bearbeiten'),
|
|
('areas_management', 'Bereiche bearbeiten')
|
|
] |