from django.db import models

from notes.models import Note
from . import managers

# Extends 'notes' module

class Observation(Note):
    """ Note dans le cadre d'une rencontre """

    objects = managers.ObservationManager()
    rencontre = models.ForeignKey(  'maraudes.Rencontre',
                                    related_name="observations",
                                    on_delete=models.CASCADE
                                )

    # Note attributes proxies
    def note_author(self):      return self.rencontre.maraude.referent
    def note_date(self):        return self.rencontre.date
    def note_time(self):        return self.rencontre.heure_debut
    def note_labels(self):      return [self.rencontre.lieu, self.rencontre.heure_debut]
    def note_bg_colors(self):   return ("info", "info")


class Signalement(Note):

    source = models.CharField('Source', max_length=128)

    def note_labels(self): return [self.source]
    def note_bg_colors(self): return ('warning', 'alert')