from cryptoquant.trader.constant import Direction, Exchange, Interval, Offset, Status, Product, OptionType, OrderType
import pandas as pd
from cryptoquant.app.data_manage.data_manager import save_data_to_cryptoquant

if __name__ == '__main__':
    df = pd.read_csv('IF9999.csv')
    symbol = 'IF9999'
    save_data_to_cryptoquant(symbol, df, Exchange.CFFEX)
    

from datetime import datetime
from cryptoquant.app.cta_backtester.engine import BacktestingEngine, OptimizationSetting
from cryptoquant.app.cta_strategy.strategies.atr_rsi_strategy import (
    AtrRsiStrategy,
)
#%%
engine = BacktestingEngine()

engine.set_parameters(
    vt_symbol="IF9999.CFFEX",
    interval="1m",
    start=datetime(2020, 1, 1),
    end=datetime(2020, 4, 30),
    rate=0.3/10000,
    slippage=0.5,
    size=300,
    pricetick=0.2,
    capital=1_000_0,
)
setting = {}
engine.add_strategy(AtrRsiStrategy,setting)
# 导入数据
engine.load_data()
# 开始回测
engine.run_backtesting()
#计算收益
df = engine.calculate_result()
# 开始统计
engine.calculate_statistics()
# 开始画图
engine.show_chart()

from cryptoquant.config.config import ok_api_key, ok_seceret_key, ok_passphrase,binance_api_key,binance_secret_key
from cryptoquant import get_exchange

"""
Attention:
to run this code file , your python may need to be python3.9. It can be run by my environment
of  python 3.9. Hope you can run it successfully. Many Thanks 
"""

if __name__ == "__main__":
    setting ={
        'symbol':"EOS/USDT",
        'api_key':binance_api_key,
        'secret':binance_secret_key,
        'base_asset':'EOS',
        'quote_asset':'USDT',
        'sleep_time':5,
        'time_frame':'5m'
    }

    apikey = binance_api_key
    secret = binance_secret_key
    symbol = "EOS/USDT"
    time_frame = '5m'
    strategy_name = 'apidemo'

    exchange = get_exchange(symbol, apikey, secret, time_frame, strategy_name, setting)

    print('GEt Trades', exchange.GetTrades())
    print('GEt Ticker',exchange.GetTicker())
    print('GEt Depth',exchange.GetDepth())
    print('GetAccount',exchange.GetAccount())
    print('获取K线',exchange.GetKline(time_frame))
    print('get Orders',exchange.GetOrders())
    print('get open Orders',exchange.GetOpenOrders())

    # 买单
    buy_order = exchange.Buy(Price = 3,Amount = 4)
    print(f"获取订单{exchange.GetOrder(buy_order.id)}")

    # 撤单
    cancel_order = exchange.CancelOrder(buy_order.id)
    print(f"取消订单{cancel_order}")

    # 卖单
    sell_order = exchange.Sell(Price = 5,Amount = 4)
    print(f"获取订单{exchange.GetOrder(buy_order.id)}")

    # 撤单
    cancel_order = exchange.CancelOrder(sell_order.id)
    print(f"取消订单{cancel_order}")

CCXT GateWay Init
GEt Trades [Trade(Id='170714627', Time=1658938662633, Price=1.138, Amount=1933.3, Type=<Direction.ORDER_TYPE_BUY: 0>), 
..                  ...    ...    ...    ...    ...       ...
495 2022-07-28 00:15:00  1.135  1.141  1.134  1.137  133360.3
496 2022-07-28 00:20:00  1.138  1.140  1.137  1.138   22433.0
497 2022-07-28 00:25:00  1.139  1.144  1.138  1.141  100433.5
498 2022-07-28 00:30:00  1.141  1.143  1.140  1.141   42677.0
499 2022-07-28 00:35:00  1.141  1.143  1.141  1.142   27621.5

"""
自动交易简易的流程：  
根据个人的策略情况不同，自行调整
 1 - 更新账户信息
 2 - 获取TICKER
 3 - 获取K线
 4 - 处理K线数据形成交易信号


 5 - 定时运行
    用户可参考代码添加定时运行模块，来完善，如需要完整脚本，可以添加微信 studyquant88 发送cryptoquant 来获取，这个示例应用的代码。
 wechat: studyquant88
"""


from cryptoquant import *
# - 策略导入
from cryptoquant.app.cta_strategy.strategies.double_ma_strategy import (
    DoubleMaStrategy
    )
from cryptoquant.config.strategy_config import ma_strategy_spot_setting
from squtils import sync_current_minute



def refresh_data(exchange, strategy:DoubleMaStrategy,time_frame):
    """
    更新数据
    """
    ticker = exchange.GetTicker()
    # 推送TICKER给策略
    strategy.on_tick_data(ticker)
    # 获取账户信息
    account = exchange.GetAccount()
    # # 推送到策略
    strategy.on_account(account)
    # 获取K线数据
    kline_df = exchange.GetKline(time_frame)
    strategy.on_kline(kline_df)


if __name__ == "__main__":
    # 参数设置
    setting = ma_strategy_spot_setting
    # symbol
    symbol = setting['symbol']
    # 策略
    strategy_name = 'Trend_strategy'
    # 策略周期
    time_frame = setting['time_frame']
    # API SECRET
    secret_key = setting['secret']
    # api_key
    api_key = setting['api_key']

    # 接口实例
    exchange = get_exchange(symbol, api_key, secret_key, time_frame, strategy_name, setting)
    exchange.log('开始运行')
    logger = exchange.logger

    # 策略实例
    strategy = DoubleMaStrategy(exchange, strategy_name, symbol, setting)
    strategy.trading = True  # 打开实盘交易
    strategy.fixed_order_amount = setting['order_amount']

    # 更新数据
    refresh_data(exchange,strategy,time_frame)

    """
    # todo
    后期可以添加定时运行模块，来完善，如需要完整脚本，可以添加微信 studyquant88 发送cryptoquant 来获取，这个示例应用的代码。
    """
    while True:
        refresh_data(exchange, strategy, time_frame)
        # 简易的时间驱动！ 用户可自行实现运行周期
        time.sleep(30)