Dict <-> String parser for NICE - KOREA INFORMATION SERVICE

nicecoder.py

#
# Dict <-> String parser for NICE - KOREA INFORMATION SERVICE (sigh)
#
# [email protected]
#
from itertools import izip


def encode(d):
    res = []
    for key in d.keys():
        res.append('%d:%s%d:%s' % (len(key), key, len(str(d[key])), str(d[key])))
    return ''.join(res)


def decode(s):
    tokens = []
    try:
        while s is not '':
            t = s.split(':', 1)
            s = s[len(t[0])+1:]
            tokens.append(s[:int(t[0])])
            s = s[int(t[0]):]
    except:
        return None

    i = iter(tokens)
    return dict(izip(i, i))
    

def test():
    s = '3:foo3:bar'
    d = {'foo':'bar'}
    assert(s == encode(d))
    assert(d == decode(s))