-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathdirector.py
225 lines (210 loc) · 8.74 KB
/
director.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
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
# -*- coding: utf-8 -*-
# @Time : 22/01/2018 2:39 PM
# @Author : Akio
# import time
# from post.post import *
from post.post_subclasses import *
from log import Log
import traceback
import sys
class Director:
def __init__(self, user):
self.user = user
self.next_step = 'login'
self.wrong_num = 0
self.tag = False
self.wrong_info = set()
# TODO decide where to create and close log
self.log = Log()
self.log.log('Running ' + self.user.data['phone'])
def run(self):
try:
while (not self.tag) and self.wrong_num < 3:
self.do_post()
# self.get_field_info()
except Exception as err:
# TODO check if need \n, then delete
# traceback.print_exception(*sys.exc_info())
self.log.log('\n----- Unexpected Error-----\n')
for line in traceback.TracebackException(*sys.exc_info()).format():
self.log.log(line)
self.log.log('\n----- End -----')
self.log.close()
exit(1)
# TODO check
self.log.flush()
# @staticmethod
# def sleep():
# time.sleep(0.2)
def do_post(self):
self.log.log('Beginning' + ' ' + self.next_step)
post_obj = None
if self.next_step == 'login':
post_obj = Login(self.user, self.log)
elif self.next_step == 'getRatio':
post_obj = GetRatio(self.user, self.log)
elif self.next_step == 'getFieldEggs':
post_obj = GetFieldEggs(self.user, self.log)
elif self.next_step == 'hatchField':
post_obj = HatchField(self.user, self.log)
elif self.next_step == 'cleanFriend':
post_obj = CleanFriend(self.user, self.log)
elif self.next_step == 'end':
GetFieldInfo(self.user, self.log)
self.tag = True
return
self.next_step = post_obj.next_step
if not post_obj.wrong_info == '':
self.wrong_num += 1
self.wrong_info.add(post_obj.wrong_info)
# sleep
# if self.next_step == 'login':
# self.login()
# elif self.next_step == 'getRatio':
# self.get_ratio()
# elif self.next_step == 'getFieldEggs':
# self.get_field_eggs()
# elif self.next_step == 'hatchField':
# self.hatch_field()
# elif self.next_step == 'cleanFriend':
# self.clean_friend()
# elif self.next_step == 'end':
# self.tag = True
# self.sleep()
# # 登陆
# def login(self):
# post_dict = OrderedDict()
# post_dict['phone'] = self.user.data.get('phone')
# post_dict['pwd'] = self.user.data.get('pwd')
# post_obj = Post('login', post_dict, 'userId', 'token')
# if post_obj.success:
# response_dic = post_obj.get_response_dic()
# self.user.update(response_dic)
# self.next_step = 'getRatio'
# else:
# self.wrong_num += 1
# self.wrong_info.append('登录')
# self.next_step = 'login'
#
# # 获得利率
# def get_ratio(self):
# post_dict = OrderedDict()
# post_dict['userId'] = self.user.data.get('userId')
# post_dict['token'] = self.user.data.get('token')
# post_obj = Post('getRatio', post_dict, 'chickenCount') # , 'ratios'
# if post_obj.success:
# response_dic = post_obj.get_response_dic()
# self.user.update(response_dic)
# self.next_step = 'getFieldEggs'
# else:
# self.wrong_num += 1
# self.wrong_info.append('首页')
# self.next_step = 'login'
#
# # 收小兔
# def get_field_eggs(self):
# # 如果想要fields需要先getFieldInfo
# self.get_field_info()
# fields_list = self.user.data['fields']
# for filed_info in fields_list:
# if filed_info['hasEgg'] == '1':
# post_dict = OrderedDict()
# post_dict['userId'] = self.user.data.get('userId')
# post_dict['fieldId'] = filed_info.get('id')
# post_dict['token'] = self.user.data.get('token')
# # 没有数据需要更新到user.data
# post_obj = Post('getFieldEggs', post_dict)
# if not post_obj.success:
# self.wrong_info.append('getFieldEggs: ' + filed_info.get('id'))
# self.next_step = 'cleanFriend'
#
# # 收好友小兔
# def clean_friend(self):
# self.get_friend_list()
# if self.user.data['friends']:
# for friend in self.user.data['friends']:
# # 还没被清扫
# if friend['hasClean'] == '0':
# post_dict = OrderedDict()
# post_dict['userId'] = self.user.data.get('userId')
# post_dict['friendId'] = friend.get('userId')
# post_dict['token'] = self.user.data.get('token')
# post_obj = Post('cleanFriend', post_dict)
# # self.sleep()
# if not post_obj.success:
# # 重写错误信息格式
# # self.wrong_info.append('clean_friend')
# print('--------------error--------------')
# print(post_obj.response_json)
# print('--------------/error--------------')
# else:
# print('------------success:------------')
# for key, val in post_obj.response_json.items():
# print(key, val)
# print('------------/success:------------')
# self.next_step = 'hatchField'
#
# # 获取好友列表
# def get_friend_list(self):
# post_dict = OrderedDict()
# post_dict['userId'] = self.user.data.get('userId')
# post_dict['token'] = self.user.data.get('token')
# post_obj = Post('getFriendList', post_dict, 'friends')
# if post_obj.success:
# response_dic = post_obj.get_response_dic()
# # 作为终端没有下一步指令，只update数据
# self.user.update(response_dic)
# else:
# self.wrong_num += 1
# self.wrong_info.append('getFieldInfo')
#
# # 增养小兔子
# def hatch_field(self):
# self.get_field_info()
# max_num = 3000
# # 对小兔取整数
# egg_count = int(float(self.user.data['eggCount']))
# fields_list = self.user.data['fields']
# for filed_info in fields_list:
# if egg_count > 0 and filed_info['active']:
# # 可以孵化位置数量
# blank_space = max_num - int(filed_info['chickens'])
# if blank_space > 0:
# post_dict = OrderedDict()
# add_count = egg_count if egg_count < blank_space else blank_space
# egg_count -= add_count
# post_dict['userId'] = self.user.data.get('userId')
# post_dict['fieldId'] = filed_info.get('id')
# post_dict['addCount'] = str(add_count)
# post_dict['token'] = self.user.data.get('token')
# # 没有数据需要更新到user.data
# post_obj = Post('hatchField', post_dict)
# # self.sleep()
# if post_obj.success:
# egg_count = int(float(self.user.data['eggCount']) - add_count)
# else:
# self.wrong_info.append('hatch_field: ' + filed_info.get('id'))
# self.next_step = 'end' # 下一步是啥
#
# # 获取/更新 信息
# def get_field_info(self):
# post_dict = OrderedDict()
# post_dict['userId'] = self.user.data.get('userId')
# # 暂时不处理好友
# post_dict['friendId'] = ''
# post_dict['token'] = self.user.data.get('token')
# post_obj = Post('getFieldInfo', post_dict, 'chickenCount', 'eggCount', 'fields')
# if post_obj.success:
# response_dic = post_obj.get_response_dic()
# # 作为终端没有下一步指令，只update数据
# self.user.update(response_dic)
# else:
# print('error')
# print(post_obj.response_json)
# self.wrong_num += 1
# self.wrong_info.append('getFieldInfo')
# 判断success=='0'时候的处理方法可以根据response_json['message']来判断
def post_fail(self, post_obj):
message = post_obj.response_json['message']
if message == '已超时，请重新登录':
self.next_step = 'login'