#!/usr/bin/env python

import json

from libnmap.objects import NmapHost, NmapService, NmapReport

from libnmap.parser import NmapParser

class ReportEncoder(json.JSONEncoder):

def default(self, obj):

otype = {'NmapHost': NmapHost,

'NmapOSFingerprint': NmapHost.NmapOSFingerprint,

'NmapService': NmapService,

'NmapReport': NmapReport}

if isinstance(obj, tuple(otype.values())):

key = '__%s__' % obj.__class__.__name__

return {key: obj.__dict__}

return json.JSONEncoder.default(self, obj)

class ReportDecoder(json.JSONDecoder):

def decode(self, json_str):

r = NmapParser.parse_fromdict(json.loads(json_str))

return r