SimpleDistributedComputer/Runner.py at master · ArticuNode/SimpleDistributedComputer · 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
#-------------------------------------------------------------------------------
# Name:        module1
# Purpose:
#
# Author:      RedSpiderMkV
#
# Created:     17/06/2014
# Copyright:   (c) RedSpiderMkV 2014
# Licence:     <your licence>
#-------------------------------------------------------------------------------

import cPickle as Pickle
import socket
import sys

class UdpCommunicator:
    host = ''
    port = 0

    replyAddress = ''

    def __init__(self, host, port):
        self.host = host
        self.port = port

        self.udpListener = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
        self.udpListener.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
        self.udpListener.bind((host, port))

    def GetAndRunTask(self):
        data, addr = self.udpListener.recvfrom(1024)
        theTask = Pickle.loads(data)
        ans = theTask.Calculate()

        self.udpListener.close()
        self.replyAddress = addr[0]

        return ans

    def SendResponse(self, ans):
        senderSocket = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
        senderSocket.sendto(str(ans), (self.replyAddress, 8086))
        senderSocket.close()


def main():
    host = sys.argv[1]
    port = int(sys.argv[2])

    taskRunner = UdpCommunicator(host, port)
    taskResult = taskRunner.GetAndRunTask()

    print taskResult

    taskRunner.SendResponse(taskResult)

if __name__ == '__main__':
    main()