benedict.serializers.query_string

Complexity

Total Complexity

5

Size/Duplication

Total Lines	38
Duplicated Lines	0 %

Importance

Changes

0

3 Methods

Rating	Name	Duplication	Size	Complexity
A	QueryStringSerializer.__init__()	0	2	1
A	QueryStringSerializer.encode()	0	3	1
A	QueryStringSerializer.decode()	0	11	3

# -*- coding: utf-8 -*-

from benedict.serializers.abstract import AbstractSerializer

try:
    # python 3
    from urllib.parse import urlencode
    from urllib.parse import parse_qs
except ImportError:
    # python 2
    from urllib import urlencode
    from urlparse import parse_qs

import json
import re


class QueryStringSerializer(AbstractSerializer):

    def __init__(self):
        super(QueryStringSerializer, self).__init__()

    def decode(self, s, **kwargs):
        flat = kwargs.pop('flat', True)
        qs_re = r'^(([\w\-\%\+]+\=[\w\-\%\+]*)+([\&]{1})?)+'
        qs_pattern = re.compile(qs_re)
        if qs_pattern.match(s):
            data = parse_qs(s)
            if flat:
                data = {key: value[0] for key, value in data.items()}
            return data
        else:
            raise ValueError('Invalid query string: {}'.format(s))

    def encode(self, d, **kwargs):
        data = urlencode(d, **kwargs)
        return data