Как реализовать остановку цикла for?

Question

[Statium]

Добрый день, имеется скрипт, код ниже:

class HelloWorldSvc(win32serviceutil.ServiceFramework):
    _svc_name_ = "Monitoring"
    _svc_display_name_ = "HMonitoring"

    def __init__(self, args):
        win32serviceutil.ServiceFramework.__init__(self, args)
        self.stop_event = win32event.CreateEvent(None, 0, 0, None)
        socket.setdefaulttimeout(60)
        self.stop_requested = False

    def SvcStop(self):
        self.ReportServiceStatus(win32service.SERVICE_STOP_PENDING)
        win32event.SetEvent(self.stop_event)
        self.stop_requested = True

    def SvcDoRun(self):
        servicemanager.LogMsg(
            servicemanager.EVENTLOG_INFORMATION_TYPE,
            servicemanager.PYS_SERVICE_STARTED,
            (self._svc_name_, '')
        )
        self.main()

    def main(self):
        # Simulate a main loop
        time.sleep(5)
        for i in iter(int, 1):
            # Подключение к MySQL
            logging.info("Подключаемся к БД MySQL")
            mydb = mysql.connector.connect(host/pass)
            mycursor = mydb.cursor()
            logging.info("Подключение выполнено успешно")

            time.sleep(3)
            if self.stop_requested:
                logging.info('Получен запрос на остановку службы. Служба остановлена')
                break

            # Получаем IP-адрес сервера
            logging.info("Запрашиваем IP-адрес сервера")
            try:
                ip = requests.get('http://localhost/')  # get-запрос
                logging.info("IP-адрес получен")
                time.sleep(3)
                ip = ip.text
            except:
                ip = "Error"
                logging.info("Не могу получить IP-адрес")

            if self.stop_requested:
                logging.info('Получен запрос на остановку службы. Служба остановлена')
                break
        return


if __name__ == '__main__':
    if len(sys.argv) == 1:
        servicemanager.Initialize()
        servicemanager.PrepareToHostSingle(HelloWorldSvc)
        servicemanager.StartServiceCtrlDispatcher()
    else:
        win32serviceutil.HandleCommandLine(HelloWorldSvc)

Данный код копилируется в exe и запускается в винде как служба, это только кусочек кода, на самом деле цикл (бесконечный) выполняет множество различных запросов и отправляет данные на сервер.

После каждого запроса у меня в коде стоит

if self.stop_requested:
                logging.info('Получен запрос на остановку службы. Служба остановлена')
                break

Если попытаться остановить службу она останавливает цикл. Дак вот у меня этих кусков кода с проверкой остановки, в скрипте штук 50. Можно ли как то остановить цикл сразу после остановки службы? Если оставить этот if в одном месте, перед остановкой служба ждет завершение всего цикла.

Answer 1

[Vindicar]

Если тебе нужна проверка на останов после каждой операции - придётся-таки эту проверку производить после каждой операции. Да, можно схитрить, но от этого не уйдёшь.

Я могу посоветовать убрать time.sleep() и заменить их на threading.Event.wait() с таймаутом - в этом случае, если Event будет взведено, то не придётся "досыпать". Но проверять исход ожидания всё равно придётся самому.

Альтернативное, но ОЧЕНЬ кардинальное решение - сделать код асинхронным. Причина в том, что у асинхронных подпрограмм, завёрнутых в Task, есть механизм отмены, позволяющий принудительно выкинуть исключение в ходе выполнения любой длительной операции (любого await-вызова). По сути, это та же самая проверка - но встроенная.

Это работает, так как асинхронный код выполняется в одном потоке. Провернуть такой же фокус, скажем, в многопоточном приложении - нетривиально и довольно рискованно.

Comments

[Statium]

можно пример как это обернуть в асинхронный цикл?

[Vindicar]

Statium, ну в очень общих чертах как-то так. Это не готовый код, это только идея.

async def serve_one():
    conn = await aiomysql.connect()  # нужно использовать асинхронные библиотеки!
    ...
    await asyncio.sleep(3)  # не time.sleep()! asyncio.sleep() позволяет выполняться "соседним" задачам
    ip = await aiohttp.get('http://localhost/')  # см. выше про библиотеки
    ...
    # ну и так далее

async def serve_all():
    for i in iter(int, 1):
        await serve_one()

serve_task = None

async def monitor():  # а вот эта корутина будет висеть и тихонько мониторить признак остановки.
    global serve_task  # вместо global'ов можно отрефакторить на класс, суть не меняется.
    while not stop_requested:  # спим, пока нас не попросят остановиться.
        await asyncio.sleep(0.5)
    serve_task.cancel()  # сигнализируем, что нужно прервать главную задачу.
    # это спровоцирует исключение asyncio.CancelledError в текущем await-вызове внутри задачи.

async def main():
    global serve_task
    serve_task = asyncio.create_task(serve_all())  # запускаем serve_all() "параллельно" себе
    monitor = asyncio.create_task(monitor())  # запускаем monitor() "параллельно" себе
    try:
        await serve_task  # ждём завершения serve_all()
    except asyncio.CancelledError:
        pass  # задача была прервана, но мы этого и ожидали

if __name__ == '__main__':
    asyncio.run(main())