source: mauRepo/MolesManager/trunk/src/MolesManager/djencoder.py @ 8248

Subversion URL: http://proj.badc.rl.ac.uk/svn/ndg/mauRepo/MolesManager/trunk/src/MolesManager/djencoder.py@8280
Revision 8248, 2.0 KB checked in by mnagni, 8 years ago (diff)

Complete - # 22388: CEDA Observation - inSupportOf
 http://team.ceda.ac.uk/trac/ceda/ticket/22388

Line 
1'''
2Created on 9 Feb 2012
3
4@author: mnagni
5'''
6from json.encoder import JSONEncoder
7from MolesManager.forms.date import methodsWithDecorator
8import re
9import logging
10
11class DJEncoder(JSONEncoder):
12
13    log = logging.getLogger('DJEncoder')
14   
15    def __init__(self):
16        self.__markers = {}
17        super(DJEncoder, self).__init__()
18        self.__pattern = re.compile('\D\D__*')
19        self.__pattern2 = re.compile('\A\w+__id\Z')
20       
21
22   
23    def default(self, obj):
24        # Convert objects to a dictionary of their representation
25        d = { '__module__':obj.__module__,
26              }
27        if obj.__class__.__name__ == 'EnumSymbol':
28            d['__class__'] = obj._cls.__name__
29        else:
30            d['__class__'] = obj.__class__.__name__
31           
32        if d['__module__'].startswith('sqlalchemy'):
33            return d
34
35        if d['__class__'] == 'CEDA_Project':
36            print d
37
38        if d['__class__'] == 'Decimal':
39            d.update({'value': str(obj)})
40        else:   
41            for key in obj.__dict__.keys():
42                if not (key.startswith("_") or self.__pattern.match(key) or self.__pattern2.match(key)):
43                    d.update({key: getattr(obj, key)})
44        getters = list(methodsWithDecorator(type(obj), "property"))
45        for name in getters:
46            try:
47                d.update({name: getattr(obj, name)})
48            except Exception as e:
49                DJEncoder.log.error(e)
50       
51        for key, value in d.items():
52            if value is not None and id(value) in self.__markers and not isinstance(value, str) and not isinstance(value, int):
53                continue
54                #return {}
55            else:
56                self.__markers[id(value)] = value               
57        return d
58   
59    @classmethod
60    def escapeForJSON(self, toEscape):
61        res = toEscape.replace("'", "'")
62        #res = res.replace('"', '\\"')
63        res = res.replace('\\', '\\\\')
64        return res
65       
Note: See TracBrowser for help on using the repository browser.