1 import urlparse

2

3

4 def parse_cookies(environ):

5 return urlparse.parse_qs(environ.get("HTTP_COOKIE", "").replace("; ", "&"))

6

7

8 class PinFailError(Exception):

9 pass

10

11

12 class PinMan(object):

13 def __init__(self, key):

14 assert key

15 self._key = key

16 self._fn = "_%s" % key

17

18 def load(self):

19 ret = None

20

21 try:

22 with open(self._fn) as pin_f:

23 ret = pin_f.read().strip()

24 except IOError:

25 pass

26

27 return ret

28

29 def check(self, cookies):

30 if self._key not in cookies:

31 raise PinFailError()

32

33 pin = cookies[self._key][0]

34 if pin != self.load():

35 raise PinFailError()

36

37 return pin