import average_growth
import json
import common
import choose_api
from pathlib import Path


def over_time(detail, board):
    """
    判斷當(dāng)日持倉(cāng)的所有基金的合計(jì)漲幅是否超過滬深300
    :param code_list: list 基金代碼列表
    :return: json文件，格式如下，HS300漲幅、持倉(cāng)合計(jì)漲幅、持倉(cāng)漲幅是否超過滬深300
        {
            "2020-12-31": {
                "HS300": "1.91",
                "my_position": "1.35",
                "over_take": false
            },
            "2021-01-04": {
                "HS300": "1.08",
                "my_position": "1.33",
                "over_take": true
            }
        }
        ......
    """
    # 獲取日期
    try:
        date = detail[0]['netWorthDate']
    except:
        date = detail[1]['netWorthDate']
    hs300 = ''
    for i in board:
        if i['name'] == '滬深300':
            hs300 = i['changePercent']

    # 判斷文件是否存在，不存在則創(chuàng)建
    json_file_name = 'file/bj.json'
    file = Path(json_file_name)
    file.touch(exist_ok=True)

    # 此時(shí)更新的準(zhǔn)確凈值漲幅的平均值
    avg = average_growth.average_growth(detail)['average_dayGrowth']
    print('hs300:', hs300)
    print('avg:', avg)
    # 寫入文件
    with open("file/bj.json", 'r+') as f:
        try:
            data = json.load(f)
            f.seek(0, 0)
            f.truncate()
        except Exception as e:
            print(e)
            data = {}
        finally:
            print(f'獲取數(shù)據(jù): {data}')
            data[date] = {
                'HS300': hs300,
                'my_position': str(avg),
                'over_take': True if float(avg) > float(hs300) else False
            }
            print(f'更新數(shù)據(jù): {date}:{data[date]}')
            f.write(json.dumps(data, sort_keys=True, indent=4, separators=(',', ': ')))
            print(f'{json_file_name} 已更新!')


if __name__ == '__main__':
    code_list = common.get_codelist('product')
    data = choose_api.choose_api(code_list)
    detail = data['detail']
    board = data['board']
    over_time(detail, board)

import requests


def nasdaq():
    """
    爬取sina財(cái)經(jīng)nasdaq基礎(chǔ)數(shù)據(jù)
    :return: 構(gòu)建的標(biāo)準(zhǔn)返回格式，只包含當(dāng)日的數(shù)據(jù)，無歷史數(shù)據(jù)
    """
    url = "http://hq.sinajs.cn/?rn=1609213839262list=gb_$ndx"
    r = requests.get(url)
    response = r.text
    if r.status_code == 200:
        data = response.split('=')[1].split(',')
        nasdaq_data = {'name': data[0].strip('"'), 'code': '040046', 'price': data[1], 'priceChange': data[4], 'expectGrowth': data[2], 'dayGrowth': data[2], 'lastWeekGrowth': '-', 'lastMonthGrowth': '-', 'lastThreeMonthsGrowth': '-', 'date': ''}
        return nasdaq_data
    else:
        print(f'nasdaq return error: \n {response}')


if __name__ == '__main__':
    nasdaq_data = nasdaq()
    for k, v in nasdaq_data.items():
        print(f'{k}: {v}')