mirror of
https://github.com/status-im/spiff-arena.git
synced 2025-02-11 09:16:45 +00:00
7c219fd731
git-subtree-dir: SpiffWorkflow git-subtree-split: 63db3e45947ec66b8d0efc2c74064004f8ff482c
82 lines
2.7 KiB
Python
82 lines
2.7 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
# This library is free software; you can redistribute it and/or
|
|
# modify it under the terms of the GNU Lesser General Public
|
|
# License as published by the Free Software Foundation; either
|
|
# version 2.1 of the License, or (at your option) any later version.
|
|
#
|
|
# This library is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
|
# Lesser General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU Lesser General Public
|
|
# License along with this library; if not, write to the Free Software
|
|
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
|
|
# 02110-1301 USA
|
|
import json
|
|
import uuid
|
|
from ..operators import Attrib
|
|
from .dict import DictionarySerializer
|
|
|
|
class JSONSerializer(DictionarySerializer):
|
|
|
|
def serialize_workflow_spec(self, wf_spec, **kwargs):
|
|
thedict = super(JSONSerializer, self).serialize_workflow_spec(
|
|
wf_spec, **kwargs)
|
|
return self._dumps(thedict)
|
|
|
|
def deserialize_workflow_spec(self, s_state, **kwargs):
|
|
thedict = self._loads(s_state)
|
|
return super(JSONSerializer, self).deserialize_workflow_spec(
|
|
thedict, **kwargs)
|
|
|
|
def serialize_workflow(self, workflow, **kwargs):
|
|
thedict = super(JSONSerializer, self).serialize_workflow(
|
|
workflow, **kwargs)
|
|
return self._dumps(thedict)
|
|
|
|
def deserialize_workflow(self, s_state, **kwargs):
|
|
thedict = self._loads(s_state)
|
|
return super(JSONSerializer, self).deserialize_workflow(
|
|
thedict, **kwargs)
|
|
|
|
def _object_hook(self, dct):
|
|
if '__uuid__' in dct:
|
|
return uuid.UUID(dct['__uuid__'])
|
|
|
|
if '__bytes__' in dct:
|
|
return dct['__bytes__'].encode('ascii')
|
|
|
|
if '__attrib__' in dct:
|
|
return Attrib(dct['__attrib__'])
|
|
|
|
return dct
|
|
|
|
def _jsonableHandler(self, obj):
|
|
if hasattr(obj, 'jsonable'):
|
|
return obj.jsonable()
|
|
|
|
raise 'Object of type %s with value of %s is not JSON serializable' % (
|
|
type(obj), repr(obj))
|
|
|
|
|
|
def _default(self, obj):
|
|
if isinstance(obj, uuid.UUID):
|
|
return {'__uuid__': obj.hex}
|
|
|
|
if isinstance(obj, bytes):
|
|
return {'__bytes__': obj.decode('ascii')}
|
|
|
|
if isinstance(obj, Attrib):
|
|
return {'__attrib__': obj.name}
|
|
|
|
raise TypeError('%r is not JSON serializable' % obj)
|
|
|
|
def _loads(self, text):
|
|
return json.loads(text, object_hook=lambda o: self._object_hook(o))
|
|
|
|
def _dumps(self, dct):
|
|
return json.dumps(dct, sort_keys=True, default=lambda o:
|
|
self._default(o))
|