mirror of
https://github.com/ubf/ubf.git
synced 2026-04-17 02:06:03 +00:00
72 lines
2.4 KiB
Python
72 lines
2.4 KiB
Python
|
|
##
|
|
##
|
|
|
|
from pyubf import Integer, Float, Atom, String, Binary, Record, Tuple, List, PropList
|
|
|
|
## pyjson -> pyubf
|
|
def from_pyjson(value):
|
|
if value is None:
|
|
return value
|
|
elif isinstance(value, bool):
|
|
return value
|
|
elif isinstance(value, int) or isinstance(value, long):
|
|
return Integer(value)
|
|
elif isinstance(value, float):
|
|
return Float(value)
|
|
elif isinstance(value, str) or isinstance(value, unicode):
|
|
return Binary(value)
|
|
elif isinstance(value, dict):
|
|
if value.keys() == ['$A']:
|
|
return Atom(value['$A'])
|
|
elif value.keys() == ['$S']:
|
|
return String(value['$S'])
|
|
elif value.keys() == ['$T']:
|
|
return Tuple(tuple(from_pyjson(value['$T'])))
|
|
elif value.has_key('$R') and len(value.keys()) > 1:
|
|
name = value['$R']
|
|
del value['$R']
|
|
fields = dict([ [k, from_pyjson(v)] for k, v in value.iteritems() ])
|
|
return Record(name, fields)
|
|
elif value.has_key('$P') and len(value.keys()) > 1:
|
|
l = [ [k, from_pyjson(v)] for k, v in value.iteritems() ]
|
|
return PropList(dict(l))
|
|
else:
|
|
l = [ [k, from_pyjson(v)] for k, v in value.iteritems() ]
|
|
return dict(l)
|
|
elif isinstance(value, list):
|
|
return [ from_pyjson(v) for v in value ]
|
|
else:
|
|
raise RuntimeError, 'unknown object ~s' % repr(value)
|
|
|
|
## pyubf -> pyjson
|
|
def to_pyjson(value):
|
|
if value is None:
|
|
return value
|
|
elif isinstance(value, bool):
|
|
return value
|
|
elif isinstance(value, int) or isinstance(value, long):
|
|
return value
|
|
elif isinstance(value, float):
|
|
return value
|
|
elif isinstance(value, Atom):
|
|
return {'$A' : str(value)}
|
|
elif isinstance(value, String):
|
|
return {'$S' : str(value)}
|
|
elif isinstance(value, str):
|
|
return value
|
|
elif isinstance(value, unicode):
|
|
return value
|
|
elif isinstance(value, tuple):
|
|
return {'$T' : [ to_pyjson(v) for v in list(value) ]}
|
|
elif isinstance(value, Record):
|
|
record = dict([ [k, to_pyjson(v)] for k, v in value.iteritems() ])
|
|
record.update({'$R' : value.name})
|
|
return record
|
|
elif isinstance(value, list):
|
|
return [ to_pyjson(v) for v in value ]
|
|
elif isinstance(value, dict):
|
|
return dict([ [k, to_pyjson(v)] for k, v in value.iteritems() ])
|
|
else:
|
|
raise RuntimeError, 'unknown object ~s' % repr(value)
|