###################################################################
#           Copyright (c) 2016 by TAOS Technologies, Inc.
#                     All rights reserved.
#
#  This file is proprietary and confidential to TAOS Technologies.
#  No part of this file may be reproduced, stored, transmitted,
#  disclosed or used in any form or by any means other than as
#  expressly provided by the written permission from Jianhui Tao
#
###################################################################

# -*- coding: utf-8 -*-

import sys
import taos
from util.log import *
from util.cases import *
from util.sql import *
import numpy as np

# constant define
WAITS = 5 # wait seconds

class TDTestCase:
    def init(self, conn, logSql):
        tdLog.debug("start to execute %s" % __file__)
        tdSql.init(conn.cursor())

        self.rowNum = 10
        self.ts = 1537146000000

    def checkColumnSorted(self, col, order):
        frame = inspect.stack()[1]
        callerModule = inspect.getmodule(frame[0])
        callerFilename = callerModule.__file__

        if col < 0:
            tdLog.exit(
                "%s failed: sql:%s, col:%d is smaller than zero" %
                (callerFilename, tdSql.sql, col))
        if col > tdSql.queryCols:
            tdLog.exit(
                "%s failed: sql:%s, col:%d is larger than queryCols:%d" %
                (callerFilename, tdSql.sql, col, tdSql.queryCols))

        matrix = np.array(tdSql.queryResult)
        list = matrix[:, 0]

        if order == "" or order.upper() == "ASC":
            if all(sorted(list) == list):
                tdLog.info(
                    "sql:%s, column :%d is sorted in accending order as expected" %
                    (tdSql.sql, col))
            else:
                tdLog.exit(
                    "%s failed: sql:%s, col:%d is not sorted in accesnind order" %
                    (callerFilename, tdSql.sql, col))
        elif order.upper() == "DESC":
            if all(sorted(list, reverse=True) == list):
                tdLog.info(
                    "sql:%s, column :%d is sorted in decending order as expected" %
                    (tdSql.sql, col))
            else:
                tdLog.exit(
                    "%s failed: sql:%s, col:%d is not sorted in decending order" %
                    (callerFilename, tdSql.sql, col))
        else:
            tdLog.exit(
                "%s failed: sql:%s, the order provided for col:%d is not correct" %
                (callerFilename, tdSql.sql, col))

    def test_base(self):
        print("======= step 1: create table and insert data =========")
        tdLog.debug(
            ''' create table st(ts timestamp, tbcol1 tinyint, tbcol2 smallint, tbcol3 int, tbcol4 bigint, tbcol5 float, tbcol6 double,
                tbcol7 bool, tbcol8 nchar(20), tbcol9 binary(20), tbcol11 tinyint unsigned, tbcol12 smallint unsigned, tbcol13 int unsigned, tbcol14 bigint unsigned) tags(tagcol1 tinyint, tagcol2 smallint, tagcol3 int, tagcol4 bigint, tagcol5 float,
                tagcol6 double, tagcol7 bool, tagcol8 nchar(20), tagcol9 binary(20),  tagcol11 tinyint unsigned, tagcol12 smallint unsigned, tagcol13 int unsigned, tagcol14 bigint unsigned)''')
        tdSql.execute(
            ''' create table st(ts timestamp, tbcol1 tinyint, tbcol2 smallint, tbcol3 int, tbcol4 bigint, tbcol5 float, tbcol6 double,
                tbcol7 bool, tbcol8 nchar(20), tbcol9 binary(20), tbcol11 tinyint unsigned, tbcol12 smallint unsigned, tbcol13 int unsigned, tbcol14 bigint unsigned) tags(tagcol1 tinyint, tagcol2 smallint, tagcol3 int, tagcol4 bigint, tagcol5 float,
                tagcol6 double, tagcol7 bool, tagcol8 nchar(20), tagcol9 binary(20),  tagcol11 tinyint unsigned, tagcol12 smallint unsigned, tagcol13 int unsigned, tagcol14 bigint unsigned)''')

        for i in range(self.rowNum):
            tdSql.execute("create table st%d using st tags(%d, %d, %d, %d, %f, %f, %d, 'tag%d', '标签%d', %d, %d, %d, %d)" % (
                i + 1, i + 1, i + 1, i + 1, i + 1, 1.1 * (i + 1), 1.23 * (i + 1), (i + 1) % 2, i + 1, i + 1, i + 1, i + 1, i + 1, i + 1))
            for j in range(self.rowNum):
                tdSql.execute("insert into st%d values(%d, %d, %d, %d, %d, %f, %f, %d, 'taosdata%d', '涛思数据%d', %d, %d, %d, %d)" % (
                    i + 1, self.ts + 10 * (i + 1) + j + 1, j + 1, j + 1, j + 1, j + 1, 1.1 * (j + 1), 1.23 * (j + 1), (j + 1) % 2, j + 1, j + 1, i + 1, i + 1, i + 1, i + 1))

        print("======= step 2: verify order for each column =========")
        # sort for timestamp in asc order
        tdSql.query("select * from st order by ts asc")
        self.checkColumnSorted(0, "asc")

        # sort for timestamp in desc order
        tdSql.query("select * from st order by ts desc")
        self.checkColumnSorted(0, "desc")

        print("======= step 2: verify order for special column =========")

        tdSql.query("select tbcol1 from st order by ts desc")

        tdSql.query("select tbcol6 from st order by ts desc")

        for i in range(1, 10):
            tdSql.error("select * from st order by tbcol%d" % i)
            tdSql.error("select * from st order by tbcol%d asc" % i)
            tdSql.error("select * from st order by tbcol%d desc" % i)

            tdSql.query(
                "select avg(tbcol1) from st group by tagcol%d order by tagcol%d" %
                (i, i))
            self.checkColumnSorted(1, "")

            tdSql.query(
                "select avg(tbcol1) from st group by tagcol%d order by tagcol%d asc" %
                (i, i))
            self.checkColumnSorted(1, "asc")

            tdSql.query(
                "select avg(tbcol1) from st group by tagcol%d order by tagcol%d desc" %
                (i, i))
            self.checkColumnSorted(1, "desc")

        # order by rules: https://jira.taosdata.com:18090/pages/viewpage.action?pageId=123455481
        tdSql.error("select tbcol1 from st order by 123")
        tdSql.error("select tbcol1 from st order by tbname")
        tdSql.error("select tbcol1 from st order by tagcol1")
        tdSql.error("select tbcol1 from st order by ''")
        tdSql.error("select top(tbcol1, 12) from st1 order by tbcol1,ts")
        tdSql.error("select top(tbcol1, 12) from st order by tbcol1,ts,tbcol2")
        tdSql.error("select top(tbcol1, 12) from st order by ts, tbcol1")
        #tdSql.error("select top(tbcol1, 2) from st1 group by tbcol1 order by tbcol2")

        fun_list = ['avg','count','twa','sum','stddev','leastsquares','min',
                    'max','first','last','top','bottom','percentile','apercentile',
                    'last_row','diff','spread','distinct']
        key = ['tbol','tagcol']
        for i in range(1,15):
            for k in key:
                for j in fun_list:
                    if j == 'leastsquares':
                        pick_func=j+'('+ k + str(i) +',1,1)'
                    elif j == 'top' or j == 'bottom' : continue
                    elif j == 'percentile' or j == 'apercentile':
                        pick_func=j+'('+ k + str(i) +',1)'
                    else:
                        pick_func=j+'('+ k + str(i) +')'
                    sql = 'select %s from st group by %s order by %s' % (pick_func , k+str(i), k+str(i))
                    tdSql.error(sql)
                    sql = 'select %s from st6 group by %s order by %s ' % (pick_func , k+str(i), k+str(i))
                    tdSql.error(sql)

        tdSql.query("select top(tbcol1, 2) from st1 group by tbcol2 order by tbcol2")
        tdSql.query("select top(tbcol1, 12) from st order by tbcol1, ts")

        tdSql.query("select avg(tbcol1) from st group by tbname order by tbname")
        tdSql.checkData(1, 0, 5.5)
        tdSql.checkData(5, 1, "st6")

        tdSql.query("select top(tbcol1, 2) from st group by tbname order by tbname")
        tdSql.checkData(1, 1, 10)
        tdSql.checkData(2, 2, "st2")

        tdSql.query("select top(tbcol1, 12) from st order by tbcol1")
        tdSql.checkData(1, 1, 9)

        tdSql.error("select top(tbcol1, 12) from st1 order by tbcol1,ts")
        tdSql.error("select top(tbcol1, 12),tbname from st order by tbcol1,tbname")

        tdSql.query("select top(tbcol1, 12) from st group by tbname order by tbname desc")
        tdSql.checkData(1, 2, "st10")
        tdSql.checkData(10, 2, "st9")

        tdSql.query("select top(tbcol1, 2) from st group by tbname order by tbname desc,ts")
        tdSql.checkData(1, 2, "st10")
        tdSql.checkData(10, 2, "st5")
        tdSql.checkData(0, 0, "2018-09-17 09:00:00.109")
        tdSql.checkData(1, 0, "2018-09-17 09:00:00.110")
        tdSql.checkData(2, 0, "2018-09-17 09:00:00.099")

    def stop(self):
        tdSql.close()
        tdLog.success("%s successfully executed" % __file__)

    def create_insert_data(self):
        sql = "create table sb(ts timestamp, i1 int) tags(t_b1 binary(32), t_n1 nchar(64));"
        tdSql.execute(sql)
        for i in range(20):
            value = "%s%s%s" % (chr(64+i),chr(64+i),chr(65+i))
            if i%10 == 5:
              value +=  "_%s" % (chr(66+i))
            if i%10 == 0:
              value +=  "_%s" % (chr(67+i))
            sql = "insert into b%d using sb tags('A%s', 'B%s') values(now, %d);"%(i, value, value, i+1000)
            tdSql.execute(sql)

    def test_groupby_order(self):
        # create and insert data
        print("======= group order step 1: create table and insert data =========")
        self.create_insert_data()

        # do query
        print("======= group order step 2: do query =========")
        # ASC
        sql = "select count(*) from sb group by t_b1 order by t_b1 asc;"
        tdSql.execute(sql)
        tdSql.waitedQuery(sql, 20, WAITS)
        tdSql.checkData(0,  1, "A@@A_C")
        tdSql.checkData(5,  1, "AEEF_G")
        tdSql.checkData(19, 1, "ASST")
        #DESC
        sql = "select count(*) from sb group by t_b1 order by t_b1 desc;"
        tdSql.execute(sql)
        tdSql.waitedQuery(sql, 20, WAITS)
        tdSql.checkData(19,  1, "A@@A_C")
        tdSql.checkData(4, 1, "AOOP_Q")
        tdSql.checkData(0, 1, "ASST")

    def run(self):
        tdSql.prepare()
        self.test_base()
        self.test_groupby_order()

tdCases.addWindows(__file__, TDTestCase())
tdCases.addLinux(__file__, TDTestCase())
