diff --git a/aqua_importer.py b/aqua_importer.py index 4e5b241..d338653 100644 --- a/aqua_importer.py +++ b/aqua_importer.py @@ -753,24 +753,10 @@ class Importer: for user_udeame_row in user_udemae_result: user_udeame = user_udeame_row._asdict() - udemae = { - "rate": user_udeame["rate"], - "maxRate": user_udeame["max_rate"], - "classValue": user_udeame["class_value"], - "maxClassValue": user_udeame["max_class_value"], - "totalWinNum": user_udeame["total_win_num"], - "totalLoseNum": user_udeame["total_lose_num"], - "maxWinNum": user_udeame["max_win_num"], - "maxLoseNum": user_udeame["max_lose_num"], - "winNum": user_udeame["win_num"], - "loseNum": user_udeame["lose_num"], - "npcTotalWinNum": user_udeame["npc_total_win_num"], - "npcTotalLoseNum": user_udeame["npc_total_lose_num"], - "npcMaxWinNum": user_udeame["npc_max_win_num"], - "npcMaxLoseNum": user_udeame["npc_max_lose_num"], - "npcWinNum": user_udeame["npc_win_num"], - "npcLoseNum": user_udeame["npc_lose_num"], - } + user_udeame.pop("id") + user_udeame.pop("user_id") + + udemae = {inflection.camelize(k, False): v for k, v in user_udeame.items()} return ( ratings["recent_rating"], @@ -1100,6 +1086,7 @@ class Importer: f"Imported {version_str} userFriendSeasonRanking: {artemis_user}, {tmp['seasonId']}" ) + def main(): parser = argparse.ArgumentParser(description="AQUA to ARTEMiS") parser.add_argument(