-
Notifications
You must be signed in to change notification settings - Fork 0
/
ChessClient.py
210 lines (180 loc) · 7.59 KB
/
ChessClient.py
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
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
# -*- coding: utf-8 -*-
import requests
import cookielib
from bs4 import BeautifulSoup
import json
import time
import cPickle as pickle
import ChessHelper
from ChessBoard import ChessBoard
from Hall import GameRoom
from Hall import User
import random
class ChessClient():
def __init__(self, server_url):
self.session = requests.Session()
self.session.cookies = cookielib.CookieJar()
agent = 'Mozilla/5.0 (Windows NT 10.0; WOW64) AppleWebKit/537.36 (KHTML, like Gecko) Maxthon/5.1.2.3000 Chrome/55.0.2883.75 Safari/537.36'
self.headers = {
"Host": server_url,
"Origin": server_url,
"Referer": server_url,
'User-Agent': agent
}
self.server_url = server_url
self.board = ChessBoard()
self.last_status_signature = ""
def send_get(self, url):
return self.session.get(self.server_url + url, headers=self.headers)
def send_post(self, url, data):
return self.session.post(self.server_url + url, data, headers=self.headers)
def login_in_guest(self):
response = self.send_get('/login?action=login_in_guest')
soup = BeautifulSoup(response.content, "html.parser")
username_span = soup.find('span', attrs={'id': 'username'})
if username_span:
return username_span.text
else:
return None
def login(self, username, password):
response = self.send_post('/login?action=login',
data={'username': username, 'password': password})
soup = BeautifulSoup(response.content, "html.parser")
username_span = soup.find('span', attrs={'id': 'username'})
if username_span:
return username_span.text
else:
return None
def logout(self):
self.send_get('/login?action=logout')
def join_room(self, roomid):
response = self.send_post('/action?action=joinroom',
data={'roomid': roomid})
action_result = json.loads(response.content)
return action_result
def join_game(self):
response = self.send_get('/action?action=joingame')
action_result = json.loads(response.content)
return action_result
def put_piece(self, row, col):
response = self.send_get(
'/action?action=gameaction&actionid=%s&piece_i=%d&piece_j=%d' % ('put_piece', row, col))
action_result = json.loads(response.content)
return action_result
def get_room_info(self):
response = self.send_get(
'/action?action=gameaction&actionid=%s' % 'get_room_info')
action_result = json.loads(response.content)
room = pickle.loads(str(action_result['info']))
return room
def get_game_info(self):
response = self.send_get(
'/action?action=gameaction&actionid=%s' % 'get_game_info')
action_result = json.loads(response.content)
room = pickle.loads(str(action_result['info']))
return room
def get_user_info(self):
response = self.send_get(
'/action?action=gameaction&actionid=%s' % 'get_user_info')
action_result = json.loads(response.content)
room = pickle.loads(str(action_result['info']))
return room
def wait_game_info_changed(self, interval=0.5, max_time=100):
wait_time = 0
assert interval > 0, "interval must be positive"
while True:
response = self.send_get(
'/action?action=gameaction&actionid=%s' % ('get_status_signature'))
action_result = json.loads(response.content)
if action_result['id'] == 0:
status_signature = action_result['info']
if self.last_status_signature != status_signature:
self.last_status_signature = status_signature
break
else:
print "ERROR get_status_signature,", action_result['id'], action_result['info']
break
time.sleep(interval)
wait_time += interval
if wait_time > max_time:
break
return wait_time
def get_all_rooms(self):
response = self.send_get(
'/action?action=get_all_rooms')
action_result = json.loads(response.content)
all_rooms = action_result['info']
return all_rooms
def answer_take_back(self, agree=True):
response = self.send_get(
'/action?action=gameaction&actionid=answer_take_back&agree=' + ('true' if agree else 'false'))
action_result = json.loads(response.content)
return action_result
class GameStrategy_random():
def __init__(self):
self._chess_helper_move_set = []
for i in range(15):
for j in range(15):
self._chess_helper_move_set.append((i, j))
random.shuffle(self._chess_helper_move_set)
self.try_step = 0
def play_one_piece(self, user, gameboard):
move = self._chess_helper_move_set[self.try_step]
while gameboard.get_piece(move[0], move[1]) != 0 and self.try_step < 15 * 15:
self.try_step += 1
move = self._chess_helper_move_set[self.try_step]
self.try_step += 1
return move
def go_play():
import argparse
parser = argparse.ArgumentParser()
parser.add_argument('--room_name', type=str, default='test_room')
parser.add_argument('--server_url', default='http://192.168.7.61:11111')
parser.add_argument('--ai', default='random')
args = parser.parse_args()
if args.ai == 'random':
strategy = GameStrategy_random()
else:
assert False, "No other ai, you can add one or import the AICollection's ai."
client = ChessClient(args.server_url)
client.login_in_guest()
client.join_room(args.room_name)
client.join_game()
user = client.get_user_info()
print "加入游戏成功，你是:" + ("黑方" if user.game_role == 1 else "白方")
while True:
wait_time = client.wait_game_info_changed()
print 'wait_time:', wait_time
room = client.get_room_info()
# room=GameRoom()
user = client.get_user_info()
# user=User()
gameboard = client.get_game_info()
# gameboard = ChessBoard()
print 'room.get_status():', room.get_status()
print 'user.game_status():', user.game_status
print 'gameboard.game_status():'
ChessHelper.printBoard(gameboard)
if room.get_status() == GameRoom.ROOM_STATUS_NOONE or room.get_status() == GameRoom.ROOM_STATUS_ONEWAITING:
print "等待另一个对手加入游戏:"
continue
elif room.get_status() == GameRoom.ROOM_STATUS_PLAYING:
if room.ask_take_back != 0 and room.ask_take_back != user.game_role:
client.answer_take_back()
break
if gameboard.get_current_user() == user.game_role:
print "轮到你走："
one_legal_piece = strategy.play_one_piece(user, gameboard)
action_result = client.put_piece(*one_legal_piece)
if action_result['id'] != 0:
print "走棋失败:"
print ChessHelper.numToAlp(one_legal_piece[0]), ChessHelper.numToAlp(one_legal_piece[1])
print action_result['info']
else:
print "轮到对手走...."
continue
elif room.get_status() == GameRoom.ROOM_STATUS_FINISH:
print "游戏已经结束了," + ("黑方" if gameboard.get_winner() == 1 else "白方") + " 赢了"
break
if __name__ == "__main__":
go_play()