Commit | Line | Data |
---|---|---|
92c7413b | 1 | # -*- encoding: utf-8 -*- |
b7a741ad | 2 | import simplejson, uuid, datetime, caldav, vobject, uuid |
6d885e0c | 3 | from django.contrib.auth.models import User |
d15017b2 | 4 | from django.db import models |
f12cc7fb | 5 | from django.db.models import Q |
b7a741ad | 6 | from django.db.models.signals import pre_delete |
92c7413b | 7 | from timezones.fields import TimeZoneField |
6d885e0c | 8 | from auf_savoirs_en_partage.backend_config import RESOURCES |
da9020f3 | 9 | from savoirs.globals import META |
b7a741ad | 10 | from settings import CALENDRIER_URL |
e3c3296e | 11 | from datamaster_modeles.models import Thematique, Pays, Region |
b7a741ad | 12 | from lib.calendrier import combine |
13 | from caldav.lib import error | |
d15017b2 CR |
14 | |
15 | class Discipline(models.Model): | |
16 | id = models.IntegerField(primary_key=True, db_column='id_discipline') | |
17 | nom = models.CharField(max_length=765, db_column='nom_discipline') | |
6ef8ead4 CR |
18 | |
19 | def __unicode__ (self): | |
92c7413b | 20 | return self.nom |
6ef8ead4 | 21 | |
d15017b2 CR |
22 | class Meta: |
23 | db_table = u'discipline' | |
24 | ordering = ["nom",] | |
25 | ||
79c398f6 CR |
26 | class SourceActualite(models.Model): |
27 | nom = models.CharField(max_length=255) | |
28 | url = models.CharField(max_length=255) | |
ccbc4363 | 29 | |
30 | def __unicode__(self,): | |
31 | return u"%s" % self.nom | |
79c398f6 | 32 | |
2f9c4d6c EMS |
33 | class ActualiteManager(models.Manager): |
34 | ||
35 | def get_query_set(self): | |
36 | return ActualiteQuerySet(self.model) | |
37 | ||
da44ce68 EMS |
38 | def search(self, text): |
39 | return self.get_query_set().search(text) | |
40 | ||
2f9c4d6c EMS |
41 | class ActualiteQuerySet(models.query.QuerySet): |
42 | ||
43 | def search(self, text): | |
c9cbc784 | 44 | return self.filter(Q(titre__icontains=text) | Q(texte__icontains=text)) |
2f9c4d6c | 45 | |
d15017b2 | 46 | class Actualite(models.Model): |
4f262f90 | 47 | id = models.AutoField(primary_key=True, db_column='id_actualite') |
d15017b2 CR |
48 | titre = models.CharField(max_length=765, db_column='titre_actualite') |
49 | texte = models.TextField(db_column='texte_actualite') | |
50 | url = models.CharField(max_length=765, db_column='url_actualite') | |
d15017b2 | 51 | date = models.DateField(db_column='date_actualite') |
f554ef70 | 52 | visible = models.BooleanField(db_column='visible_actualite', default = False) |
3b6456b0 | 53 | ancienid = models.IntegerField(db_column='ancienId_actualite', blank = True, null = True) |
ccbc4363 | 54 | source = models.ForeignKey(SourceActualite, blank = True, null = True) |
6ef8ead4 | 55 | |
2f9c4d6c EMS |
56 | objects = ActualiteManager() |
57 | ||
6ef8ead4 | 58 | def __unicode__ (self): |
f554ef70 | 59 | return "%s" % (self.titre) |
6ef8ead4 | 60 | |
d15017b2 CR |
61 | class Meta: |
62 | db_table = u'actualite' | |
63 | ordering = ["-date",] | |
92c7413b | 64 | |
92c7413b | 65 | class Evenement(models.Model): |
c5b3da8b | 66 | uid = models.CharField(max_length = 255, default = str(uuid.uuid1())) |
92c7413b CR |
67 | approuve = models.BooleanField(default = False) |
68 | titre = models.CharField(max_length=255) | |
69 | discipline = models.ForeignKey('Discipline', related_name = "discipline", | |
70 | blank = True, null = True) | |
71 | discipline_secondaire = models.ForeignKey('Discipline', related_name = \ | |
72 | "discipline_secondaire", | |
73 | verbose_name = \ | |
74 | "Discipline secondaire", | |
75 | blank = True, null = True) | |
76 | mots_cles = models.TextField('Mots-Clés', blank = True, null = True) | |
77 | type = models.CharField(max_length = 255, choices = \ | |
78 | (('Colloque', 'Colloque'), | |
79 | ('Conférence', 'Conférence'), | |
80 | ('Appel à contribution', 'Appel à contribution'), | |
81 | ('Journée d\'étude', 'Journée d\'étude'), | |
82 | (None, 'Autre') | |
83 | )) #TODO: choices | |
84 | fuseau = TimeZoneField(verbose_name = 'Fuseau horaire') | |
85 | debut = models.DateTimeField(default = datetime.datetime.now) | |
86 | fin = models.DateTimeField(default = datetime.datetime.now) | |
87 | lieu = models.TextField() | |
88 | description = models.TextField(blank = True, null = True) | |
89 | #fichiers = TODO? | |
90 | contact = models.TextField(blank = True, null = True) | |
91 | url = models.CharField(max_length=255, blank = True, null = True) | |
92 | ||
020f79a9 | 93 | def __unicode__(self,): |
94 | return "[%s] %s" % (self.uid, self.titre) | |
95 | ||
b7a741ad | 96 | def save(self, *args, **kwargs): |
97 | """Sauvegarde l'objet dans django et le synchronise avec caldav s'il a été | |
98 | approuvé""" | |
99 | self.update_vevent() | |
100 | super(Evenement, self).save(*args, **kwargs) | |
101 | ||
102 | # methodes de commnunications avec CALDAV | |
103 | def as_ical(self,): | |
104 | """Retourne l'evenement django sous forme d'objet icalendar""" | |
105 | cal = vobject.iCalendar() | |
106 | cal.add('vevent') | |
107 | ||
108 | # fournit son propre uid | |
7f56d0d4 | 109 | if self.uid in [None, ""]: |
b7a741ad | 110 | self.uid = str(uuid.uuid1()) |
111 | ||
112 | cal.vevent.add('uid').value = self.uid | |
113 | ||
114 | cal.vevent.add('summary').value = self.titre | |
115 | ||
116 | if self.mots_cles is None: | |
117 | kw = [] | |
118 | else: | |
119 | kw = self.mots_cles.split(",") | |
120 | ||
121 | try: | |
122 | kw.append(self.discipline.nom) | |
123 | kw.append(self.discipline_secondaire.nom) | |
124 | kw.append(self.type) | |
125 | except: pass | |
126 | ||
79b400f0 | 127 | kw = [x.strip() for x in kw if len(x.strip()) > 0 and x is not None] |
b7a741ad | 128 | for k in kw: |
129 | cal.vevent.add('x-auf-keywords').value = k | |
130 | ||
131 | description = self.description | |
132 | if len(kw) > 0: | |
133 | if len(self.description) > 0: | |
134 | description += "\n" | |
028f548f | 135 | description += u"Mots-clés: " + ", ".join(kw) |
b7a741ad | 136 | |
137 | cal.vevent.add('dtstart').value = combine(self.debut, self.fuseau) | |
138 | cal.vevent.add('dtend').value = combine(self.fin, self.fuseau) | |
139 | cal.vevent.add('created').value = combine(datetime.datetime.now(), "UTC") | |
140 | cal.vevent.add('dtstamp').value = combine(datetime.datetime.now(), "UTC") | |
79b400f0 | 141 | if len(description) > 0: |
b7a741ad | 142 | cal.vevent.add('description').value = description |
143 | if len(self.contact) > 0: | |
144 | cal.vevent.add('contact').value = self.contact | |
145 | if len(self.url) > 0: | |
146 | cal.vevent.add('url').value = self.url | |
147 | if len(self.lieu) > 0: | |
148 | cal.vevent.add('location').value = self.lieu | |
149 | return cal | |
150 | ||
151 | def update_vevent(self,): | |
152 | """Essaie de créer l'évènement sur le serveur ical. | |
153 | En cas de succès, l'évènement local devient donc inactif et approuvé""" | |
154 | try: | |
155 | if self.approuve: | |
156 | event = self.as_ical() | |
157 | client = caldav.DAVClient(CALENDRIER_URL) | |
158 | cal = caldav.Calendar(client, url = CALENDRIER_URL) | |
159 | e = caldav.Event(client, parent = cal, data = event.serialize(), id=self.uid) | |
160 | e.save() | |
161 | except: | |
162 | self.approuve = False | |
163 | ||
164 | def delete_vevent(self,): | |
165 | """Supprime l'evenement sur le serveur caldav""" | |
166 | try: | |
167 | if self.approuve: | |
168 | event = self.as_ical() | |
169 | client = caldav.DAVClient(CALENDRIER_URL) | |
170 | cal = caldav.Calendar(client, url = CALENDRIER_URL) | |
171 | e = cal.event(self.uid) | |
172 | e.delete() | |
173 | except error.NotFoundError: | |
174 | pass | |
175 | ||
176 | ||
177 | # Surcharge du comportement de suppression | |
178 | # La méthode de connexion par signals est préférable à surcharger la méthode delete() | |
179 | # car dans le cas de la suppression par lots, cell-ci n'est pas invoquée | |
180 | def delete_vevent(sender, instance, *args, **kwargs): | |
181 | instance.delete_vevent() | |
182 | ||
183 | pre_delete.connect(delete_vevent, sender = Evenement) | |
184 | ||
185 | ||
d972b61d | 186 | class ListSet(models.Model): |
187 | spec = models.CharField(primary_key = True, max_length = 255) | |
188 | name = models.CharField(max_length = 255) | |
189 | server = models.CharField(max_length = 255) | |
9eda5d6c | 190 | validated = models.BooleanField(default = True) |
d972b61d | 191 | |
10d37e44 | 192 | def __unicode__(self,): |
193 | return self.name | |
194 | ||
da44ce68 EMS |
195 | class RecordManager(models.Manager): |
196 | ||
197 | def get_query_set(self): | |
198 | return RecordQuerySet(self.model) | |
199 | ||
200 | def search(self, text): | |
201 | return self.get_query_set().search(text) | |
202 | ||
203 | class RecordQuerySet(models.query.QuerySet): | |
204 | ||
205 | def search(self, text): | |
f12cc7fb | 206 | qs = self |
da44ce68 | 207 | words = text.split() |
da44ce68 | 208 | |
f12cc7fb EMS |
209 | # Ne garder que les ressources qui contiennent tous les mots |
210 | # demandés. | |
211 | for word in words: | |
212 | qs = qs.filter(Q(title__icontains=word) | Q(description__icontains=word) | | |
213 | Q(creator__icontains=word) | Q(contributor__icontains=word) | | |
214 | Q(subject__icontains=word)) | |
215 | ||
216 | # On donne un point pour chaque mot présent dans le titre. | |
217 | score_expr = ' + '.join(['(title LIKE %s)'] * len(words)) | |
218 | score_params = ['%' + word + '%' for word in words] | |
219 | return qs.extra( | |
220 | select={'score': score_expr}, | |
221 | select_params=score_params | |
222 | ).order_by('-score') | |
223 | ||
224 | def search_auteur(self, text): | |
225 | qs = self | |
226 | for word in text.split(): | |
227 | qs = qs.filter(Q(creator__icontains=word) | Q(contributor__icontains=word)) | |
228 | return qs | |
229 | ||
230 | def search_sujet(self, text): | |
231 | qs = self | |
232 | for word in text.split(): | |
233 | qs = qs.filter(subject__icontains=word) | |
234 | return qs | |
235 | ||
236 | def search_titre(self, text): | |
237 | qs = self | |
238 | for word in text.split(): | |
239 | qs = qs.filter(title__icontains=word) | |
240 | return qs | |
241 | ||
0cc5f772 | 242 | class Record(models.Model): |
23b5b3d5 | 243 | |
244 | #fonctionnement interne | |
0cc5f772 | 245 | id = models.AutoField(primary_key = True) |
23b5b3d5 | 246 | server = models.CharField(max_length = 255) |
247 | last_update = models.CharField(max_length = 255) | |
248 | last_checksum = models.CharField(max_length = 255) | |
c88d78dc | 249 | validated = models.BooleanField(default = True) |
23b5b3d5 | 250 | |
251 | #OAI | |
252 | title = models.TextField(null = True, blank = True) | |
253 | creator = models.TextField(null = True, blank = True) | |
254 | description = models.TextField(null = True, blank = True) | |
255 | modified = models.CharField(max_length = 255, null = True, blank = True) | |
256 | identifier = models.CharField(max_length = 255, null = True, blank = True, unique = True) | |
257 | uri = models.CharField(max_length = 255, null = True, blank = True, unique = True) | |
258 | source = models.TextField(null = True, blank = True) | |
259 | contributor = models.TextField(null = True, blank = True) | |
260 | subject = models.TextField(null = True, blank = True) | |
261 | publisher = models.TextField(null = True, blank = True) | |
262 | type = models.TextField(null = True, blank = True) | |
263 | format = models.TextField(null = True, blank = True) | |
264 | language = models.TextField(null = True, blank = True) | |
da9020f3 | 265 | |
c88d78dc | 266 | listsets = models.ManyToManyField(ListSet, null = True, blank = True) |
d972b61d | 267 | |
da9020f3 | 268 | #SEP 2 (aucune données récoltées) |
23b5b3d5 | 269 | alt_title = models.TextField(null = True, blank = True) |
270 | abstract = models.TextField(null = True, blank = True) | |
271 | creation = models.CharField(max_length = 255, null = True, blank = True) | |
272 | issued = models.CharField(max_length = 255, null = True, blank = True) | |
273 | isbn = models.TextField(null = True, blank = True) | |
274 | orig_lang = models.TextField(null = True, blank = True) | |
da9020f3 | 275 | |
276 | # Metadata AUF multivaluées | |
277 | disciplines = models.ManyToManyField(Discipline) | |
278 | thematiques = models.ManyToManyField(Thematique) | |
e3c3296e | 279 | pays = models.ManyToManyField(Pays) |
280 | regions = models.ManyToManyField(Region) | |
0cc5f772 | 281 | |
da44ce68 EMS |
282 | # Manager |
283 | objects = RecordManager() | |
284 | ||
6d885e0c | 285 | def est_complet(self,): |
286 | """teste si le record à toutes les données obligatoires""" | |
287 | return self.disciplines.count() > 0 and \ | |
288 | self.thematiques.count() > 0 and \ | |
289 | self.pays.count() > 0 and \ | |
290 | self.regions.count() > 0 | |
291 | ||
0cc5f772 | 292 | def __unicode__(self): |
f554ef70 | 293 | return "[%s] %s" % (self.server, self.title) |
da9020f3 | 294 | |
6d885e0c | 295 | class Serveur(models.Model): |
b7a741ad | 296 | """Identification d'un serveur d'ou proviennent les références""" |
6d885e0c | 297 | nom = models.CharField(primary_key = True, max_length = 255) |
298 | ||
299 | def __unicode__(self,): | |
300 | return self.nom | |
301 | ||
302 | def conf_2_db(self,): | |
303 | for k in RESOURCES.keys(): | |
304 | s, created = Serveur.objects.get_or_create(nom=k) | |
305 | s.nom = k | |
306 | s.save() | |
307 | ||
308 | class Profile(models.Model): | |
309 | user = models.ForeignKey(User, unique=True) | |
310 | serveurs = models.ManyToManyField(Serveur, null = True, blank = True) | |
0cc5f772 CR |
311 | |
312 | class HarvestLog(models.Model): | |
23b5b3d5 | 313 | context = models.CharField(max_length = 255) |
314 | name = models.CharField(max_length = 255) | |
0cc5f772 | 315 | date = models.DateTimeField(auto_now = True) |
23b5b3d5 | 316 | added = models.IntegerField(null = True, blank = True) |
317 | updated = models.IntegerField(null = True, blank = True) | |
a85ba76e | 318 | processed = models.IntegerField(null = True, blank = True) |
23b5b3d5 | 319 | record = models.ForeignKey(Record, null = True, blank = True) |
320 | ||
321 | @staticmethod | |
322 | def add(message): | |
323 | logger = HarvestLog() | |
324 | if message.has_key('record_id'): | |
325 | message['record'] = Record.objects.get(id=message['record_id']) | |
326 | del(message['record_id']) | |
327 | ||
328 | for k,v in message.items(): | |
329 | setattr(logger, k, v) | |
330 | logger.save() |