|
0
|
1 |
# -*- coding: utf-8 -*- |
|
|
2 |
''' |
|
|
3 |
Created on Jun 10, 2013 |
|
|
4 |
|
|
|
5 |
@author: ymh |
|
|
6 |
''' |
|
33
|
7 |
from .models import TermLabel |
|
0
|
8 |
from dateutil import parser |
|
|
9 |
import re |
|
|
10 |
|
|
|
11 |
class ImportProcessor(object): |
|
|
12 |
|
|
|
13 |
def __init__(self, field): |
|
|
14 |
self.field = field |
|
|
15 |
|
|
|
16 |
def process(self, obj, value): |
|
|
17 |
return {} |
|
|
18 |
|
|
|
19 |
|
|
|
20 |
class CharFieldProcessor(ImportProcessor): |
|
|
21 |
|
|
|
22 |
def process(self, obj, value): |
|
|
23 |
setattr(obj, self.field, value) |
|
|
24 |
return {} |
|
|
25 |
|
|
28
|
26 |
class TrimCharFieldProcessor(CharFieldProcessor): |
|
|
27 |
|
|
|
28 |
def process(self, obj, value): |
|
|
29 |
return super(TrimCharFieldProcessor, self).process(obj, value.strip()) |
|
|
30 |
|
|
0
|
31 |
class BooleanFieldProcessor(ImportProcessor): |
|
|
32 |
|
|
|
33 |
def process(self, obj, value): |
|
|
34 |
setattr(obj, self.field, value and value.strip().lower() in ['oui', '1', 't', 'yes', 'y', 'o']) |
|
|
35 |
|
|
|
36 |
class DateFieldProcessor(ImportProcessor): |
|
|
37 |
|
|
|
38 |
def process(self, obj, value): |
|
28
|
39 |
setattr(obj, self.field, parser.parse(value) if value else None) |
|
0
|
40 |
|
|
|
41 |
class TermProcessor(ImportProcessor): |
|
|
42 |
|
|
28
|
43 |
def __init__(self, field, context, notice_term_klass, re_split = r"[\;\,\:\(\)]", re_sub = "\(.+?\)"): |
|
0
|
44 |
ImportProcessor.__init__(self, field) |
|
28
|
45 |
self.re_split = re.compile(re_split) |
|
|
46 |
self.re_sub = re.compile(re_sub) if re_sub else None |
|
0
|
47 |
self.context = context |
|
|
48 |
self.notice_term_klass = notice_term_klass |
|
|
49 |
|
|
|
50 |
def build_notice_term(self, token, obj): |
|
33
|
51 |
|
|
|
52 |
termlabels = list(TermLabel.objects.filter(label=token, term__thesaurus__uri=self.context).select_related()) |
|
|
53 |
if termlabels: |
|
|
54 |
term_obj = termlabels[0].term |
|
|
55 |
if not self.notice_term_klass.objects.filter(notice=obj,term=term_obj).exists(): |
|
|
56 |
return self.notice_term_klass(notice=obj,term=term_obj) |
|
|
57 |
else: |
|
|
58 |
return None |
|
|
59 |
else: |
|
|
60 |
return None |
|
0
|
61 |
|
|
|
62 |
def process(self, obj, value): |
|
|
63 |
res = {} |
|
|
64 |
#remove everything between () |
|
|
65 |
value = getattr(obj, self.field) |
|
|
66 |
if self.re_sub: |
|
28
|
67 |
value = self.re_sub.sub("", value) |
|
|
68 |
for token in self.re_split.split(value): |
|
0
|
69 |
token = token.strip() |
|
|
70 |
nt = self.build_notice_term(token, obj) |
|
|
71 |
if nt is not None: |
|
|
72 |
res.setdefault(self.notice_term_klass,[]).append(nt) |
|
|
73 |
return res |