src/core/import_processor.py
author ymh <ymh.work@gmail.com>
Tue, 25 Jun 2013 00:00:03 +0200
changeset 33 61c3ffd94f11
parent 28 5918a9d353d0
child 34 b1fd0e0197c8
permissions -rw-r--r--
- correct imports. - optimize import csv - add srep thesaurus
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
0
4095911a7830 Jocondelab first commit before design
ymh <ymh.work@gmail.com>
parents:
diff changeset
     1
# -*- coding: utf-8 -*-
4095911a7830 Jocondelab first commit before design
ymh <ymh.work@gmail.com>
parents:
diff changeset
     2
'''
4095911a7830 Jocondelab first commit before design
ymh <ymh.work@gmail.com>
parents:
diff changeset
     3
Created on Jun 10, 2013
4095911a7830 Jocondelab first commit before design
ymh <ymh.work@gmail.com>
parents:
diff changeset
     4
4095911a7830 Jocondelab first commit before design
ymh <ymh.work@gmail.com>
parents:
diff changeset
     5
@author: ymh
4095911a7830 Jocondelab first commit before design
ymh <ymh.work@gmail.com>
parents:
diff changeset
     6
'''
33
61c3ffd94f11 - correct imports.
ymh <ymh.work@gmail.com>
parents: 28
diff changeset
     7
from .models import TermLabel
0
4095911a7830 Jocondelab first commit before design
ymh <ymh.work@gmail.com>
parents:
diff changeset
     8
from dateutil import parser
4095911a7830 Jocondelab first commit before design
ymh <ymh.work@gmail.com>
parents:
diff changeset
     9
import re
4095911a7830 Jocondelab first commit before design
ymh <ymh.work@gmail.com>
parents:
diff changeset
    10
4095911a7830 Jocondelab first commit before design
ymh <ymh.work@gmail.com>
parents:
diff changeset
    11
class ImportProcessor(object):
4095911a7830 Jocondelab first commit before design
ymh <ymh.work@gmail.com>
parents:
diff changeset
    12
    
4095911a7830 Jocondelab first commit before design
ymh <ymh.work@gmail.com>
parents:
diff changeset
    13
    def __init__(self, field):
4095911a7830 Jocondelab first commit before design
ymh <ymh.work@gmail.com>
parents:
diff changeset
    14
        self.field = field
4095911a7830 Jocondelab first commit before design
ymh <ymh.work@gmail.com>
parents:
diff changeset
    15
        
4095911a7830 Jocondelab first commit before design
ymh <ymh.work@gmail.com>
parents:
diff changeset
    16
    def process(self, obj,  value):
4095911a7830 Jocondelab first commit before design
ymh <ymh.work@gmail.com>
parents:
diff changeset
    17
        return {}
4095911a7830 Jocondelab first commit before design
ymh <ymh.work@gmail.com>
parents:
diff changeset
    18
    
4095911a7830 Jocondelab first commit before design
ymh <ymh.work@gmail.com>
parents:
diff changeset
    19
    
4095911a7830 Jocondelab first commit before design
ymh <ymh.work@gmail.com>
parents:
diff changeset
    20
class CharFieldProcessor(ImportProcessor):
4095911a7830 Jocondelab first commit before design
ymh <ymh.work@gmail.com>
parents:
diff changeset
    21
    
4095911a7830 Jocondelab first commit before design
ymh <ymh.work@gmail.com>
parents:
diff changeset
    22
    def process(self, obj, value):
4095911a7830 Jocondelab first commit before design
ymh <ymh.work@gmail.com>
parents:
diff changeset
    23
        setattr(obj, self.field, value)
4095911a7830 Jocondelab first commit before design
ymh <ymh.work@gmail.com>
parents:
diff changeset
    24
        return {}
4095911a7830 Jocondelab first commit before design
ymh <ymh.work@gmail.com>
parents:
diff changeset
    25
28
5918a9d353d0 Correction in importing csv.
ymh <ymh.work@gmail.com>
parents: 0
diff changeset
    26
class TrimCharFieldProcessor(CharFieldProcessor):
5918a9d353d0 Correction in importing csv.
ymh <ymh.work@gmail.com>
parents: 0
diff changeset
    27
    
5918a9d353d0 Correction in importing csv.
ymh <ymh.work@gmail.com>
parents: 0
diff changeset
    28
    def process(self, obj, value):
5918a9d353d0 Correction in importing csv.
ymh <ymh.work@gmail.com>
parents: 0
diff changeset
    29
        return super(TrimCharFieldProcessor, self).process(obj, value.strip())
5918a9d353d0 Correction in importing csv.
ymh <ymh.work@gmail.com>
parents: 0
diff changeset
    30
0
4095911a7830 Jocondelab first commit before design
ymh <ymh.work@gmail.com>
parents:
diff changeset
    31
class BooleanFieldProcessor(ImportProcessor):
4095911a7830 Jocondelab first commit before design
ymh <ymh.work@gmail.com>
parents:
diff changeset
    32
    
4095911a7830 Jocondelab first commit before design
ymh <ymh.work@gmail.com>
parents:
diff changeset
    33
    def process(self, obj, value):
4095911a7830 Jocondelab first commit before design
ymh <ymh.work@gmail.com>
parents:
diff changeset
    34
        setattr(obj, self.field, value and value.strip().lower() in ['oui', '1', 't', 'yes', 'y', 'o'])
4095911a7830 Jocondelab first commit before design
ymh <ymh.work@gmail.com>
parents:
diff changeset
    35
4095911a7830 Jocondelab first commit before design
ymh <ymh.work@gmail.com>
parents:
diff changeset
    36
class DateFieldProcessor(ImportProcessor):
4095911a7830 Jocondelab first commit before design
ymh <ymh.work@gmail.com>
parents:
diff changeset
    37
    
4095911a7830 Jocondelab first commit before design
ymh <ymh.work@gmail.com>
parents:
diff changeset
    38
    def process(self, obj, value):
28
5918a9d353d0 Correction in importing csv.
ymh <ymh.work@gmail.com>
parents: 0
diff changeset
    39
        setattr(obj, self.field, parser.parse(value) if value else None)
0
4095911a7830 Jocondelab first commit before design
ymh <ymh.work@gmail.com>
parents:
diff changeset
    40
4095911a7830 Jocondelab first commit before design
ymh <ymh.work@gmail.com>
parents:
diff changeset
    41
class TermProcessor(ImportProcessor):
4095911a7830 Jocondelab first commit before design
ymh <ymh.work@gmail.com>
parents:
diff changeset
    42
    
28
5918a9d353d0 Correction in importing csv.
ymh <ymh.work@gmail.com>
parents: 0
diff changeset
    43
    def __init__(self, field, context, notice_term_klass, re_split = r"[\;\,\:\(\)]", re_sub = "\(.+?\)"):
0
4095911a7830 Jocondelab first commit before design
ymh <ymh.work@gmail.com>
parents:
diff changeset
    44
        ImportProcessor.__init__(self, field)
28
5918a9d353d0 Correction in importing csv.
ymh <ymh.work@gmail.com>
parents: 0
diff changeset
    45
        self.re_split = re.compile(re_split)
5918a9d353d0 Correction in importing csv.
ymh <ymh.work@gmail.com>
parents: 0
diff changeset
    46
        self.re_sub = re.compile(re_sub) if re_sub else None
0
4095911a7830 Jocondelab first commit before design
ymh <ymh.work@gmail.com>
parents:
diff changeset
    47
        self.context = context
4095911a7830 Jocondelab first commit before design
ymh <ymh.work@gmail.com>
parents:
diff changeset
    48
        self.notice_term_klass = notice_term_klass
4095911a7830 Jocondelab first commit before design
ymh <ymh.work@gmail.com>
parents:
diff changeset
    49
    
4095911a7830 Jocondelab first commit before design
ymh <ymh.work@gmail.com>
parents:
diff changeset
    50
    def build_notice_term(self, token, obj):
33
61c3ffd94f11 - correct imports.
ymh <ymh.work@gmail.com>
parents: 28
diff changeset
    51
        
61c3ffd94f11 - correct imports.
ymh <ymh.work@gmail.com>
parents: 28
diff changeset
    52
        termlabels = list(TermLabel.objects.filter(label=token, term__thesaurus__uri=self.context).select_related())
61c3ffd94f11 - correct imports.
ymh <ymh.work@gmail.com>
parents: 28
diff changeset
    53
        if termlabels:
61c3ffd94f11 - correct imports.
ymh <ymh.work@gmail.com>
parents: 28
diff changeset
    54
            term_obj = termlabels[0].term
61c3ffd94f11 - correct imports.
ymh <ymh.work@gmail.com>
parents: 28
diff changeset
    55
            if not self.notice_term_klass.objects.filter(notice=obj,term=term_obj).exists():
61c3ffd94f11 - correct imports.
ymh <ymh.work@gmail.com>
parents: 28
diff changeset
    56
                return self.notice_term_klass(notice=obj,term=term_obj)
61c3ffd94f11 - correct imports.
ymh <ymh.work@gmail.com>
parents: 28
diff changeset
    57
            else:
61c3ffd94f11 - correct imports.
ymh <ymh.work@gmail.com>
parents: 28
diff changeset
    58
                return None
61c3ffd94f11 - correct imports.
ymh <ymh.work@gmail.com>
parents: 28
diff changeset
    59
        else:
61c3ffd94f11 - correct imports.
ymh <ymh.work@gmail.com>
parents: 28
diff changeset
    60
            return None            
0
4095911a7830 Jocondelab first commit before design
ymh <ymh.work@gmail.com>
parents:
diff changeset
    61
4095911a7830 Jocondelab first commit before design
ymh <ymh.work@gmail.com>
parents:
diff changeset
    62
    def process(self, obj, value):
4095911a7830 Jocondelab first commit before design
ymh <ymh.work@gmail.com>
parents:
diff changeset
    63
        res = {}
4095911a7830 Jocondelab first commit before design
ymh <ymh.work@gmail.com>
parents:
diff changeset
    64
        #remove everything between ()
4095911a7830 Jocondelab first commit before design
ymh <ymh.work@gmail.com>
parents:
diff changeset
    65
        value = getattr(obj, self.field)
4095911a7830 Jocondelab first commit before design
ymh <ymh.work@gmail.com>
parents:
diff changeset
    66
        if self.re_sub:
28
5918a9d353d0 Correction in importing csv.
ymh <ymh.work@gmail.com>
parents: 0
diff changeset
    67
            value = self.re_sub.sub("", value)
5918a9d353d0 Correction in importing csv.
ymh <ymh.work@gmail.com>
parents: 0
diff changeset
    68
        for token in self.re_split.split(value):
0
4095911a7830 Jocondelab first commit before design
ymh <ymh.work@gmail.com>
parents:
diff changeset
    69
            token = token.strip()
4095911a7830 Jocondelab first commit before design
ymh <ymh.work@gmail.com>
parents:
diff changeset
    70
            nt = self.build_notice_term(token, obj)
4095911a7830 Jocondelab first commit before design
ymh <ymh.work@gmail.com>
parents:
diff changeset
    71
            if nt is not None:
4095911a7830 Jocondelab first commit before design
ymh <ymh.work@gmail.com>
parents:
diff changeset
    72
                res.setdefault(self.notice_term_klass,[]).append(nt)
4095911a7830 Jocondelab first commit before design
ymh <ymh.work@gmail.com>
parents:
diff changeset
    73
        return res