Dead simple python script to parse Google style search query

google_query_parser.py

#!/usr/bin/env python

import shlex

def parse_query(query):
    tokens = shlex.split(query)
    dsl = {'search': []}
    key = ''
    for token in tokens:
        if token.endswith(':'):
            key = token[:-1]
        else:
            if ':' in token:
                key, word = token.split(':', 1)
                dsl[key] = word
            elif key:
                dsl[key] = token or ''
            else:
                dsl['search'].append(token)

            key = ''

    if key:
        dsl[key] = ''

    return dsl

if __name__ == '__main__':
    print(parse_query('site:wooyun.org keyword1 author: "White Hat"'))
    # {'search': ['keyword1'], 'site': 'wooyun.org', 'author': 'White Hat'}