hubl/src/server/models.py

from django.db import models
from django.contrib.auth.models import User
from django.db.models.signals import post_save
from django.dispatch import receiver


class Member(models.Model):
    avatar = models.URLField(blank=True)
    available = models.NullBooleanField(blank=True)
    bio = models.CharField(max_length=255, blank=True, null=True)
    cell = models.ForeignKey("Cell", on_delete=models.DO_NOTHING, blank=True, null=True)
    jabberID = models.CharField(max_length=255, blank=True, null=True)
    jabberRoom = models.NullBooleanField(blank=True)
    user = models.OneToOneField(User, on_delete=models.CASCADE)
    number = models.CharField(max_length=255, blank=True, null=True)
    projects = models.ManyToManyField("Project", blank=True)
    pseudo = models.CharField(max_length=255, blank=True, null=True)
    roles = models.ManyToManyField("Role", blank=True)
    skills = models.ManyToManyField("Skill", blank=True)
    website = models.URLField(blank=True)

    def __str__(self):
        return 'Profil de {} '.format(self.user.get_full_name())

@receiver(post_save, sender=User)
def create_user_profile(sender, instance, created, **kwargs):
    if created:
        Member.objects.create(user=instance)
    else:
        instance.member.save()


class Skill(models.Model):
    name = models.CharField(max_length=255, default='')

    def __str__(self):
        return self.name


class Role(models.Model):
    name = models.CharField(max_length=255, default='')

    def __str__(self):
        return self.name


class Project(models.Model):
    name = models.CharField(max_length=255, default='')
    number = models.PositiveIntegerField(default='0', blank=True)
    client = models.ForeignKey("Client", on_delete=models.DO_NOTHING)
    team = models.ManyToManyField("Member", blank=True)

    def __str__(self):
        return self.name
        

class Client(models.Model):
    name = models.CharField(max_length=255, default='')
    address = models.CharField(max_length=255, default='')
    logo = models.URLField()

    def __str__(self):
        return self.name


class Cell(models.Model):
    name = models.CharField(max_length=255, default='')
    city = models.CharField(max_length=255, default='')
    country = models.CharField(max_length=255, default='')

    def __str__(self):
        return self.name


class Channel(models.Model):
    name = models.CharField(max_length=255, default='')
    description = models.CharField(max_length=255, default='')
    members = models.ManyToManyField("Member", blank=True)
    owner = models.ForeignKey("Member", related_name="owned_chatrooms", on_delete=models.DO_NOTHING)

    def __str__(self):
        return self.name