from pathlib import Path
from configs.path_config import DATA_PATH
import nonebot
try:
    import ujson as json
except ModuleNotFoundError:
    import json

nonebot.load_plugins("plugins/shop")

# 修改旧数据

statistics_group_file = Path(DATA_PATH) / "statistics" / "_prefix_count.json"
statistics_user_file = Path(DATA_PATH) / "statistics" / "_prefix_user_count.json"

for file in [statistics_group_file, statistics_user_file]:
    if file.exists():
        with open(file, 'r', encoding='utf8') as f:
            data = json.load(f)
            if not (statistics_group_file.parent / f"{file}.bak").exists():
                with open(f"{file}.bak", 'w', encoding='utf8') as wf:
                    json.dump(data, wf, ensure_ascii=False, indent=4)
            for x in ['total_statistics', 'day_statistics']:
                for key in data[x].keys():
                    num = 0
                    if data[x][key].get('我的道具') is not None:
                        num += data[x][key]['我的道具']
                        del data[x][key]['我的道具']
                    if data[x][key].get('使用道具') is not None:
                        num += data[x][key]['使用道具']
                        del data[x][key]['使用道具']
                    if data[x][key].get('我的金币') is not None:
                        num += data[x][key]['我的金币']
                        del data[x][key]['我的金币']
                    if data[x][key].get('购买') is not None:
                        num += data[x][key]['购买']
                        del data[x][key]['购买']
                    if data[x][key].get('商店') is not None:
                        data[x][key]['商店'] += num
                    else:
                        data[x][key]['商店'] = num
            for x in ['week_statistics', 'month_statistics']:
                for key in data[x].keys():
                    if key == 'total':
                        if data[x][key].get('我的道具') is not None:
                            num += data[x][key]['我的道具']
                            del data[x][key]['我的道具']
                        if data[x][key].get('使用道具') is not None:
                            num += data[x][key]['使用道具']
                            del data[x][key]['使用道具']
                        if data[x][key].get('我的金币') is not None:
                            num += data[x][key]['我的金币']
                            del data[x][key]['我的金币']
                        if data[x][key].get('购买') is not None:
                            num += data[x][key]['购买']
                            del data[x][key]['购买']
                        if data[x][key].get('商店') is not None:
                            data[x][key]['商店'] += num
                        else:
                            data[x][key]['商店'] = num
                    else:
                        for day in data[x][key].keys():
                            num = 0
                            if data[x][key][day].get('我的道具') is not None:
                                num += data[x][key][day]['我的道具']
                                del data[x][key][day]['我的道具']
                            if data[x][key][day].get('使用道具') is not None:
                                num += data[x][key][day]['使用道具']
                                del data[x][key][day]['使用道具']
                            if data[x][key][day].get('我的金币') is not None:
                                num += data[x][key][day]['我的金币']
                                del data[x][key][day]['我的金币']
                            if data[x][key][day].get('购买') is not None:
                                num += data[x][key][day]['购买']
                                del data[x][key][day]['购买']
                            if data[x][key][day].get('商店') is not None:
                                data[x][key][day]['商店'] += num
                            else:
                                data[x][key][day]['商店'] = num
        with open(file, 'w', encoding='utf8') as f:
            json.dump(data, f, ensure_ascii=False, indent=4)