import unittest
from refill import queueListener, ChannelReader, ChannelWriter, open_hdf5, mktime,day_stamp
from refill import MerchantReader,MerchantWriter
import signal as sig
import sys,getopt
import logging

class TestCommand(unittest.TestCase):
    def test_copy_channel(self):
        def copy(reader,writer,paths):
            for path in paths:
                data = reader.read_path(path)
                writer.write_set(path,data)
            pass

        latest_time = mktime('2023-03-01 00:00:00')
        lastest_day = day_stamp(latest_time)

        reader = ChannelReader()
        hDFive = open_hdf5('/var/www/html/data/stdata/channel_bak.hdf5', True)
        writer = ChannelWriter(hDFive)
        days = reader.days()
        for day in days:
            day = int(day)
            if day < lastest_day:
                continue
            paths = reader.datasets(day)
            copy(reader,writer,paths)
            print(paths)
        pass

    def test_copy_mch(self):
        def copy(reader,writer,paths):
            for path in paths:
                data = reader.read_path(path)
                writer.write_set(path,data)
            pass

        latest_time = mktime('2023-03-01 00:00:00')
        lastest_day = day_stamp(latest_time)

        reader = MerchantReader()
        hDFive = open_hdf5('/var/www/html/data/stdata/merchant_bak.hdf5', True)
        writer = MerchantWriter(hDFive)
        days = reader.days()
        for day in days:
            day = int(day)
            if day < lastest_day:
                continue
            paths = reader.datasets(day)
            copy(reader,writer,paths)
            print(paths)
        pass

if __name__ == '__main__':
    unittest.main()