battlechess/server.py at master · guyver2/battlechess · 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
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
#!/usr/bin/python

import socket
import sys
import traceback
import threading
import time
from core.Board import Board
from communication import sendData, recvData, waitForMessage


class GameThread(threading.Thread):
    def __init__(self, client_1, client_2):
        super(GameThread, self).__init__()
        self.client_1 = client_1
        self.client_2 = client_2
        self.nick_1 = ""
        self.nick_2 = ""
        self.board = Board()

    def run(self):
        self.nick_1 = waitForMessage(self.client_1, 'NICK')
        self.nick_2 = waitForMessage(self.client_2, 'NICK')
        filename = time.strftime("games/%Y_%m_%d_%H_%M_%S")
        filename += "_"+self.nick_1+"_Vs_"+self.nick_2+".txt"
        log = open(filename, "w")
        log.write(self.nick_1+' Vs. '+self.nick_2+'\n')

        sendData(self.client_1, 'URLR', "http://git.sxbn.org/battleChess/"+filename)
        sendData(self.client_2, 'URLR', "http://git.sxbn.org/battleChess/"+filename)

        sendData(self.client_1, 'NICK', self.nick_2)
        sendData(self.client_2, 'NICK', self.nick_1)

        loop = True
        try:
            while loop:
                valid = False
                while not valid:
                    head, move = recvData(self.client_1)
                    if head == "OVER":
                        loop = False
                        raise ValueError()  # jump to finaly
                    elif head == 'MOVE':
                        i, j, ii, jj = move
                        valid, pos, msg = self.board.move(i, j, ii, jj, 'w')
                        # print "got move from", [i,j], "to", [ii,jj], "from white", valid
                        sendData(self.client_1, 'VALD', valid)
                    else:
                        print('error : server was expecting MOVE, not', head)
                        raise ValueError()

                log.write("%d %d %d %d\n" % (i, j, ii, jj))
                if self.board.winner:  # if we have a winner, send the whole board
                    endBoard = self.board.toString()
                    sendData(self.client_1, 'BORD', endBoard)
                    sendData(self.client_2, 'BORD', endBoard)
                    break  # game is over
                else:
                    sendData(self.client_1, 'BORD', self.board.toString('w'))
                    sendData(self.client_2, 'BORD', self.board.toString('b'))

                valid = False
                while not valid:
                    head, move = recvData(self.client_2)
                    if head == "OVER":
                        loop = False
                        raise ValueError()  # jump to finaly
                    elif head == 'MOVE':
                        i, j, ii, jj = move
                        valid, pos, msg = self.board.move(i, j, ii, jj, 'b')
                        # print "got move from", [i,j], "to", [ii,jj], "from black", valid
                        sendData(self.client_2, 'VALD', valid)
                    else:
                        print('error : server was expecting MOVE, not', head)
                        raise ValueError()

                log.write("%d %d %d %d\n" % (i, j, ii, jj))
                if self.board.winner:  # if we have awinner, send the whole board
                    endBoard = self.board.toString()
                    sendData(self.client_1, 'BORD', endBoard)
                    sendData(self.client_2, 'BORD', endBoard)
                    break  # game is over
                else:
                    sendData(self.client_1, 'BORD', self.board.toString('w'))
                    sendData(self.client_2, 'BORD', self.board.toString('b'))
        except Exception as e:
            print(e)
            traceback.print_exc(file=sys.stdout)
            pass
        finally:  # Always close the game
            # print "finishing the game"
            log.flush()
            log.close()
            sendData(self.client_1, 'OVER', None)
            sendData(self.client_2, 'OVER', None)
            self.client_1.close()
            self.client_2.close()


if __name__ == '__main__':
    PORT = 8887
    HOST = ''  # default value
    if len(sys.argv) == 1:
        print("Usage:\n\t", sys.argv[0], "PORT")
    if len(sys.argv) > 1:
        PORT = int(sys.argv[1])
    if len(sys.argv) > 2:
        HOST = sys.argv[2]

    # print " starting server on "+socket.gethostname()+":"+str(PORT)

    # create an INET, STREAMing socket
    serversocket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
    serversocket.bind((HOST, PORT))

    # register SIGINT to close socket
    def signal_handler(signal, frame):
        print('You pressed Ctrl+C!')
        serversocket.close()
        sys.exit(0)

    # become a server socket
    serversocket.listen(5)
    loop = True
    while loop:
        try:
            # accept connections from outside
            (client_1, address) = serversocket.accept()
            sendData(client_1, 'COLR', 'white')
            (client_2, address) = serversocket.accept()
            sendData(client_2, 'COLR', 'black')
            game = GameThread(client_1, client_2)
            game.start()
        except Exception as e:
            traceback.print_exc(file=sys.stdout)
            print(e)
            loop = False
            serversocket.close()