challenges/challenge1.py at master · thepythonclub/challenges · GitHub

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
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
#!/usr/bin/env python

from twisted.web.server import Site, Session
from twisted.web.resource import Resource
from twisted.internet import reactor
import random
import base64
import json


class ShortSession(Session):
    sessionTimeout = 2


class Challenge1(Resource):
    sessions = set()

    encoded = ''
    solution = ''
    session = None

    def get_answer(self, challenge):
        answers = open('answers.json')
        data = json.load(answers)
        answers.close()
        return str(data[challenge])

    def make_string(self):
        decoded = ''
        alpha = 'abcdefghijklmnopqrstuvwxyz'
        for x in range(32):
            decoded += random.choice(alpha)
        encoded = base64.b64encode(decoded)
        return encoded, decoded

    def render_GET(self, request):
        if 'answer' in str(request):
            return 'Maybe you should try POST ;-)'
        self.encoded, self.solution = self.make_string()
        self.session = request.getSession()
        if self.session.uid not in self.sessions:
            self.sessions.add(self.session.uid)
            self.session.notifyOnExpire(lambda: self._expired(self.session.uid))
        return 'I am Jack\'s base64 encoded string: "' + self.encoded + '"'

    def render_POST(self, request):
        if self.session.uid not in self.sessions:
            self.sessions.add(self.session.uid)
            self.session.notifyOnExpire(lambda: self._expired(self.session.uid))
            return 'Too slow...'

        try:
            answer = json.loads(request.content.getvalue())['answer']
        except ValueError:
            return 'You should be submitting as JSON'
        print answer, self.solution
        if answer == self.solution:
            return self.get_answer('challenge1')
        else:
            return 'If at first you don\'t succeed... try, try again.'

    def _expired(self, uid):
        try:
            self.sessions.remove(uid)
        except KeyError:
            pass


rootResource = Resource()
rootResource.putChild("challenge1", Challenge1())
factory = Site(rootResource)
factory.sessionFactory = ShortSession

reactor.listenTCP(8081, factory)
reactor.run()