import re
import os
import time
import math
import sqlite3
from datetime import datetime

from hts.HTS import HTS
from hts.OrderType import OrderType

from hts.BuySellChecker import BuySellChecker
from hts.OrderChecker import OrderChecker
from stock.util.TelegramBot import TelegramBot
from stock.analysis.StockStatus import StockStatus

class HTS_Stocks (HTS):

    RESOURCE_PATH = None
    orderChecker = None
    buySellChecker = None
    labelChecker = None
    bot = None
    stockStatus = None
    analyzed_day = None
    MAX_BUY_PRICE = None

    conn_stock = None
    cursor_stock = None

    def __init__(self, RESOURCE_PATH):
        super().__init__(RESOURCE_PATH)
        self.bot = TelegramBot()

        self.RESOURCE_PATH = RESOURCE_PATH
        self.stockStatus = StockStatus(RESOURCE_PATH)

        self.buySellChecker = BuySellChecker()
        self.orderChecker = OrderChecker(self.RESOURCE_PATH, "STOCK")

        self.analyzed_day = 120
        self.MAX_BUY_PRICE = 300000

        return

    def connect2StockDB(self):

        self.conn_stock = sqlite3.connect(os.path.join(self.RESOURCE_PATH, "resources/stock.db"))
        self.cursor_stock = self.conn_stock.cursor()

        return

    def disconnectStockDB(self):

        self.cursor_stock.close()
        self.conn_stock.close()
        return

    def getCompanyInfo(self, today_stock):

        self.cursor_stock.execute('SELECT distinct code, name FROM stock order by code')
        all_stocks = self.cursor_stock.fetchall()

        valid_company = dict()
        self.cursor_stock.execute('select CODE, NAME, max(ymd) as ymd from fnguide where type != "E" and sales > 0 group by 1 order by sales desc')
        items = self.cursor_stock.fetchall()
        for i, item in enumerate(items):
            if item[0] in today_stock:
                valid_company[item[0]] = i

        return all_stocks, valid_company

    def buyRealTime(self, today, n = 200):
        print ("START...")
        THIS_TIME = datetime.now()

        today_stock = self.get_today_stock()
        all_stocks, valid_company = self.getCompanyInfo(today_stock)

        while datetime.strptime(today + " 070000", '%Y%m%d %H%M%S') < THIS_TIME < datetime.strptime(today + " 153100", '%Y%m%d %H%M%S'):

            # 1515 까지만 매수를 시도한다.
            if datetime.strptime(today + " 090000", '%Y%m%d %H%M%S') < THIS_TIME < datetime.strptime(today + " 151500", '%Y%m%d %H%M%S'):

                # 매도를 체크한다.
                #self.sellStocks()

                for idx, item in enumerate(all_stocks):
                    if THIS_TIME < datetime.strptime(today + " 090000", '%Y%m%d %H%M%S') or datetime.strptime(today + " 151500", '%Y%m%d %H%M%S') < THIS_TIME:
                        break

                    time.sleep(0.1)

                    stock_code = item[0]
                    stock_name = item[1]
                    if (stock_code not in valid_company):
                        continue

                    print("%5d: %8s, %-50s" % (idx, stock_code, stock_name))

                    stock = self.stockStatus.fetchLastData(self.cursor_stock, stock_code, n)
                    try:
                        self.getRealTime_DailyCheck(today, stock_code, stock)
                        data = self.stockStatus.analyze(stock, self.analyzed_day)
                    except:
                        print("#ERROR:", stock_code, stock_name)
                        continue

                    # 분석일 데이터만 활용한다 (이전 데이터는 제거)
                    data.drop(data.index[:len(data) - self.analyzed_day], inplace=True)

                    bs_buy_price = data["close"][len(data["close"]) - 1]

                    # 미체결 기록을 가져와서 10분 이상 된 매수 주문을 취소 한다.
                    ORDER_LIST = self.requestOrderList()
                    orderListToCancel = self.orderChecker.cancel(today, "A" + stock_code, ORDER_LIST, mins=10)
                    if len(orderListToCancel) > 0:
                        self.cancelOrderList(orderListToCancel)

                    # 다음 조건이면 매수한다.
                    if bs_buy_price > 1000:

                        if not self.orderChecker.exist(today, "A" + stock_code, hours=5):
                            buy_count = int(self.MAX_BUY_PRICE / bs_buy_price)

                            if buy_count > 0:

                                # 매수를 주문한다.
                                orderNum = self.requestOrder(OrderType.buy, stock_code, buy_count, bs_buy_price)
                                self.orderChecker.buy(today, "A" + stock_code, buy_count, bs_buy_price, orderNum)

                                # bot에 메시지를 보냄
                                self.bot.post(stock_code, stock_name, "BUY", bs_buy_price, buy_count)

                                # 로그 출력
                                print("BUY", THIS_TIME.strftime('%Y%m%d %H%M%S'), orderNum, stock_code, stock_name, bs_buy_price, buy_count)

            time.sleep(3600)
            THIS_TIME = datetime.now()

        return

    def updteTodayStock(self, db_filename, stock_code, today_str):
        bsLine, data = self.labelChecker.makeCandidate(stock_code, today_str)
        self.labelChecker.updateLabel(db_filename, stock_code, bsLine, data, today_str)
        return

    def get_today_stock(self):
        today_stock = set()
        today_stock.add("311320")
        today_stock.add("311390")
        today_stock.add("290380")
        today_stock.add("333430")
        today_stock.add("200350")
        today_stock.add("009410")
        today_stock.add("053950")
        today_stock.add("027710")
        today_stock.add("246690")
        today_stock.add("096530")
        today_stock.add("201490")
        today_stock.add("008420")
        today_stock.add("302550")
        today_stock.add("039240")
        today_stock.add("377330")
        today_stock.add("025550")
        today_stock.add("006340")
        today_stock.add("001440")
        return today_stock

if __name__ == "__main__":
    today = datetime.today()

    PROJECT_HOME = os.getcwd()
    RESOURCE_PATH = os.path.join(PROJECT_HOME, "resources")

    hts = HTS_Stocks(RESOURCE_PATH)
    hts.connect2DB("hts.db")
    hts.connect2StockDB()

    today_str = today.strftime('%Y%m%d')
    hts.buyRealTime(today_str)

    hts.disconnectStockDB()
    hts.disconnect()

    print ("done...")