Python Scheduler.parse_conf_file方法代码示例

# 需要导入模块: from scheduler import Scheduler [as 别名]
# 或者: from scheduler.Scheduler import parse_conf_file [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 parse_conf_file [as 别名]

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

        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):
        """Get UUID for the scenario controller"""
        inventory = self.connection.get_inventory()

        for uuid in inventory['devices']:
            if inventory['devices'][uuid]['devicetype'] == 'scenariocontroller':
                self.log.debug("Found Scenario Controller {}".format(uuid))
                return uuid

        return None

    def new_day(self, weekdayno):
        """ Load schedules for a new day

        @param weekdayno: 1-7
        @return: nothing
        """
        self.weekday = weekday = all_days[weekdayno - 1]
        self.log.trace("new_day() weekday={}".format(weekday))
        no_activities = self.scheduler.new_day(weekday)
        self.next_item = self.scheduler.get_first("00:00")
        self.weekdayno = weekdayno

        if self.print_schedule:
            self.scheduler.schedules.list_full_day()
        self.log.info("New day: {}. Loaded {} schedule items for today.".format(weekday, no_activities))

    def setup_app(self):
        self.log.trace("setup_app start")
        self.connection.add_event_handler(self.event_handler)
        app = "scheduler"

        self.scenario_controllerUUID = self.get_scenario_controller_uuid()
        if self.scenario_controllerUUID is None:
            self.log.error("Scenario Controller not found.")

        mapfile = self.get_config_option('schedule', None, section=app, app=app)
        self.print_schedule = self.get_config_option('print_daily_schedule', "Yes", section=app, app=app) == "Yes"
        self.log.info("Print daily schedule? {}".format(self.print_schedule))

        if mapfile is None:
            self.mapfile = None
            self.log.error("No Schedule file found in config file. Idling.")  # TODO: Follow up - set flag to avoid processing!
        else:
            self.mapfile = agoclient.config.CONFDIR + '/conf.d/' + mapfile
            self.log.info("Reading schedules from {}".format(self.mapfile))

        self.groups = Groups(agoclient.config.CONFDIR + '/conf.d/' + 'groups.json')  # TODO: Change to proper file

        lat = self.get_config_option("lat", "47.07", "system")
        lon = self.get_config_option("lon", "15.42", "system")

        self.log.info("Getting position. lat={} lon={}".format(lat, lon))
        self.scheduler = Scheduler(lat=float(lat), lon=float(lon), log=self.log, groups=self.groups)

        self.scheduler.parse_conf_file(self.mapfile)
        dl, day_no = self.scheduler.get_weekday()
        self.scheduler.weekday = dl
        self.log.trace("day_no={}".format(day_no))
        self.new_day(day_no)

        self.next_item = self.scheduler.get_first(self.scheduler.now())
        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))

# 需要导入模块: from scheduler import Scheduler [as 别名]
# 或者: from scheduler.Scheduler import parse_conf_file [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