#coding=utf-8
import os

import getGlobalVar
import config

g_order_id = 0

def get_orderid():
    global  g_order_id
    g_order_id += 1
    return g_order_id

def set_orderid():
    global g_order_id
    if os.path.exists('./output/maxorderid.txt') and os.path.getsize('./output/maxorderid.txt') > 0:
        f = open('./output/maxorderid.txt', 'w')
        try:
            g_order_id = f.read()
        finally:
            f.close()

def save_orderid():
    global g_order_id
    f = open('./output/maxorderid.txt', 'w')
    f.write(g_order_id)
    f.flush()
    f.close()

def setprogressbar(ret):
    window = getGlobalVar.getWindow()
    if window:
        cur = window.progressBar.value()
        if ret > cur:
            window.progressBar.setValue(ret)

        if ret < window.progressBar.maximum():
            window.labelret.setText('计算中。。。')
        else:
            window.labelret.setText('计算完毕！！！')

def cls_progressbar():
    window = getGlobalVar.getWindow()
    if window:
        window.progressBar.setValue(0)
        window.labelret.setText('')

def init_succ_var():
    meettotalCount = 0
    meettotal = {day:{'totalrate':0, 'successcount':0} for day in config.ZF_DAY}
    return meettotalCount, meettotal

def compute_succ_zf(df, target, meettotalCount, meettotal):
    meettotalCount += 1
    taggetclose = df.iloc[target]['close']
    for day in config.ZF_DAY:
        if (target - day) > -1:
            meettotal[day]['totalrate'] += (df.iloc[target-day]['close']-taggetclose)/taggetclose*100
            if df.iloc[target-day]['close'] > taggetclose:
                meettotal[day]['successcount'] += 1
    return meettotalCount