summaryrefslogtreecommitdiff
path: root/tko/unique_cookie.py
blob: 292d12ae86653f6991dd722c2968fb27952b1f53 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
#!/usr/bin/env python

import os, random


def unique_id(cookie_key):
    """
    Find out if remote caller has cookie set on the key.
    If not, set cookie on client side: evaluate this key by a random string.
    ( unique user identifier )
    In both scenarios return value of the cookie, be it old or newly set one
    """
    uid = ''
    ## try to retrieve uid from Cookie
    if 'HTTP_COOKIE' in os.environ:
        ## parse os.environ['HTTP_COOKIE']
        cookies = os.environ['HTTP_COOKIE'].split(';')
        key = '%s=' % cookie_key
        uid_cookies = [c for c in cookies if c.strip().startswith(key)]

        if uid_cookies:
            assert(len(uid_cookies) == 1)
            uid_cookie = uid_cookies[0]
            uid = uid_cookie.replace(key, '')

    if not uid:
        uid = str(random.random())[2:16] # random string of 14 digits
        set_cookie_statement = 'Set-Cookie:%s=%s;' % (cookie_key, uid)
        set_cookie_statement += 'expires=Thu, 26-Dec-2013 22:03:25 GMT;'
        print set_cookie_statement

    return uid