From 11c8f7064db7d5b2088d867ccdcff2a3fbe2ee38 Mon Sep 17 00:00:00 2001 From: NK2 <74492155+FireFighter1027@users.noreply.github.com> Date: Sat, 3 Apr 2021 20:52:26 +0530 Subject: [PATCH] Added stats management file This module will be able to create, update the data and html webpage document which are stats.json and stats_webpage.html :) --- dist/ba_root/mods/stats/mystats.py | 204 +++++++++++++++++++++++++++++ 1 file changed, 204 insertions(+) diff --git a/dist/ba_root/mods/stats/mystats.py b/dist/ba_root/mods/stats/mystats.py index e69de29..cd4fc4a 100644 --- a/dist/ba_root/mods/stats/mystats.py +++ b/dist/ba_root/mods/stats/mystats.py @@ -0,0 +1,204 @@ +damage_data = {} +#Don't touch the above line +""" +mystats module for BombSquad version 1.5.29 +Provides functionality for dumping player stats to disk between rounds. +""" + +import threading,json,os,urllib.request,ba,_ba,setting +from ba._activity import Activity +from ba._music import setmusic, MusicType +from ba._enums import InputType, UIScale +# False-positive from pylint due to our class-generics-filter. +from ba._player import EmptyPlayer # pylint: disable=W0611 +from ba._team import EmptyTeam # pylint: disable=W0611 +from typing import Any, Dict, Optional +from ba._lobby import JoinInfo +from ba import _activitytypes as ba_actypes +from ba._activitytypes import * +our_settings = setting.get_setting() +# where our stats file and pretty html output will go +base_path = os.path.join(_ba.env()['python_directory_user'],"stats" + os.sep) +statsfile = base_path + 'stats.json' +htmlfile = base_path + 'stats_page.html' +table_style = "{width:100%;border: 3px solid black;border-spacing: 5px;border-collapse:collapse;text-align:center;background-color:#fff}" +heading_style = "{border: 3px solid black;text-align:center;}" +html_start = f''' + + + + Test Server + + + +

Top 200 Players of {our_settings['server_name']}

+ + + + + + + + + +''' +# #removed this line as it isn't crt data +def refreshStats(): + # lastly, write a pretty html version. + # our stats url could point at something like this... + stats = setting.getStats() + f=open(htmlfile, 'w') + f.write(html_start) + entries = [(a['scores'], a['kills'], a['deaths'], a['games'], a['name_html'], a['aid']) for a in stats.values()] + # this gives us a list of kills/names sorted high-to-low + entries.sort(reverse=True) + rank = 0 + toppers = {} + pStats = stats + toppersIDs=[] + for entry in entries: + if True: + rank += 1 + scores = str(entry[0]) + kills = str(entry[1]) + deaths = str(entry[2]) + games = str(entry[3]) + name = str(entry[4]) + aid = str(entry[5]) + if rank < 6: toppersIDs.append(aid) + #The below kd and avg_score will not be added to website's html document, it will be only added in stats.json + try: + kd = str(float(kills) / float(deaths)) + kd_int = kd.split('.')[0] + kd_dec = kd.split('.')[1] + p_kd = kd_int + '.' + kd_dec[:3] + except Exception: + p_kd = "0" + try: + avg_score = str(float(scores) / float(games)) + avg_score_int = avg_score.split('.')[0] + avg_score_dec = avg_score.split('.')[1] + p_avg_score = avg_score_int + '.' + avg_score_dec[:3] + except Exception: + p_avg_score = "0" + if damage_data and aid in damage_data: + dmg = damage_data[aid] + dmg = str(str(dmg).split('.')[0] + '.' + str(dmg).split('.')[1][:3]) + else: dmg = 0 + pStats[str(aid)]["rank"] = int(rank) + pStats[str(aid)]["scores"] = int(scores) + pStats[str(aid)]["total_damage"] += float(dmg) #not working properly + pStats[str(aid)]["games"] = int(games) + pStats[str(aid)]["kills"] = int(kills) + pStats[str(aid)]["deaths"] = int(deaths) + pStats[str(aid)]["kd"] = float(p_kd) + pStats[str(aid)]["avg_score"] = float(p_avg_score) + + if rank < 201: + # #removed this line as it isn't crt data + f.write(f''' + + + + + + + + ''') + f.write(''' +
RankNameScoreKillsDeathsGames Played
Total Damage{str(dmg)}
{str(rank)}{str(name)}{str(scores)}{str(kills)}{str(deaths)}{str(games)}
+ +''') + f.close() + + f2 = open(statsfile, "w") + f2.write(json.dumps(pStats, indent=4)) + f2.close() + + from playersData import pdata + pdata.update_toppers(toppersIDS) + +def update(score_set): + """ + Given a Session's ScoreSet, tallies per-account kills + and passes them to a background thread to process and + store. + """ + # look at score-set entries to tally per-account kills for this round + + account_kills = {} + account_deaths = {} + account_scores = {} + + for p_entry in score_set.get_records().values(): + account_id = p_entry.player.get_account_id() + if account_id is not None: + account_kills.setdefault(account_id, 0) # make sure exists + account_kills[account_id] += p_entry.accum_kill_count + account_deaths.setdefault(account_id, 0) # make sure exists + account_deaths[account_id] += p_entry.accum_killed_count + account_scores.setdefault(account_id, 0) # make sure exists + account_scores[account_id] += p_entry.accumscore + # Ok; now we've got a dict of account-ids and kills. + # Now lets kick off a background thread to load existing scores + # from disk, do display-string lookups for accounts that need them, + # and write everything back to disk (along with a pretty html version) + # We use a background thread so our server doesn't hitch while doing this. + if account_scores: UpdateThread(account_kills, account_deaths, account_scores).start() + +class UpdateThread(threading.Thread): + def __init__(self, account_kills, account_deaths, account_scores): + threading.Thread.__init__(self) + self._account_kills = account_kills + self.account_deaths = account_deaths + self.account_scores = account_scores + def run(self): + # pull our existing stats from disk + try: + if os.path.exists(statsfile): + with open(statsfile) as f: + stats = json.loads(f.read()) + except: + return + + # now add this batch of kills to our persistant stats + for account_id, kill_count in self._account_kills.items(): + # add a new entry for any accounts that dont have one + if account_id not in stats: + # also lets ask the master-server for their account-display-str. + # (we only do this when first creating the entry to save time, + # though it may be smart to refresh it periodically since + # it may change) + '''url = 'http://bombsquadgame.com/accountquery?id=' + account_id + response = json.loads( + urllib.request.urlopen(urllib.Request(url)).read()) + print('response variable from mystats.py line 183:') + print(response) + name_html = response['name_html']''' + stats[account_id] = {'rank': 0, + 'name_html': str(account_id), + 'scores': 0, + 'total_damage': 0, + 'kills': 0, + 'deaths': 0, + 'games': 0, + 'kd': 0, + 'avg_score': 0, + 'aid': str(account_id)} + # now increment their kills whether they were already there or not + stats[account_id]['kills'] += kill_count + stats[account_id]['deaths'] += self.account_deaths[account_id] + stats[account_id]['scores'] += self.account_scores[account_id] + # also incrementing the games played and adding the id + stats[account_id]['games'] += 1 + stats[account_id]['aid'] = str(account_id) + # dump our stats back to disk + tempppp = None + from datetime import datetime + with open(statsfile, 'w') as f: + f.write(json.dumps(stats)) + # aaand that's it! There IS no step 27! + now = datetime.now() + update_time = now.strftime("%S:%M:%H - %d %b %y") + print(f"Added {str(len(self._account_kills))} account's stats entries. || {str(update_time)}") + refreshStats() \ No newline at end of file