werkzeug parse_set_cookie

parse_set_cookie.py

import re
import datetime
import werkzeug

rct = re.compile('((?P<name>[^=;]+)(=(?P<value>("([^";]+|\\.)*"|[^";]+))?)?($|; ))')

def parse_set_cookie(headers, dt=None):
    '''
    :type headers: werkzeug.http.Headers
    :type dt: datetime.datetime
    :rtype: dict
    '''
    if dt is None:
        dt = datetime.datetime.now()
    cookies = {}
    for header in headers.get_all('Set-Cookie'):
        items = [item.groupdict() for item in rct.finditer(header)]
        name, items[0]['name'] = items[0]['name'], 'value'
        cookies[name] = {
            key: (
                werkzeug.parse_date(value) if key == 'Expires' else
                int(value) if key == 'Max-Age' else
                True if key in ('Secure', 'HttpOnly') else
                value[1:-1] if '"' == value[0] == value[-1] else
                value
                )
            for key, value in (
                (item['name'], item['value'].strip() if item['value'] else None)
                for item in items
                )
            }
    return cookies

if __name__ == '__main__':
    import pprint
    headers = werkzeug.Headers()
    headers.add('Set-Cookie', 'one=patata; Max-Age=0')
    headers.add('Set-Cookie', 'two=patata; Expires=Wed, 21 Oct 2015 07:28:00 GMT; Secure; HttpOnly')
    pprint.pprint(parse_set_cookie(headers))