challenges/challenge5.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
76
77
78
79
80
81
82
#!/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 json


class ShortSession(Session):
    sessionTimeout = 4


class Challenge5(Resource):
    sessions = set()

    sentence = ''
    solution = ''
    session = None

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

    def pick_language(self):
        sentence = ''
        f = open('verses.json','r')
        text = json.loads(f.read())
        f.close()

        language = random.choice(text.keys())
        sentence = text[language]

        print language, sentence

        return sentence, language

    def render_GET(self, request):
        if 'answer' in str(request):
            return 'Maybe you should try POST ;-)'
        self.sentence, self.solution = self.pick_language()
        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))

        data = (self.sentence).encode('UTF-8')
        string = '<html><meta http-equiv="Content-Type" content="text/html;charset=UTF-8">' + \
                'I am Jack\'s random language: "' + data + '"</html>'
        return string

    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'
        if answer == self.solution:
            return self.get_answer('challenge5')
        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("challenge5", Challenge5())
factory = Site(rootResource)
factory.sessionFactory = ShortSession

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