-
Notifications
You must be signed in to change notification settings - Fork 0
/
Player_class.py
206 lines (156 loc) · 11.5 KB
/
Player_class.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
import requests
import json
import expectedValueWN8
import averageExpectedOverallValueWN8
from operator import itemgetter
class Player:
def __init__(self, server: str, user_name: str):
# load all_tank_data.json file, this is a dictionary of all tank information (e.g. tank name, tier, etc. ), key is based on Tank ID
self.allTankopediaData = (json.load(open("all_tank_data.json")))["data"]
self.username = user_name
self.playerServer = server.lower()
# TODO API error handling
if self.playerServer == "na":
player_info = ((requests.get("https://api.worldoftanks.com/wot/account/list/?application_id=bd644ca5adf8dc631b1598528a4b7fc1&search=" + self.username)).json())\
["data"][0] # DICT value keys for userID
else:
player_info = ((requests.get("https://api.worldoftanks." + self.playerServer + "/wot/account/list/?application_id=bd644ca5adf8dc631b1598528a4b7fc1&search=" + self.username)).json())\
["data"][0]
self.userID = player_info["account_id"]
#---------------------------------------------------------------------------------------------------------------------------------------------------------
# overall stats
if self.playerServer == "na":
overall_info = ((requests.get("https://api.worldoftanks.com/wot/account/info/?application_id=bd644ca5adf8dc631b1598528a4b7fc1&account_id=" + str(self.userID))).json())\
["data"][str(self.userID)] # DICT value keys for overall player info
else:
overall_info = ((requests.get("https://api.worldoftanks." + self.playerServer + "/wot/account/info/?application_id=bd644ca5adf8dc631b1598528a4b7fc1&account_id=" + str(self.userID))).json())\
["data"][str(self.userID)]
self.wgRating = overall_info["global_rating"]
self.overallStats = {}
self.overallStats.update(overall_info["statistics"]["all"])
self.totalBattles = self.overallStats["battles"]
self.winRate = round((self.overallStats["wins"] / self.totalBattles) * 100, 2)
self.dpg = round(self.overallStats["damage_dealt"] / self.totalBattles)
#---------------------------------------------------------------------------------------------------------------------------------------------------------
def fetchStats(self):
# getting tank-specific stats from WG API
if self.playerServer == "na":
# lists of all the tanks a player has nested in a dictionary
self.allTankStats = ((requests.get("https://api.worldoftanks.com/wot/tanks/stats/?application_id=bd644ca5adf8dc631b1598528a4b7fc1&account_id=" + str(self.userID) +\
"&fields=-in_garage%2C+-frags%2C+-max_frags%2C+-team%2C+-stronghold_defense%2C+-globalmap%2C+-clan%2C+-stronghold_skirmish%2C+-company%2C+-regular_team%2C+-account_id%2C+-max_xp"\
)).json())["data"][str(self.userID)] # dictionaries nested in list
else:
self.allTankStats = ((requests.get("https://api.worldoftanks." + self.playerServer + "/wot/tanks/stats/?application_id=bd644ca5adf8dc631b1598528a4b7fc1&account_id=" + str(self.userID) +\
"&fields=-in_garage%2C+-frags%2C+-max_frags%2C+-team%2C+-stronghold_defense%2C+-globalmap%2C+-clan%2C+-stronghold_skirmish%2C+-company%2C+-regular_team%2C+-account_id%2C+-max_xp"\
)).json())["data"][str(self.userID)]
# number of tanks a player has ACCORDING to the WOT API, may not be the same once cross-referenced with the XVM expected values JSON file
#print(type(self.allTankStats))
#print(self.allTankStats)
self.numberOfTanks = len(self.allTankStats)
# list of all tank wn8 values, list is empty in constructor but gets filled when the tankWN8 method is called
self.allTankWN8 = []
self.skippedTankID = []
#self.tankWN8()
#self.overallAccountWn8 = int(self.overallWN8())
#self.colorIcon = Color_icon_class.ColorIcon(self.overallAccountWn8, self.winRate, self.wgRating, self.playerServer)
self.allTankBattles = {tank["tank_id"]: tank["all"] for tank in self.allTankStats} # format is tank_id: {"all" tank stats}
#print(len(self.allTankBattles))
#print(self.allTankBattles)
# returns a dictionary of overall stats for a specific tank at that instant of time
def individualTank(self, tank_id):
return self.allTankBattles[tank_id]
#---------------------------------------------------------------------------------------------------------------------------------------------------------
# calculate wn8 for each individual tank
def tankWN8(self):
# for loop iterates through each tank's random battles, variable randBattleStats is a nested dictionary containing tank stats
for i in self.allTankStats:
randBattleStats = i["all"]
if randBattleStats["battles"] == 0:
pass
else:
try:
# Calculate wn8, parameters are (id, avgDamage, avgDef, avgFrag, avgSpots, winrate)
wn8 = expectedValueWN8.calculateWn8(i["tank_id"], (randBattleStats["damage_dealt"]/randBattleStats["battles"]), (randBattleStats["dropped_capture_points"]/randBattleStats["battles"])\
, (randBattleStats["frags"]/randBattleStats["battles"]), (randBattleStats["spotted"]/randBattleStats["battles"]), ((randBattleStats["wins"]/randBattleStats["battles"]) * 100))
tank_stat = {"Tank ID": i["tank_id"], "Tank WN8": int(wn8), "Tank Battles": randBattleStats["battles"],\
"Tank Name": self.allTankopediaData[str(i["tank_id"])]["name"], "Tier": self.allTankopediaData[str(i["tank_id"])]["tier"]}
self.allTankWN8.append(tank_stat)
#print(tank_stat)
#print(len(self.allTankWN8))
# in the event that tank_id from WG API cannot be found in JSON file list, tank is skipped
except AttributeError:
self.skippedTankID.append(i["tank_id"])
#print(str(i["tank_id"]) + " was skipped!")
# some tank_id's from xvm JSON cant seem to be found in tankopedia request, but can found in overall player stats
except KeyError:
#print("key error")
# Calculate wn8, parameters are (id, avgDamage, avgDef, avgFrag, avgSpots, winrate)
wn8 = expectedValueWN8.calculateWn8(i["tank_id"], (randBattleStats["damage_dealt"]/randBattleStats["battles"]), (randBattleStats["dropped_capture_points"]/randBattleStats["battles"])\
, (randBattleStats["frags"]/randBattleStats["battles"]), (randBattleStats["spotted"]/randBattleStats["battles"]), ((randBattleStats["wins"]/randBattleStats["battles"]) * 100))
tank_stat = {"Tank ID": i["tank_id"], "Tank WN8": int(wn8), "Tank Battles": randBattleStats["battles"]}
self.allTankWN8.append(tank_stat)
#print(tank_stat)
#print(len(self.allTankWN8))
#---------------------------------------------------------------------------------------------------------------------------------------------------------
# calculate overall wn8 by using overall stats and weighted averages of expected values
def overallWN8(self):
overall_stats_input = []
total_damage = total_def = total_frag = total_spot = total_wr = 0
#print("-------------------------------------------------")
#print("All Tank ID and Battles from WOT API for the Player: ")
#print("-------------------------------------------------")
for i in self.allTankStats:
tank_values = {"Tank ID": i["tank_id"], "Battles": i["all"]["battles"]}
#print(tank_values)
overall_stats_input.append(tank_values)
total_damage += i["all"]["damage_dealt"]
total_def += i["all"]["dropped_capture_points"]
total_frag += i["all"]["frags"]
total_spot += i["all"]["spotted"]
total_wr += (i["all"]["wins"]) * 100
#print("-------------------------------------------------")
return averageExpectedOverallValueWN8.calculateOverallWn8(overall_stats_input, total_damage, total_def, total_frag, total_spot, total_wr)
#---------------------------------------------------------------------------------------------------------------------------------------------------------
# calculated overall wn8 of a player by weighted averages
def overallWeightedWn8(self):
# calculates the overall wn8 of a player, done by averaging wn8 per tank weighted by number of games per tank
sum_of_damage_times_battles = 0
total_battles = 0
for i in self.allTankStats:
sum_of_damage_times_battles += (i["Tank WN8"] * i["Tank Battles"])
total_battles += i["Tank Battles"]
print("Total battles from cross-referencing JSON: " + str(total_battles))
return (sum_of_damage_times_battles / total_battles)
#---------------------------------------------------------------------------------------------------------------------------------------------------------
def sortedListOfTanks(self):
return sorted(self.allTankWN8, key = itemgetter("Tank Battles"), reverse = True)
#---------------------------------------------------------------------------------------------------------------------------------------------------------
# returns overall account stats in a newline format
def overallDiscordAccountStats(self):
return (f"> **Username:** {self.username} \n> **Player server:** {str(self.playerServer)} {self.colorIcon.serverIcon()} \n> **Overall winrate:** {str(self.winRate)} {self.colorIcon.colorWN8()}"
f"\n> **Overall wn8:** {str(self.overallAccountWn8)} {self.colorIcon.colorWN8()} \n> **WG rating:** {str(self.wgRating)} {self.colorIcon.colorWGRating()}")
def accountTankStats(self):
print("-------------------------------------------------")
print("Sorted list of Tanks: ")
print("-------------------------------------------------")
for i in self.sortedListOfTanks():
print(i)
print("-------------------------------------------------")
def print(self):
print("-------------------------------------------------")
print(self.username)
print(self.playerServer)
print("User ID is: " + str(self.userID))
print("WG rating is: " + str(self.wgRating))
print("Total battles is: " + str(self.totalBattles))
print("Overall win rate is " + str(self.winRate))
print("Overall dpg is " + str(self.dpg))
print("-------------------------------------------------")
print("Sorted list of Tanks: ")
print("-------------------------------------------------")
for i in self.sortedListOfTanks():
print(i)
print("-------------------------------------------------")
#test = Player("na", "waikin_reppinKL")
#test.fetchStats()
#print(test.print())