mirror of
https://github.com/FAUSheppy/skillbird
synced 2025-12-06 06:51:34 +01:00
112 lines
3.5 KiB
Python
112 lines
3.5 KiB
Python
import Player
|
|
import TrueSkillWrapper as TS
|
|
from fuzzywuzzy import fuzz
|
|
from datetime import datetime, timedelta
|
|
|
|
known_players = dict()
|
|
|
|
# care for cpu load #
|
|
player_ranks = dict()
|
|
last_rank_update = datetime.now()
|
|
|
|
def load_save():
|
|
with open("score.log") as f:
|
|
for l in f:
|
|
p = Player.PlayerFormDatabase(l)
|
|
known_players[p.steamid] = p
|
|
|
|
def save_to_file(fname="score.log"):
|
|
with open(fname,"w") as f:
|
|
for p in known_players.values():
|
|
f.write(p.toCSV()+'\n')
|
|
|
|
def get_player_rank(p):
|
|
global player_ranks
|
|
try:
|
|
return str(player_ranks[p])
|
|
except KeyError:
|
|
return "N/A"
|
|
|
|
|
|
def dumpRatings(top=0, forceMeanSort=False, enforceWhitelist=None):
|
|
global known_players
|
|
ret = ""
|
|
if forceMeanSort:
|
|
sort = sorted(known_players.values(),key=lambda x: x.rating.mu,reverse=True)
|
|
else:
|
|
sort = sorted(known_players.values(),key=lambda x: TS.get_env().expose(x.rating),reverse=True)
|
|
if enforceWhitelist:
|
|
sort = list(filter(lambda x: x.name in enforceWhitelist, sort))
|
|
tmp = ["{} {} mean: {} var: {} WinRatio: {}% ({} Games)".format(x.get_name().ljust(30),\
|
|
str(int(TS.get_env().expose(x.rating))).rjust(5),str(int(x.rating.mu)).rjust(4),\
|
|
str(int(x.rating.sigma)).rjust(4),x.winratio().rjust(4),str(x.games).rjust(3))\
|
|
for x in sort]
|
|
if top != 0:
|
|
tmp = tmp[:top]
|
|
count = 0
|
|
for s in tmp:
|
|
count += 1
|
|
ret += ("Rank: "+str(count).rjust(4) +" " + s + "\n")
|
|
return ret
|
|
|
|
def sync_from_database(players):
|
|
for p in players:
|
|
#print("BKnP: {}".format(p))
|
|
if p in known_players:
|
|
#print("KnP: {}".format(p))
|
|
p.rating = known_players[p].rating
|
|
if type(players) == dict:
|
|
players[p] = p.rating
|
|
else:
|
|
known_players.update({Player.DummyPlayer(p.steamid, p.name):Player.PlayerForDatabase(None,None,None,player=p)})
|
|
|
|
def sync_to_database(players, win):
|
|
for p in players:
|
|
known_players[p].rating = players[p]
|
|
if win:
|
|
known_players[p].wins += 1
|
|
known_players[p].games += 1
|
|
updatePlayerRanks()
|
|
|
|
|
|
def updatePlayerRanks(force=False):
|
|
global last_rank_update
|
|
global player_ranks
|
|
|
|
if force or last_rank_update - datetime.now() > timedelta(seconds=240):
|
|
last_rank_update = datetime.now()
|
|
s = sorted(known_players.values(),key=lambda x: TS.get_env().expose(x.rating),reverse=True)
|
|
rank = 1
|
|
for p in s:
|
|
if p in player_ranks:
|
|
player_ranks[p] = rank
|
|
else:
|
|
player_ranks.update({p:rank})
|
|
rank += 1
|
|
|
|
def _get_known_players():
|
|
return known_players
|
|
|
|
|
|
def save_event(event):
|
|
return
|
|
|
|
def save_psql():
|
|
f=open("pass.secret",'r')
|
|
pw=f.readline().strip();
|
|
f.close()
|
|
PSQL.save("insurgency", "insurgencyUser", "localhost", pw, known_players)
|
|
|
|
def fuzzy_find_player(name):
|
|
ret = ""
|
|
tup_list = []
|
|
TS.lock()
|
|
try:
|
|
for p in known_players.values():
|
|
sim = fuzz.token_set_ratio(name.lower(),p.name.lower())
|
|
tup_list += [(sim,p)]
|
|
finally:
|
|
TS.unlock()
|
|
tmp = sorted(tup_list, key=lambda x: x[0], reverse=True)
|
|
return list(filter(lambda x: x[0] > 80, tmp))
|