xyzshop/plot/refill/CalcBase.py

import redis
import time as time
import logging
import json
from .DataStream import span_days

import traceback

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

class CalcBase(object):
    def __init__(self):
        self._mQuit = False
        self._mRHost = ''
        self._mRPort = 6379

    def set_redis(self, rhost, rport):
        self._mRHost = rhost
        self._mRPort = rport

    def stop(self):
        self._mquit = True

    def _calc_handler(self, rclient):
        return 1

    def _reader(self):
        return None

    def run(self):
        def redis_client():
            pool = redis.ConnectionPool(host=self._mRHost, port=self._mRPort)
            client = redis.Redis(connection_pool=pool)
            return client

        client = None
        sleep_time = 1
        while self._mQuit == False:
            try:
                if client is None:
                    client = redis_client()
                sleep_time = self._calc_handler(client)
            except redis.RedisError as ex:
                logger.error(ex)
            except Exception as ex:
                logger.error(ex)
                logger.error(traceback.format_exc())
            finally:
                time.sleep(sleep_time)

    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

        days = span_days(start_time, end_time)

        # stime = lambda t: time.strftime('%d-%H:%M:%S', time.localtime(t))
        # sdays = [stime(day) for day in days]
        # logger.debug(sdays)

        return days, start_time, end_time