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

'''
Created on 9 Feb 2012

@author: mnagni
'''
from json.encoder import JSONEncoder
from MolesManager.forms.date import methodsWithDecorator
import re
import logging

class DJEncoder(JSONEncoder):

    log = logging.getLogger('DJEncoder')
    
    def __init__(self):
        self.__markers = {}
        super(DJEncoder, self).__init__()
        self.__pattern = re.compile('\D\D__*')
        self.__pattern2 = re.compile('\A\w+__id\Z')
        

    def default(self, obj):
        # Convert objects to a dictionary of their representation
        d = { '__module__':obj.__module__,
              }
        if obj.__class__.__name__ == 'EnumSymbol':
            d['__class__'] = obj._cls.__name__
        else:
            d['__class__'] = obj.__class__.__name__
            
        if d['__module__'].startswith('sqlalchemy'):
            return d

        if d['__class__'] == 'Decimal':
            d.update({'value': str(obj)})
        else:    
            for key in obj.__dict__.keys():
                if not (key.startswith("_") or self.__pattern.match(key) or self.__pattern2.match(key)):
                    d.update({key: getattr(obj, key)})
        getters = list(methodsWithDecorator(type(obj), "property"))
        for name in getters:
            try:
                d.update({name: getattr(obj, name)})
            except Exception as e:
                DJEncoder.log.error(e)
        
        for key, value in d.items():
            if value is not None and id(value) in self.__markers and not isinstance(value, str) and not isinstance(value, int):
                continue
                #return {}
            else:
                self.__markers[id(value)] = value
                '''
                if isinstance(value, str) or isinstance(value, unicode):                    
                    self.__markers[id(value)] = escapeForJSON(value)
                else:
                    self.__markers[id(value)] = value
                '''
        return d

def encodeToJSON(toEncode):
    return escapeForJSON(DJEncoder().encode(toEncode))    
    
def escapeForJSON(toEscape):
    res = toEscape.replace('"', '"')
    res = res.replace("'", "'")        
    #res = res.replace("'", "'")
    res = res.replace('\\', '\\')
    #res = res.replace('(', '(')        
    #res = res.replace(')', ')')                
    return res