detailed-cod-tracker/get_cod_stats.py

380 lines
16 KiB
Python

import json
import os
import argparse
from cod_api import API, platforms
import asyncio
# Prevent Async error from showing
if os.name == 'nt':
asyncio.set_event_loop_policy(asyncio.WindowsSelectorEventLoopPolicy())
replacements = {
# Gamemodes
"dom": "Domination",
"hc_dom": "Hardcore Domination",
"war": "Team Deathmatch",
"hc_war": "Hardcore Team Deathmatch",
"hq": "Headquarters",
"hc_hq": "Hardcore Headquarters",
"conf": "Kill Confirmed",
"hc_conf": "Hardcore Kill Confirmed",
"koth": "Hardpoint",
"koth_hc": "Hardcore Hardpoint",
"sd": "Search and Destroy",
"hc_sd": "Hardcore Search and Destroy",
"cyber": "Cyber Attack",
"hc_cyber": "Hardcore Cyber Attack",
"grnd": "Grind",
"arm": "Ground War",
"infect": "Infected",
"gun": "Gun Game",
"arena": "Gunfight",
"br": "Battle Royale (Warzone)",
"br_dmz": "Plunder",
"br_all": "Battle Royale (Warzone & Plunder)",
# Weapons
"weapon_assault_rifle": "Assault Rifles",
"weapon_shotgun": "Shotguns",
"weapon_marksman": "Marksman Rifles",
"weapon_sniper": "Snipers",
"tacticals": "Tactical Equipment",
"lethals": "Lethal Equipment",
"weapon_lmg": "LMGs",
"weapon_launcher": "Launchers",
"supers": "Field Upgrades",
"weapon_pistol": "Pistols",
"weapon_other": "Primary Melee",
"weapon_smg": "SMGs",
"weapon_melee": "Melee",
"scorestreakData": "Scorestreaks",
"lethalScorestreakData": "Lethal Scorestreaks",
"supportScorestreakData": "Support Scorestreaks",
# Guns
## Assault Rifles
"iw8_ar_tango21": "RAM-7",
"iw8_ar_mike4": "M4A1",
"iw8_ar_valpha": "AS VAL",
"iw8_ar_falpha": "FR 5.56",
"iw8_ar_mcharlie": "M13",
"iw8_ar_akilo47": "AK-47",
"iw8_ar_asierra12": "Oden",
"iw8_ar_galima": "CR-56 AMAX",
"iw8_ar_sierra552": "Grau 5.56",
"iw8_ar_falima": "FAL",
"iw8_ar_anovember94": "AN-94",
"iw8_ar_kilo433": "Kilo 141",
"iw8_ar_scharlie": "FN Scar 17",
"iw8_sh_mike26": "VLK Rogue",
## Shotguns
"iw8_sh_charlie725": "725",
"iw8_sh_oscar12": "Origin 12 Shotgun",
"iw8_sh_aalpha12": "JAK-12",
"iw8_sh_romeo870": "Model 680",
"iw8_sh_dpapa12": "R9-0 Shotgun",
## Marksman Rifles
"iw8_sn_sbeta": "MK2 Carbine",
"iw8_sn_crossbow": "Crossbow",
"iw8_sn_romeo700": "SP-R 208",
"iw8_sn_kilo98": "Kar98k",
"iw8_sn_mike14": "EBR-14",
"iw8_sn_sksierra": "SKS",
## Sniper Rifles
"iw8_sn_alpha50": "AX-50",
"iw8_sn_hdromeo": "HDR",
"iw8_sn_delta": "Dragunov",
"iw8_sn_xmike109": "Rytec AMR",
## Tactical Equipment
"equip_gas_grenade": "Gas Grenade",
"equip_snapshot_grenade": "Snapshot Grenade",
"equip_decoy": "Decoy Grenade",
"equip_smoke": "Smoke Grenade",
"equip_concussion": "Concussion Grenade",
"equip_hb_sensor": "Heartbeat Sensor",
"equip_flash": "Flash Grenade",
"equip_adrenaline": "Stim",
## Lethal Equipment
"equip_frag": "Frag Grenade",
"equip_thermite": "Thermite",
"equip_semtex": "Semtex",
"equip_claymore": "Claymore",
"equip_c4": "C4",
"equip_at_mine": "Proximity Mine",
"equip_throwing_knife": "Throwing Knife",
"equip_molotov": "Mototov Cocktail",
## LMGs
"iw8_lm_kilo121": "M91",
"iw8_lm_mkilo3": "Bruen Mk9",
"iw8_lm_mgolf34": "MG34",
"iw8_lm_lima86": "SA87",
"iw8_lm_pkilo": "PKM",
"iw8_lm_sierrax": "FiNN LMG",
"iw8_lm_mgolf36": "Holger-26",
# "": "", ### RAAL LMG not implemented
## Launchers
"iw8_la_gromeo": "PILA",
"iw8_la_rpapa7": "RPG-7",
"iw8_la_juliet": "JOKR",
"iw8_la_kgolf": "Strela-P",
# "": "", ### Unknown Launcher
## Field Upgrades
"super_emp_drone": "EMP Drone",
"super_trophy": "Trophy System",
"super_ammo_drop": "Munitions Box",
"super_weapon_drop": "Weapon Drop",
"super_fulton": "Cash Deposit Balloon",
"super_armor_drop": "Armor Box",
"super_select": "Field Upgrade Pro (Any)",
"super_tac_insert": "Tactical Insertion",
"super_recon_drone": "Recon Drone",
"super_deadsilence": "Dead Silence",
"super_supply_drop": "Loadout Drop", ### Unsure if this is Loadout Drop
"super_tac_cover": "Deployable Cover",
"super_support_box": "Stopping Power Rounds",
## Pistols
"iw8_pi_cpapa": ".357",
"iw8_pi_mike9": "Renetti",
"iw8_pi_mike1911": "1911",
"iw8_pi_golf21": "X16",
"iw8_pi_decho": ".50 GS",
"iw8_pi_papa320": "M19",
# "": "", ### Sykov not implemented
## Primary Melee
"iw8_me_riotshield": "Riot Shield",
## SMGs
"iw8_sm_mpapa7": "MP7",
"iw8_sm_augolf": "AUG",
"iw8_sm_papa90": "P90",
"iw8_sm_charlie9": "ISO",
"iw8_sm_mpapa5": "MP5",
"iw8_sm_smgolf45": "Striker 45",
"iw8_sm_beta": "PP19 Bizon",
"iw8_sm_victor": "Fennec",
"iw8_sm_uzulu": "Uzi",
# "": "", ### CX9 not implemented
## Melee
"iw8_me_akimboblunt": "Kali Sticks",
"iw8_me_akimboblades": "Dual Kodachis",
"iw8_knife": "Knife",
# Scorestreaks
"precision_airstrike": "Precision Airstrike",
"cruise_predator": "Cruise Missile",
"manual_turret": "Shield Turret",
"white_phosphorus": "White Phosphorus",
"hover_jet": "VTOL Jet",
"chopper_gunner": "Chopper Gunner",
"gunship": "Gunship",
"sentry_gun": "Sentry Gun",
"toma_strike": "Cluster Strike",
"nuke": "Nuke",
"juggernaut": "Juggernaut",
"pac_sentry": "Wheelson",
"chopper_support": "Support Helo",
"bradley": "Infantry Assault Vehicle",
"airdrop": "Care Package",
"radar_drone_overwatch": "Personal Radar",
"scrambler_drone_guard": "Counter UAV",
"uav": "UAV",
"airdrop_multiple": "Emergency Airdrop",
"directional_uav": "Advanced UAV",
# Accolades
# "accoladeData": "Accolades",
# "classChanges": "Most classes changed (Evolver)",
# "highestAvgAltitude": "Highest average altitude (High Command)",
# "killsFromBehind": "Most kills from behind (Flanker)",
# "lmgDeaths": "Most LMG deaths (Target Practice)",
# "riotShieldDamageAbsorbed": "Most damage absorbed with Riot Shield (Guardian)",
# "flashbangHits": "Most Flashbang hits (Blinder)",
# "meleeKills": "Most Melee kills (Brawler)",
# "tagsLargestBank": "Largest bank (Bank Account)",
# "shotgunKills": "Most Shotgun kills (Buckshot)",
# "sniperDeaths": "Most Sniper deaths (Zeroed In)",
# "timeProne": "Most time spent Prone (Grassy Knoll)",
# "killstreakWhitePhosphorousKillsAssists": "Most kills and assists with White Phosphorus (Burnout)",
# "shortestLife": "Shortest life (Terminal)",
# "deathsFromBehind": "Most deaths from behind (Blindsided)",
# "higherRankedKills": "Most kills on higher ranked scoreboard players (Upriser)",
# "mostAssists": "Most assists (Wingman)",
# "leastKills": "Fewest kills (The Fearful)",
# "tagsDenied": "Denied the most tags (Denied)",
# "killstreakWheelsonKills": "Most Wheelson kills",
# "sniperHeadshots": "Most Sniper headshots (Dead Aim)",
# "killstreakJuggernautKills": "Most Juggernaut kills (Heavy Metal)",
# "smokesUsed": "Most Smoke Grenades used (Chimney)",
# "avengerKills": "Most avenger kills (Avenger)",
# "decoyHits": "Most Decoy Grenade hits (Made You Look)",
# "killstreakCarePackageUsed": "Most Care Packages called in (Helping Hand)",
# "molotovKills": "Most Molotov kills (Arsonist)",
# "gasHits": "Most Gas Grenade hits (Gaseous)",
# "comebackKills": "Most comebacks (Rally)",
# "lmgHeadshots": "Most LMG headshots (LMG Expert)",
# "smgDeaths": "Most SMG deaths (Run and Gunned)",
# "carrierKills": "Most kills as carrier (Carrier)",
# "deployableCoverUsed": "Most Deployable Covers used (Combat Engineer)",
# "thermiteKills": "Most Thermite kills (Red Iron)",
# "arKills": "Most assault rifle kills (AR Specialist)",
# "c4Kills": "Most C4 kills (Handle With Care)",
# "suicides": "Most suicides (Accident Prone)",
# "clutch": "Most kills as the last alive (Clutched)",
# "survivorKills": "Most kills as survivor (Survivalist)",
# "killstreakGunshipKills": "Most Gunship kills (Death From Above)",
# "timeSpentAsPassenger": "Most time spent as a passenger (Navigator)",
# "returns": "Most flags returned (Flag Returner)",
# "smgHeadshots": "Most SMG headshots (SMG Expert)",
# "launcherDeaths": "Most launcher deaths (Fubar)",
# "oneShotOneKills": "Most one shot kills (One Shot Kill)",
# "ammoBoxUsed": "Most Munitions Boxes used (Provider)",
# #"spawnSelectSquad": "",
# "weaponPickups": "Most picked up weapons (Loaner)",
# "pointBlankKills": "Most point blank kills (Personal Space)",
# "tagsCaptured": "Collected the most tags (Confirmed Kills)",
# "killstreakGroundKills": "Most ground based killstreak kills (Ground Control)",
# "distanceTraveledInVehicle": "Longest distance travelled in a vehicle (Cross Country)",
# "longestLife": "Longest life (Lifer)",
# "stunHits": "Most Stun Grenade hits (Stunner)",
# "spawnSelectFlag": "Most FOB Spawns (Objective Focused)", # Unsure
# "shotgunHeadshots": "Most Shotgun headshots (Boomstick)",
# "bombDefused": "Most defuses (Defuser)",
# "snapshotHits": "Most Snapshot Grenade hits (Photographer)",
# "noKillsWithDeath": "No kills with at least 1 death (Participant)",
# "killstreakAUAVAssists": "Most Advanced UAV assists (Target Rich Environment)",
# "killstreakPersonalUAVKills": "Most kills with a Personal Radar active (Nothing Personal)",
# "tacticalInsertionSpawns": "Most Tactical Insertions used (Revenant)",
# "launcherKills": "Most Launcher kills (Explosive)",
# "spawnSelectVehicle": "Most vehicle spawns (Oscar Mike)",
# "mostKillsLeastDeaths": "Most kills and fewest deaths (MVP)",
# "mostKills": "Most kills (The Feared)",
# "defends": "Most defend kills (Defense)",
# "timeSpentAsDriver": "Most time spent driving (Driver)",
# "": "" # WIP - Still adding more
}
# Initiating the API class
api = API()
COOKIE_FILE = 'cookie.txt'
def get_and_save_data(player_name=None):
# Create the stats directory if it doesn't exist
DIR_NAME = 'stats'
if not os.path.exists(DIR_NAME):
os.makedirs(DIR_NAME)
# Check if cookie file exists
if os.path.exists(COOKIE_FILE):
with open(COOKIE_FILE, 'r') as f:
api_key = f.read().strip()
else:
api_key = input("Please enter your ACT_SSO_COOKIE: ")
with open(COOKIE_FILE, 'w') as f:
f.write(api_key)
# If player_name is not provided via command line, get it from user input
if not player_name:
player_name = input("Please enter the player's username (with #1234567): ")
# Login with sso token
api.login(api_key)
# Retrieve data from API
player_stats = api.ModernWarfare.fullData(platforms.Activision, player_name)
match_info = api.ModernWarfare.combatHistory(platforms.Activision, player_name)
season_loot = api.ModernWarfare.seasonLoot(platforms.Activision, player_name)
map_list = api.ModernWarfare.mapList(platforms.Activision)
identities = api.Me.loggedInIdentities()
# Save results to a JSON file inside the stats directory
with open(os.path.join(DIR_NAME, 'stats.json'), 'w') as json_file:
json.dump(player_stats, json_file, indent=4)
with open(os.path.join(DIR_NAME, 'match_info.json'), 'w') as json_file:
json.dump(match_info, json_file, indent=4)
with open(os.path.join(DIR_NAME, 'season_loot.json'), 'w') as json_file:
json.dump(season_loot, json_file, indent=4)
with open(os.path.join(DIR_NAME, 'map_list.json'), 'w') as json_file:
json.dump(map_list, json_file, indent=4)
with open(os.path.join(DIR_NAME, 'identities.json'), 'w') as json_file:
json.dump(identities, json_file, indent=4)
def recursive_key_replace(obj, replacements):
if isinstance(obj, dict):
new_obj = {}
for key, value in obj.items():
new_key = replacements.get(key, key)
if isinstance(value, str):
new_value = replacements.get(value, value)
new_obj[new_key] = recursive_key_replace(new_value, replacements)
else:
new_obj[new_key] = recursive_key_replace(value, replacements)
return new_obj
elif isinstance(obj, list):
return [recursive_key_replace(item, replacements) for item in obj]
else:
return replacements.get(obj, obj) if isinstance(obj, str) else obj
def sort_data(data):
if isinstance(data, dict):
for key, value in data.items():
if key == "mode":
data[key] = dict(sorted(value.items(), key=lambda item: item[1]['properties']['timePlayed'], reverse=True))
elif key in ["Assault Rifles", "Shotguns", "Marksman Rifles", "Snipers", "LMGs", "Launchers", "Pistols", "SMGs", "Melee"]:
data[key] = dict(sorted(value.items(), key=lambda item: item[1]['properties']['kills'], reverse=True))
elif key in ["Field Upgrades"]:
data[key] = dict(sorted(value.items(), key=lambda item: item[1]['properties']['uses'], reverse=True))
elif key in ["Tactical Equipment", "Lethal Equipment"]:
data[key] = dict(sorted(value.items(), key=lambda item: item[1]['properties']['uses'], reverse=True))
elif key == "Scorestreaks":
for subcategory, scorestreaks in value.items():
data[key][subcategory] = dict(sorted(scorestreaks.items(), key=lambda item: item[1]['properties']['awardedCount'], reverse=True))
elif key == "Accolades":
if 'properties' in value:
data[key]['properties'] = dict(sorted(value['properties'].items(), key=lambda item: item[1], reverse=True))
else:
# Recursive call to handle nested dictionaries
data[key] = sort_data(value)
return data
def beautify_data():
file_path = "stats.json"
with open(file_path, 'r') as file:
data = json.load(file)
data = recursive_key_replace(data, replacements)
data = sort_data(data)
with open(file_path, 'w') as file:
json.dump(data, file, indent=4)
print(f"Keys sorted and replaced in {file_path}!")
def beautify_match_data():
file_path = "match_info.json"
with open(file_path, 'r') as file:
data = json.load(file)
data = recursive_key_replace(data, replacements)
with open(file_path, 'w') as file:
json.dump(data, file, indent=4)
print(f"Keys replaced in {file_path}!")
def main():
# Define the block of quote text to display in the help command
help_text = """
Obtaining your ACT_SSO_COOKIE
- Go to https://www.callofduty.com and login with your account
- Once logged in, press F12 for your browsers developer tools. Then go to Application --> Storage --> Cookies --> https://www.callofduty.com and find ACT_SSO_COOKIE
- Enter the value when prompted
"""
parser = argparse.ArgumentParser(description="COD Data Tool", epilog=help_text, formatter_class=argparse.RawDescriptionHelpFormatter)
# Add arguments for your commands
parser.add_argument("--replace-data", action="store_true", help="Beautify the data in stats.json")
parser.add_argument("--replace-match-data", action="store_true", help="Beautify the match data in match_info.json")
parser.add_argument("--player-name", type=str, help="Player's username (with #1234567)")
args = parser.parse_args()
if args.replace_data:
beautify_data()
elif args.replace_match_data:
beautify_match_data()
else:
get_and_save_data(args.player_name)
if __name__ == "__main__":
main()