#框架内部

try:
    sys.path.append(os.getcwd())
    print('loading strategy from'+ os.getcwd())
    import user_strategy
    from user_strategy import before_backtest, before_trading, handle_bar, end_trading, end_backtest
except:
    QA_util_log_info(Exception)


def strategy_dec(func, *a, **b):

    def deoc(*a, **b):
        before_trading()
        handle_bar()
        func(*a, **b)
        end_trading()
    return deoc


def backtest_dec(func, *a, **b):

    def inside_backtest(*a, **b):
        before_backtest()
        func(*a, **b)
        end_backtest()
    return inside_backtest


class backtest():

    @backtest_dec
    @strategy_dec
    def exec_bid(self):
        print('from backtest_framework: exec bid market and account')


"""
loading strategy fromC:\Users\yutiansut\quantaxis\test\new_test
import successfully
from strategy: befor_backtest
from strategy: befor_trading
from strategy: handle_bar
from backtest_framework: exec bid market and account
from strategy: end_trading
from strategy: end_bar
"""

print('import successfully')

def before_backtest():
    print('from strategy: befor_backtest')
def before_trading():
    print('from strategy: befor_trading')
    
def handle_bar():
    print('from strategy: handle_bar')


def end_trading():
    print('from strategy: end_trading')

def end_backtest():
    print('from strategy: end_bar')
    

class backtest():
    cash=10000
    def __init__(self,):
        pass
    
    @classmethod
    def before_backtest(backtest,func,*a,**b):
        #yield backtest.cash
        return func(backtest,*a,**b)

    
    @classmethod
    def before_trading(backtest,func,*a,**b):
        #yield backtest.cash
        return func(backtest,*a,**b)



    @classmethod
    def strategy(backtest,func,*a,**b):
        
        return func(backtest,*a,**b)
    @classmethod
    def end_trading(backtest,func,*a,**b):
        #yield backtest.cash
        return func(backtest,*a,**b)
    
    def exec_bid(self,bid):
        print(bid)
        self.cash+=bid
        

from xxx import backtest

@backtest.before_trading
def before_trading(backtest):
    print(backtest.cash)


@backtest.strategy
def handle_bar(backtest):
    print('x')
    print(backtest.cash)
    backtest.exec_bid(backtest,1000)
    print(backtest.cash)


if __name__=='__main__':
    before_trading
    handle_bar