wanzb_v3_api/src/tools/update_initfund.py

#coding:utf-8
import os
import time
import datetime
import sys
import django
from django.core.cache import cache
from django.db import connection

sys.path.append('/mnt/wzbapi/src')
os.environ['DJANGO_SETTINGS_MODULE'] = 'settings'
django.setup()

import common.models as cm

def update_initfund():
    """
    """
    prset = cm.PlayerRecord.objects.filter(match_id=10,stock_date="2022-04-26")
    for index,pr in enumerate(prset):
        print(pr.id)
        pr.init_fund = pr.today_fund
        pr.today_income = 0.0
        pr.total_income = 0.0
        pr.yesterday_fund = pr.today_fund
        pr.save()
        player_id = pr.player_id
        cm.Player.objects.filter(id=player_id).update(fund=pr.today_fund)

def update_income():
    """
    """
    prset = cm.PlayerRecord.objects.filter(match_id=10,stock_date="2022-04-27")
    for index,pr in enumerate(prset):
        player_id = pr.player_id
        today_fund = pr.today_fund

        yesterday = cm.PlayerRecord.objects.filter(match_id=10,player_id=player_id,stock_date="2022-04-26")
        if yesterday:
            yesterday = yesterday.first()
            yesterday_fund = yesterday.today_fund
            init_fund = yesterday.init_fund

            today_income = (today_fund - yesterday_fund)/float(yesterday_fund)
            total_income = (today_fund - init_fund)/float(init_fund)


            pr.init_fund = init_fund
            pr.today_income = round(today_income,4)
            pr.total_income = round(total_income,4)
            pr.save()


if __name__ == "__main__":
    print "start update init_fund..."
    st = time.time()
    #update_initfund()
    update_income()
    print "time cost:",time.time()-st