趋势类策略入门-2.ipynb

{
 "cells": [
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "# 趋势策略入门"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "### 关键函数：\n",
    " - 1 交易函数（接收一个交易字典，包含交易价格，交易量，币种，交易方向等）\n",
    " - 2 持仓计算函数（根据仓位情况和风险设定，计算限制交易量以及交易价格）\n",
    " - 3 趋势条件计算函数（根据品种价格情况，判断是否执行交易，交易方向，交易量，以及交易价格）"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 1,
   "metadata": {},
   "outputs": [
    {
     "ename": "NameError",
     "evalue": "name 'PERIOD_M5' is not defined",
     "output_type": "error",
     "traceback": [
      "\u001b[0;31m---------------------------------------------------------------------------\u001b[0m",
      "\u001b[0;31mNameError\u001b[0m                                 Traceback (most recent call last)",
      "\u001b[0;32m<ipython-input-1-b479206ff9fa>\u001b[0m in \u001b[0;36m<module>\u001b[0;34m()\u001b[0m\n\u001b[0;32m----> 1\u001b[0;31m \u001b[1;32mclass\u001b[0m \u001b[0mmid_class\u001b[0m\u001b[1;33m(\u001b[0m\u001b[1;33m)\u001b[0m\u001b[1;33m:\u001b[0m\u001b[1;33m\u001b[0m\u001b[0m\n\u001b[0m\u001b[1;32m      2\u001b[0m     \u001b[1;32mdef\u001b[0m \u001b[0m__init__\u001b[0m\u001b[1;33m(\u001b[0m\u001b[0mself\u001b[0m\u001b[1;33m,\u001b[0m \u001b[0mthis_exchange\u001b[0m\u001b[1;33m)\u001b[0m\u001b[1;33m:\u001b[0m\u001b[1;33m\u001b[0m\u001b[0m\n\u001b[1;32m      3\u001b[0m         '''\n\u001b[1;32m      4\u001b[0m         \u001b[0m初始化数据填充交易所的信息\u001b[0m\u001b[0;31m，\u001b[0m\u001b[0m首次获取价格\u001b[0m\u001b[0;31m，\u001b[0m\u001b[0m首次获取account信息\u001b[0m\u001b[1;33m\u001b[0m\u001b[0m\n\u001b[1;32m      5\u001b[0m         \u001b[0m设定好密钥\u001b[0m\u001b[0;31m…\u001b[0m\u001b[0;31m…\u001b[0m\u001b[1;33m\u001b[0m\u001b[0m\n",
      "\u001b[0;32m<ipython-input-1-b479206ff9fa>\u001b[0m in \u001b[0;36mmid_class\u001b[0;34m()\u001b[0m\n\u001b[1;32m     85\u001b[0m \u001b[1;33m\u001b[0m\u001b[0m\n\u001b[1;32m     86\u001b[0m \u001b[1;33m\u001b[0m\u001b[0m\n\u001b[0;32m---> 87\u001b[0;31m     \u001b[1;32mdef\u001b[0m \u001b[0mget_ohlc_data\u001b[0m\u001b[1;33m(\u001b[0m\u001b[0mself\u001b[0m\u001b[1;33m,\u001b[0m \u001b[0mperiod\u001b[0m \u001b[1;33m=\u001b[0m \u001b[0mPERIOD_M5\u001b[0m\u001b[1;33m)\u001b[0m\u001b[1;33m:\u001b[0m\u001b[1;33m\u001b[0m\u001b[0m\n\u001b[0m\u001b[1;32m     88\u001b[0m         '''\n\u001b[1;32m     89\u001b[0m         \u001b[0m获取K线信息\u001b[0m\u001b[1;33m\u001b[0m\u001b[0m\n",
      "\u001b[0;31mNameError\u001b[0m: name 'PERIOD_M5' is not defined"
     ]
    }
   ],
   "source": [
    "class mid_class():\n",
    "    def __init__(self, this_exchange):\n",
    "        '''\n",
    "        初始化数据填充交易所的信息，首次获取价格，首次获取account信息\n",
    "        设定好密钥……\n",
    "        \n",
    "        Args:\n",
    "            this_exchange: FMZ的交易所结构\n",
    "        \n",
    "        '''\n",
    "        self.init_timestamp = time.time()\n",
    "        self.exchange = this_exchange\n",
    "        self.name = self.exchange.GetName()\n",
    "        self.jyd = self.exchange.GetCurrency()        \n",
    "    \n",
    "    def get_account(self):\n",
    "        '''\n",
    "        获取账户信息\n",
    "        \n",
    "        Returns:\n",
    "            获取信息成功返回True，获取信息失败返回False\n",
    "        '''\n",
    "        self.Balance = '---'\n",
    "        self.Amount = '---'\n",
    "        self.FrozenBalance = '---'\n",
    "        self.FrozenStocks = '---'\n",
    "        \n",
    "        try:\n",
    "            self.account = self.exchange.GetAccount()\n",
    "\n",
    "            self.Balance =  self.account['Balance']\n",
    "            self.Amount = self.account['Stocks']\n",
    "            self.FrozenBalance =  self.account['FrozenBalance']\n",
    "            self.FrozenStocks = self.account['FrozenStocks']\n",
    "            return True\n",
    "        except:\n",
    "            return False\n",
    "    \n",
    "    def get_ticker(self):\n",
    "        '''\n",
    "        获取市价信息\n",
    "        \n",
    "        Returns:\n",
    "            获取信息成功返回True，获取信息失败返回False\n",
    "        '''\n",
    "        self.high = '---'\n",
    "        self.low = '---'\n",
    "        self.Sell =  '---'\n",
    "        self.Buy =  '---'\n",
    "        self.last =  '---'\n",
    "        self.Volume = '---'\n",
    "        \n",
    "        try:\n",
    "            self.ticker = self.exchange.GetTicker()\n",
    "        \n",
    "            self.high = self.ticker['High']\n",
    "            self.low = self.ticker['Low']\n",
    "            self.Sell =  self.ticker['Sell']\n",
    "            self.Buy =  self.ticker['Buy']\n",
    "            self.last =  self.ticker['Last']\n",
    "            self.Volume = self.ticker['Volume']\n",
    "            return True\n",
    "        except:\n",
    "            return False\n",
    "        \n",
    "        \n",
    "    def get_depth(self):\n",
    "        '''\n",
    "        获取深度信息\n",
    "        \n",
    "        Returns:\n",
    "            获取信息成功返回True，获取信息失败返回False\n",
    "        '''\n",
    "        self.Ask = '---'\n",
    "        self.Bids = '---'\n",
    "        \n",
    "        try:\n",
    "            self.Depth = self.exchange.GetDepth()\n",
    "            self.Ask = self.Depth['Asks']\n",
    "            self.Bids = self.Depth ['Bids']\n",
    "            return True\n",
    "        except:\n",
    "            return False\n",
    "        \n",
    "        \n",
    "    \n",
    "    def get_ohlc_data(self, period = PERIOD_M5):\n",
    "        '''\n",
    "        获取K线信息\n",
    "        \n",
    "        Args:\n",
    "            period: K线周期，PERIOD_M1 指1分钟, PERIOD_M5 指5分钟, PERIOD_M15 指15分钟,\n",
    "            PERIOD_M30 指30分钟, PERIOD_H1 指1小时, PERIOD_D1 指一天。\n",
    "        '''\n",
    "        self.ohlc_data = exchange.GetRecords(period)\n",
    "        \n",
    "        \n",
    "    \n",
    "    def create_order(self, order_type, price, amount):\n",
    "        '''\n",
    "        post一个挂单信息\n",
    "        \n",
    "        Args:\n",
    "            order_type：挂单类型，'buy'指挂买单，'sell'指挂卖单\n",
    "            price：挂单价格\n",
    "            amount:挂单数量\n",
    "            \n",
    "        Returns:\n",
    "            挂单Id号，可用以取消挂单\n",
    "        '''\n",
    "        if order_type == 'buy':\n",
    "            try:\n",
    "                order_id = self.exchange.Buy( price, amount)\n",
    "            except:\n",
    "                return False\n",
    "            \n",
    "        elif order_type == 'sell':\n",
    "            try:\n",
    "                order_id = self.exchange.Sell( price, amount)\n",
    "            except:\n",
    "                return False\n",
    "        \n",
    "        return order_id\n",
    "    \n",
    "    def get_orders(self):\n",
    "        self.undo_ordes = self.exchange.GetOrders()\n",
    "        return self.undo_ordes\n",
    "    \n",
    "    def cancel_order(self, order_id):\n",
    "        '''\n",
    "        取消一个挂单信息\n",
    "        \n",
    "        Args:\n",
    "            order_id：希望取消的挂单ID号\n",
    "            \n",
    "        Returns:\n",
    "            取消挂单成功返回True，取消挂单失败返回False\n",
    "        '''\n",
    "        return self.exchange.CancelOrder(order_id)\n",
    "        \n",
    "    def refreash_data(self):\n",
    "        '''\n",
    "        刷新信息\n",
    "        \n",
    "        Returns:\n",
    "            刷新信息成功返回 'refreash_data_finish!' 否则返回相应刷新失败的信息提示\n",
    "        '''\n",
    "\n",
    "        if not self.get_account():\n",
    "            return 'false_get_account'\n",
    "        \n",
    "        if not self.get_ticker():\n",
    "            return 'false_get_ticker'\n",
    "        if not self.get_depth():\n",
    "            return 'false_get_depth'\n",
    "        try:\n",
    "            self.get_ohlc_data()\n",
    "        except:\n",
    "            return 'false_get_K_line_info'\n",
    "        \n",
    "        return 'refreash_data_finish!'\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 3,
   "metadata": {
    "collapsed": true
   },
   "outputs": [],
   "source": [
    "class qushi_class():\n",
    "    def __init__(self, mid_class, amount_N, price_N):\n",
    "        '''\n",
    "        设定好初始需要考虑的参数\n",
    "        Args:\n",
    "            mid_class: 所使用的交易所中间层\n",
    "            amount_N：数量小数点限制\n",
    "            price_N：价格小数点限制\n",
    "            \n",
    "        Attributes：\n",
    "            amount_N：数量小数点限制\n",
    "            price_N：价格小数点限制\n",
    "            init_time：初始时间\n",
    "            last_time：上一次执行操作的时间\n",
    "            trade_list:交易请求的id\n",
    "        '''\n",
    "        self.jys = mid_class\n",
    "        \n",
    "        self.init_time = time.time()\n",
    "        self.last_time = time.time()\n",
    "        \n",
    "        self.amount_N = amount_N\n",
    "        self.price_N = price_N\n",
    "        \n",
    "        self.trade_list = []\n",
    "    \n",
    "    def refreash_data(self):\n",
    "        '''\n",
    "        用来从交易所获取最新的价格和数量信息\n",
    "        \n",
    "        Attributes：\n",
    "            B：商品币数量\n",
    "            money：计价币数量\n",
    "            can_buy_B：当前理论可购买商品币数量\n",
    "            Buy_price:当前市场上最近的一单挂单买价\n",
    "            Sell_price：当前市场上最近的一单挂单卖价\n",
    "        '''\n",
    "        \n",
    "        self.jys.refreash_data()\n",
    "        self.B = self.jys.Amount\n",
    "        self.money = self.jys.Balance\n",
    "        self.Buy_price = self.jys.Buy\n",
    "        self.Sell_price = self.jys.Sell\n",
    "        self.can_buy_B = self.money/ self.Sell_price\n",
    "        self.can_buy_B = _N(self.can_buy_B, self.amount_N )\n",
    "        self.can_sell_B = _N(self.B, self.amount_N )\n",
    "        \n",
    "    def make_trade_by_dict(self, trade_dicts):\n",
    "        '''\n",
    "        用来批量完成交易订单\n",
    "        \n",
    "        Attributes：\n",
    "            trade_list:已提交的交易请求的id\n",
    "        '''\n",
    "        for this_trade in trade_dicts:\n",
    "            this_trade_id = self.jys.create_order( this_trade['side'], this_trade['price'] , this_trade['amount'] ) \n",
    "            self.trade_list.append( this_trade_id )\n",
    "    \n",
    "    def condition_chicang(self, hands_num):\n",
    "        '''\n",
    "        根据持仓情况来做交易判定的条件\n",
    "        Args:\n",
    "            hands_num：表示交易一共几手（我们假设当前每次交易不高于一手）\n",
    "            \n",
    "        Attributes：\n",
    "            min_trade_B: 一手最多交易的商品币数量\n",
    "            min_trade_money: 一手最多交易的计价币数量\n",
    "        \n",
    "        '''\n",
    "        self.min_trade_B = (self.can_buy_B + self.B) / hands_num\n",
    "        self.min_buy_B = min(self.min_trade_B, self.can_buy_B)\n",
    "        self.min_sell_B = min(self.min_trade_B, self.B)\n",
    "        \n",
    "        self.min_trade_money = self.min_trade_B* self.jys.Buy\n",
    "\n",
    "\n",
    "    \n",
    "    def condition_qushi(self, change_pct ):\n",
    "        '''\n",
    "        根据市场价格情况来做交易判定的条件\n",
    "        Args:\n",
    "            change_pct：表示价格变化多少来确定一次交易（假设根据之前一段时间均值判断）\n",
    "            \n",
    "        Returns：\n",
    "            min_trade_B: 一手最多交易的商品币数量\n",
    "            min_trade_money: 一手最多交易的计价币数量\n",
    "        \n",
    "        '''\n",
    "        mean_price = sum( [x['Close'] for x in self.jys.ohlc_data[-12*24:]])/(12*24)\n",
    "        do_buy = self.jys.Buy > mean_price* (100.0 + change_pct )/100.0\n",
    "        do_sell = self.jys.Sell < mean_price/ ((100.0 + change_pct )/100.0)\n",
    "        \n",
    "        if do_buy or do_sell:\n",
    "            rt = 'Buy' if do_buy else 'Sell'\n",
    "            return rt\n",
    "        else:\n",
    "            return False\n",
    "    \n",
    "    def make_trade_dicts(self, hands_num, change_pct ):\n",
    "        self.condition_chicang(hands_num)\n",
    "        rt = self.condition_qushi( change_pct )\n",
    "        this_trade_dicts = []\n",
    "        if rt:\n",
    "            if rt == 'Buy':\n",
    "                if self.min_buy_B > 10**-self.amount_N:\n",
    "                    this_trade_dicts.append({\n",
    "                        'side':'buy',\n",
    "                        'price':self.jys.Buy,\n",
    "                        'amount':self.min_buy_B\n",
    "                    })\n",
    "            else:\n",
    "                if self.min_sell_B > 10**-self.amount_N:\n",
    "                    this_trade_dicts.append({\n",
    "                        'side':'sell',\n",
    "                        'price':self.jys.Sell,\n",
    "                        'amount':self.min_sell_B\n",
    "                    })\n",
    "            return this_trade_dicts\n",
    "        else:\n",
    "            return False\n",
    "        "
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 2,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "1e-05"
      ]
     },
     "execution_count": 2,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "1e-5"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 1,
   "metadata": {},
   "outputs": [],
   "source": [
    "def main():\n",
    "    \n",
    "    Set_amount_N = 4\n",
    "    Set_price_N = 4\n",
    "    \n",
    "    hands_num = 20\n",
    "    price_change_percentage = 2\n",
    "    \n",
    "    test_mid = mid_class(exchange)\n",
    "    Log(test_mid.refreash_data())\n",
    "    test_qushi = qushi_class(test_mid , Set_amount_N, Set_price_N)\n",
    "    \n",
    "    while True:\n",
    "        \n",
    "        Sleep(1000)\n",
    "        try:\n",
    "            test_qushi.refreash_data()\n",
    "\n",
    "            now_trade_dicts = test_qushi.make_trade_dicts(hands_num, price_change_percentage)\n",
    "            if now_trade_dicts:\n",
    "                test_qushi.make_trade_by_dict(now_trade_dicts)\n",
    "                now_trade_dicts = False\n",
    "        except:\n",
    "            pass"
   ]
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "Python 3",
   "language": "python",
   "name": "python3"
  },
  "language_info": {
   "codemirror_mode": {
    "name": "ipython",
    "version": 3
   },
   "file_extension": ".py",
   "mimetype": "text/x-python",
   "name": "python",
   "nbconvert_exporter": "python",
   "pygments_lexer": "ipython3",
   "version": "3.7.4"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 2
}