| 12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758 |
- from .DataStream import EMchPosmap as pos_map, span_days, time_border, calc_interval
- from .MerchantReader import MerchantReader
- from .MerchantPainter import MerchantPainter, allpathes
- from matplotlib.figure import Figure
- from matplotlib import ticker
- from io import BytesIO
- import numpy as np
- from .algorithm import calc_cov_mchratios
- import time as time
- import logging
- logger = logging.getLogger('MerchantCovRatioPainter')
- class MerchantCovRatioPainter(MerchantPainter):
- def __init__(self, start_time: int, end_time: int, mchids: set = None, card_types: set = None, spec: int = None, filter_wave: int = None):
- self._reader = MerchantReader()
- filter_wave = filter_wave or 3600
- self._mchids, self._card_types, self._spec, self._filter_wave = mchids, card_types, spec, filter_wave
- days, self._start_time, self._end_time, self._interval = self.calc_time(self._reader, start_time, end_time)
- pass
- def paint(self):
- reader = self._reader
- days, _start_time, _end_time, _interval = self.calc_time(self._reader, self._start_time - self._filter_wave, self._end_time)
- if len(days) == 0:
- return BytesIO()
- tuple_pathes = self._reader.many_tuple_path(days, self._mchids, self._card_types, self._spec)
- gen = allpathes(reader, tuple_pathes, days, self._spec)
- day_stamp = days[0]
- fig_create, fig_flush = self._fig_funs()
- ax, fig = fig_create()
- x = np.array([d - self._start_time for d in range(self._start_time, self._end_time)])
- window = np.ones(self._filter_wave)
- left_len = self._start_time - _start_time
- right_len = _end_time - self._end_time
- mchid_ratios = []
- for _mchid, _card_type, _spec, _data in gen:
- succ, count, y = calc_cov_mchratios(_data, pos_map, _start_time - day_stamp, _end_time - day_stamp, window, left_len, right_len)
- label, ratio = self._label(chname=_mchid, succ=succ, count=count, card_type=_card_type, spec=_spec)
- ax.plot(x, y, ls='-', label=label)
- if _card_type is None and _spec is None and type(_mchid) is int:
- mchid_ratios.append((_mchid, ratio))
- xticks = self.calc_xticks(self._start_time, self._end_time, self._interval)
- xlables = [time.strftime('%d-%H:%M:%S', time.localtime(d + self._start_time)) for d in xticks]
- buf = fig_flush(ax, fig, xticks=xticks, xlables=xlables)
- mchid_ratios = sorted(mchid_ratios, key=lambda x: (x[1], x[0]), reverse=True)
- result = []
- for mchid,ratio in mchid_ratios:
- result.append(f'{mchid}:{ratio}')
- return buf, result
|
