Python Bot.select_exercise_and_start_time方法代码示例

# 需要导入模块: from bot import Bot [as 别名]
# 或者: from bot.Bot import select_exercise_and_start_time [as 别名]
class Server(object):
    def __init__(self, configuration, workout_logger=None, **kwargs):
        self.logger = logging.getLogger(__name__)
        self.configuration = configuration
        if workout_logger == None:
            logger_factory = LoggerFactory(self.configuration)
            self.workout_logger = logger_factory.get_logger()
        else:
            self.workout_logger = workout_logger

        if 'slack_api' in kwargs:
            self.slack_api = kwargs['slack_api']
        else:
            self.slack_api = FlexbotApiClient(configuration, token=self.configuration.slack_token())

        if 'user_manager' in kwargs:
            self.user_manager = kwargs['user_manager']
        else:
            self.user_manager = UserManager(self.slack_api, self.configuration, self.workout_logger)

        if 'bot' in kwargs:
            self.bot = kwargs['bot']
        else:
            self.bot = Bot(self.slack_api, self.configuration, self.user_manager)

        if 'web_server' in kwargs:
            self.web_server = kwargs['web_server']
        else:
            self.web_server = FlexbotWebServer(self.user_manager, self.user_manager, configuration)

    def start(self):
        self.logger.debug('Starting workout loop')
        workout_loop_thread = threading.Thread(target=self.workout_loop)
        workout_loop_thread.daemon = False
        workout_loop_thread.start()
        # Start the webserver
        self.logger.debug('Starting webserver')
        cherrypy.config.update({'server.socket_host': '0.0.0.0',
                                'server.socket_port': self.configuration.webserver_port(),
                                'log.screen': True,
                               })
        cherrypy.quickstart(self.web_server)

    def workout_loop(self):
        """
        Runs the workout bot in an infinite loop.
        """
        was_office_hours = False

        while True:
            was_office_hours = self.workout_step(was_office_hours)

    def workout_step(self, was_office_hours):
        """
        Runs a step of the workout bot, handling exceptions. Returns True iff is_office_hours
        returns true before the current workout.
        """
        is_office_hours = self.bot.is_office_hours()
        try:
            self._workout_step(was_office_hours, is_office_hours)
        except NoEligibleUsersException:
            util.sleep(minutes=5)
        return is_office_hours

    def _workout_step(self, was_office_hours, is_office_hours):
        if is_office_hours:
            # Clear the previous day's history if this is the first workout of the day
            if not was_office_hours:
                self.user_manager.clear_users()

            # Get an exercise to do
            exercise, reps, mins_to_exercise = self.bot.select_exercise_and_start_time()
            util.sleep(minutes=mins_to_exercise)

            # Assign the exercise to someone
            self.bot.assign_exercise(exercise, reps)
        else:
            # Show some stats if the final workout has just passed
            if was_office_hours:
                self.slack_api.post_flex_message(self.user_manager.stats())

            # Sleep for a bit
            if not self.configuration.debug():
                util.sleep(minutes=5) # Sleep 5 minutes
            else:
                # If debugging, check again in 5 seconds
                util.sleep(seconds=5)