import redis
import time as time
import logging
import json
from .DataStream import span_days, day_stamp, span_days, time_border, calc_interval

import logging
logger = logging.getLogger('PainterBase')

all_colors = ['#000000', '#0000FF', '#FF0000', '#FF00FF', '#00FFFF', '#008080', '#808080',
              '#800080', '#FF8000', '#800000', '#008000', '#8080FF', '#FF8080', '#8000FF', '#808000', '#0080FF', '#80FF80', '#80FF00', '#FF0080',
              '#FF80FF', '#FFFF80', '#000080', '#400000', '#400040', '#4040FF', '#FF4000', '#408000','#FF4080',
              '#404000', '#808040', '#FF8040', '#804000', '#000040', '#408040', '#FF0040', '#804040',
              '#008040', '#800040', '#FF4040', '#804080', '#400080', '#404040', '#004000', '#4080FF', '#0040FF', '#004040', '#4000FF',
              '#404080', '#004080', '#8040FF', '#408080', '#8000C0', '#C000C0', '#40C000', '#40C080',
              '#C000FF', '#FF00C0', '#40C040', '#40C0C0', '#80C040', '#4040C0',  '#4000C0', '#00C040', '#C00040',
              '#8040C0', '#40C0FF', '#C08080', '#C00080', '#8080C0', '#C04080', '#C080C0', '#C04040',
              '#C040C0', '#00C080', '#C080FF', '#C00000',
              '#80C000', '#0040C0', '#C04000', '#00C0FF', '#FF40C0', '#0080C0', '#00C000',
              '#0000C0', '#4080C0', '#C0C000', '#C08000', '#C08040']

class PainterBase(object):
    def __init__(self):
        self._redis_client = None
        pass

    def set_redis(self, rhost, rport):
        pool = redis.ConnectionPool(host=rhost, port=rport)
        self._redis_client = redis.Redis(connection_pool=pool)
        pass

    def redis_client(self):
        return self._redis_client

    def calc_time(self, reader, start_time: int, end_time: int):
        end_time = end_time or int(time.time())
        if start_time is None or start_time > end_time:
            start_time = end_time - 7200

        stime = lambda t: time.strftime('%y-%m-%d %H:%M:%S', time.localtime(t))
        interval = calc_interval(start_time, end_time)
        logger.debug("start_time %d=%s end_time=%s interval=%d", start_time, stime(start_time), stime(end_time), interval)

        days = span_days(start_time, end_time)
        sdays = [stime(day) for day in days]
        logger.debug(sdays)

        return days, start_time, end_time, interval

    def calc_xticks(self,start,end,interval):
        ticks = []
        _start = start
        while True:
            if start < end:
                ticks.append(start - _start)
                if start % interval == 0:
                    start += interval
                else:
                    start += interval - start % interval
            else:
                ticks.append(end - _start)
                break
        return ticks

    def clac_xlables(self, start, end, interval):
        xticks = self.calc_xticks(start, end, interval)
        if len(xticks) == 0:
            return xticks,[]

        min = xticks[0]
        max = xticks[-1]
        diff = max - min
        if diff > 86400 * 30:
            xlables = [time.strftime('%m-%d-%H:%M:%S', time.localtime(d + self._start_time)) for d in xticks]
        elif diff > 86400:
            xlables = [time.strftime('%d-%H:%M:%S', time.localtime(d + self._start_time)) for d in xticks]
        else:
            xlables = [time.strftime('%H:%M:%S', time.localtime(d + self._start_time)) for d in xticks]

        return xticks,xlables

    def color(self,index):
        return all_colors[index]