-
Notifications
You must be signed in to change notification settings - Fork 2k
/
lazyjson.py
51 lines (39 loc) · 1.49 KB
/
lazyjson.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
import simplejson as json
import simplejson.encoder as json_encoder
class LazyJSONObject(dict):
'''An object that behaves like a dict returned from json.loads'''
def __init__(self, json_string):
self._json_string = json_string
self._json_dict = None
def _loads(self):
if not self._json_dict:
self._json_dict = json.loads(self._json_string)
self._json_string = None
return self._json_dict
def __nonzero__(self):
return True
def for_json(self):
if self._json_string:
return JSONString(self._json_string)
return self._json_dict
def _loads_method(name):
def method(self, *args, **kwargs):
return getattr(self._loads(), name)(*args, **kwargs)
return method
for fn in ['__cmp__', '__contains__', '__delitem__', '__eq__', '__ge__',
'__getitem__', '__gt__', '__iter__', '__le__', '__len__', '__lt__',
'__ne__', '__setitem__', 'clear', 'copy', 'fromkeys', 'get',
'has_key', 'items', 'iteritems', 'iterkeys', 'itervalues', 'keys',
'pop', 'popitem', 'setdefault', 'update', 'values']:
setattr(LazyJSONObject, fn, _loads_method(fn))
class JSONString(int):
'''
A type for already-encoded JSON
Fake-out simplejson by subclassing int so that simplejson calls
our __str__ method to produce JSON.
'''
def __init__(self, s):
self.s = s
super(JSONString, self).__init__(-1)
def __str__(self):
return s