Assigner les disciplines et régions en bloc dans l'admin

[auf_savoirs_en_partage_django.git] / auf_savoirs_en_partage / sitotheque / models.py

# -*- encoding: utf-8 -*-
from django.db import models
from django.db.models import Q
from datamaster_modeles.models import *
from savoirs.models import Discipline

TYPE_SITE_CHOICES = (
    ('RV', 'Revue en ligne'), 
    ('BB', 'Bibliothèque en ligne'),
    ('FD', 'Fonds documentaire'),
    ('AR', 'Archive ouverte'),
    ('CO', 'Cours en ligne'),
    ('RE', 'Repertoire de ressource'),
    ('SA', 'Site associatif'),
    ('SC', 'Site culturel'),
    ('SI', 'Site d\'information'),
    ('AU', 'Autre type de site'),
    )

class SiteManager(models.Manager):

    def get_query_set(self):
        return SiteQuerySet(self.model)

    def search(self, text):
        return self.get_query_set().search(text)

class SiteQuerySet(models.query.QuerySet):

    def search(self, text):
        qs = self
        q = None
        for word in text.split():
            part = (Q(titre__icontains=word) |
                    Q(description__icontains=word) |
                    Q(editeur__icontains=word) |
                    Q(auteur__icontains=word) |
                    Q(mots_cles__icontains=word) |
                    Q(discipline__nom__icontains=word) |
                    Q(pays__nom__icontains=word))
            if q is None:
                q = part
            else:
                q = q & part
        if q is not None:
            qs = qs.filter(q).distinct()
        return qs

class Site(models.Model):
    """Fiche d'info d'un site web"""
    url = models.URLField(verify_exists=False)   # dc:identifier (dc:source?)
    titre = models.CharField(max_length=255, verbose_name='Titre')   # dc.title
    description = models.TextField(null=True, blank=True)
    editeur = models.CharField(max_length=255, null=True, blank=True, verbose_name='Éditeur')    # dc.publisher : organisation resp
    auteur = models.CharField(max_length=255, null=True, blank=True, verbose_name='Auteur')  # dc.creator : nom, prénom
    
    #auf_partenaire = models.BooleanField()    # dc.contributor
    
    date_publication = models.DateField(null=True, blank=True)      # dc.date : date de publication
    type = models.CharField(max_length=2, null=True, blank=True, choices=TYPE_SITE_CHOICES,
                              verbose_name = 'Type de site')    # dc.type
    discipline = models.ManyToManyField(Discipline, blank=True)
    thematique = models.ManyToManyField(Thematique, blank=True)
    
    mots_cles =  models.TextField(verbose_name='Mots-clés', null=True, blank=True)    # dc:subject    # indexation libre

    # source    # dc:source (dc:relation?)
    pays = models.ForeignKey(Pays, null = True, blank=True, db_column='pays', to_field='code', verbose_name = 'Pays')
    regions = models.ManyToManyField(Region, blank=True, related_name="sites", verbose_name='Régions')
    
    # meta
    actif = models.BooleanField()
    date_maj = models.DateField(auto_now=True)

    # Manager
    objects = SiteManager()
    
    def __unicode__(self):
        return "%s" % (self.titre)
        
    def type_display(self):
        for t in TYPE_SITE_CHOICES:
            if self.type == t[0]:
                return t[1]
        return "-"

    def assigner_regions(self, regions):
        self.regions.add(*regions)

    def assigner_disciplines(self, disciplines):
        self.discipline.add(*disciplines)