Python Scheduler.get_first方法代码示例

# 需要导入模块: from scheduler import Scheduler [as 别名]
# 或者: from scheduler.Scheduler import get_first [as 别名]
class SchedulerTest1(unittest.TestCase):
    def setUp(self):
        self.groups = Groups("groups.json")
        lat = 56.045000000000002
        lon = 12.718400000000001

        self.s = Scheduler(lat=lat, lon=lon, groups=self.groups)
        self.s.parse_conf_file("schedule.json")  # Prepped schedule with 11 schedule items and 2 rules

    def test1_start_at_midnight(self):
        """ This is a normal situation, where the schedules are reloaded when it's 00:00/a new day
        """
        no = self.s.new_day("mo")
        self.assertEqual(no, 9)  # 9 items expected for Monday

        # Start at midnight
        item = self.s.get_first()
        self.assertEqual(item["time"], "06:00")  # Monday, first item should be 06:00
        self.assertEqual(item["scenario"], "Plant lights on")

        # Start later in the day, search for exact match in time
        item = self.s.get_first("08:10")
        self.assertEqual(item["time"], "08:10")
        self.assertEqual(item["device"], "2222-2222")

    def test2_start_in_evening(self):
        """ Start later in the day. The search is inexact, expecting a schedule item at a later point-in-time
        """
        self.s.new_day("mo")

        item = self.s.get_first("21:35")
        self.assertEqual(item["time"], "21:40")
        item = self.s.get_next()
        self.assertEqual(item["time"], "22:00")

    def test2b_start_with_errors(self):
        """ Start with invalid days. 0 scheduled items expected as result
        """
        no = self.s.new_day("xx")  # Needs to be "mo", "tu", etc.
        self.assertEqual(no, 0)

        no = self.s.new_day(9)  # Needs to be 1..7
        self.assertEqual(no, 0)

    def test3_new_day(self):
        """ Normal shift of day.
        """
        self.s.new_day("mo")

        self.s.schedules.list_full_day()

        # Get last item for Monday
        item = self.s.get_first("23:00")
        self.assertEqual(item["time"], "23:00")

        # This should fail, there are no more items
        self.assertIsNone(self.s.get_next())

        # It's midnight, let's shift over to a new day
        no = self.s.new_day("tu")
        self.assertEqual(no, 11)  # 11 items expected for Tuesday
        #self.s.schedules.list_full_day()

        item = self.s.get_next()
        self.assertEqual(item["time"], "06:00")  # Tuesday, first item should be 06:00

        item = self.s.get_next()
        self.assertEqual(item["time"], "06:00")  # Tuesday, second item should also be 06:00

    def test3b_new_day(self):
        """ Normal shift of day.
        """
        self.s.new_day("mo")

        # Get last item for Monday
        item = self.s.get_first("23:00")
        self.assertEqual(item["time"], "23:00")

        # This should fail, there are no more items
        self.assertIsNone(self.s.get_next())

        # It's midnight, let's shift over to a new day
        no = self.s.new_day("tu")
        self.assertEqual(no, 11)  # 11 items expected for Tuesday

        item = self.s.get_next()
        self.assertEqual(item["time"], "06:00")  # Tuesday, first item should be 06:00

        item = self.s.get_next()
        self.assertEqual(item["time"], "06:00")  # Tuesday, second item should also be 06:00

    def test4_execute_rule(self):
        """ Locate and execute a rule, use day index to locate Monday
        """
        self.s.new_day(1)
        item = self.s.get_first("00:00")
        self.assertEqual(item["time"], "06:00")

        r = item["rule"]

#.........这里部分代码省略.........

# 需要导入模块: from scheduler import Scheduler [as 别名]
# 或者: from scheduler.Scheduler import get_first [as 别名]
class AgoScheduler(agoclient.AgoApp):

    def __init__(self):
        agoclient.AgoApp.__init__(self)
        self.weekdayno = None
        self.weekday = None
        self.nexttime = None
        self.next_item = None
        self.groups = None
        self.scheduler = None
        self.scenario_controllerUUID = None
        self.mapfile = None
        self.print_schedule = False

    def event_handler(self, subject, content):
        """event handler - Processes incoming events and looks if they are of a relevant kind
                         - For now, it's only time event that are of interest
        """
        self.log.trace("event_handler start")

        if "event.environment.timechanged" in subject:
            self.log.trace("subject=%s content=%s", subject, content)

            js = json.loads(str(content).replace("u\'", '\"').replace("\'", '\"'))  # TODO: check if replacement is necessary
            t_now = self.scheduler.now()
            self.log.trace("now={} - waiting for {}".format(t_now, self.nexttime))

            if self.nexttime is not None and t_now == self.nexttime:
                while True:  # Loop through all items with same time
                    self.log.debug("Action to be triggered: {}".format(self.next_item))
                    try:
                        self.send_message(self.next_item)
                    except NameError as e:
                        self.log.error("Oops, could not send message. Msg={}".format(e))
                    self.next_item = self.scheduler.get_next()
                    if self.next_item is None:
                        self.nexttime = None
                        break
                    if t_now != self.next_item["time"]:
                        self.nexttime = self.next_item["time"]
                        break
                if self.nexttime is not None:
                    self.log.debug("Next item scheduled for {}".format(self.nexttime))
                else:
                    self.log.debug("No more scheduled items for today")

            if self.weekdayno != int(js["weekday"]):
                # TODO: Check config if a new schedule file is to be loaded
                self.weekdayno = int(js["weekday"])
                self.new_day(self.weekdayno)
                self.next_item = self.scheduler.get_first("00:00")
                if self.next_item is not None:
                    self.nexttime = self.next_item["time"]
                else:
                    self.nexttime = None
                self.log.debug("First item scheduled for {}".format(self.nexttime))

        # TODO: Event handler for "reload" event

    def device_msg(self, uuid, action, level=None):
        content = {"uuid": uuid,  # TODO: Check unicode encoded strings
                   "action": action}
        if action == "setlevel":
            content["level"] = level
        elif action == "fade":
            pass

        if action in {"on", "off", "setlevel"}:
            msg = "event.device.statechanged"

            self.log.debug("About to set device content={}".format(content))
            self.connection.send_message(msg, content)

    def send_message(self, item):
        self.log.debug(("In send_message. item={}".format(item)))
        if item["device"] is not None:
            if item["action"] in {"on", "off"}:
                self.device_msg(item["device"], item["action"])
            elif item["action"] in {"setlevel"}:
                self.device_msg(item["device"], item["action"], item["level"])
            # todo: aDD ACTION SPECIFIC FIELDS

        if item["scenario"] is not None:
            content = {"uuid": item["scenario"],
                       "command": "run"}
            self.log.debug("About to execute scenario content={}".format(content))
            self.connection.send_message(None, content)

        if item["group"] is not None:
            self.log.info(("About to send message to device group {}".format(item["group"])))
            grp = self.groups.find(item["group"])
            if grp is not None:
                for dev in grp.devices:
                    if dev is not None:
                        if item["action"] in {"on", "off"}:
                            self.device_msg(dev, item["action"])
                        elif item["action"] in {"setlevel"}:
                            self.device_msg(dev, item["action"], item["level"])

    def get_scenario_controller_uuid(self):
#.........这里部分代码省略.........

# 需要导入模块: from scheduler import Scheduler [as 别名]
# 或者: from scheduler.Scheduler import get_first [as 别名]
class SchedulerTest2(unittest.TestCase):
    def setUp(self):
        self.groups = Groups("groups.json")
        self.s = Scheduler(12.7, 56.6, self.groups)  # Coords for south of Sweden
        self.s.parse_conf_file("winter.json")  # Prepped schedule with 9 schedule items and 2 rules

    def test1_start_at_midnight(self):
        """ This is a normal situation, where the schedules are reloaded when it's 00:00/a new day
        """
        no = self.s.new_day("mo")
        self.assertEqual(no, 7)  # 7 items expected for Monday

        # Start at midnight
        item = self.s.get_first()
        self.assertEqual(item["time"], "06:15")  # Monday, first item should be 06:15
        self.assertEqual(item["scenario"], "ce52eea8-8889-4dc6-b578-7d049b5a64ab")   # Plant lights on"

        # Start later in the day, search for exact match in time
        item = self.s.get_first("08:10")
        self.assertEqual(item["time"], "15:00")
        self.assertEqual(item["group"], "5555-5555")

    def test2_start_in_evening(self):
        """ Start later in the day. The search is inexact, expecting a schedule item at a later point-in-time
        """
        self.s.new_day("mo")

        item = self.s.get_first("21:35")
        self.assertEqual(item["time"], "21:45")
        item = self.s.get_next()
        self.assertEqual(item["time"], "21:59")

    def test2b_start_with_errors(self):
        """ Start with invalid days. 0 scheduled items expected as result
        """
        no = self.s.new_day("xx")  # Needs to be "mo", "tu", etc.
        self.assertEqual(no, 0)

        no = self.s.new_day(9)  # Needs to be 1..7
        self.assertEqual(no, 0)

    def test3_new_day(self):
        """ Normal shift of day.
        """
        self.s.new_day("mo")

        # Get last item for Monday
        item = self.s.get_first("22:00")
        self.assertEqual(item["time"], "22:00")

        # This should fail, there are no more items
        self.assertIsNone(self.s.get_next())

        # It's midnight, let's shift over to a new day
        no = self.s.new_day("tu")
        self.assertEqual(no, 10)  # 10 items expected for Tuesday

        item = self.s.get_next()
        self.assertEqual(item["time"], "06:00")  # Tuesday, first item should be 06:00

        item = self.s.get_next()
        self.assertEqual(item["time"], "06:00")  # Tuesday, second item should also be 06:00

    def test3_new_day2(self):
        """ Normal shift of day.
        """
        self.s.new_day("we")

        # Get last item for Wednesday
        item = self.s.get_first("22:00")
        self.assertEqual(item["time"], "22:00")

        # This should fail, there are no more items
        self.assertIsNone(self.s.get_next())

        # It's midnight, let's shift over to a new day
        no = self.s.new_day("th")
        self.assertEqual(no, 7)  # 10 items expected for Thursday

        item = self.s.get_next()
        self.assertEqual(item["time"], "06:25")  # Thursday, first item should be 06:25

        item = self.s.get_next()
        self.assertEqual(item["time"], "15:00")  # Thursday, second item should also be 15:00

    def test4_list(self):
        """ List full day, incl sunrise/sunset calculation"""
        self.s.schedules.list_full_day()

    def test5_sunrise(self):
        """ Test sunrise calculation"""
        now = datetime.now()
        then = now.replace(year=2017, month=02, day=14)
        self.assertEqual(self.s.calc_sunrise(then), "07:40")

        then = now.replace(year=2017, month=07, day=22)
        self.assertEqual(self.s.calc_sunrise(then), "03:51")

    def tearDown(self):
        pass