Курс Python → Ограничение ресурсов в Python
Для ограничения использования процессора и памяти в Python можно воспользоваться модулем resource. Для начала необходимо получить значения нежёсткого и жёсткого лимитов для конкретного ресурса (например, RLIMIT_CPU для ограничения использования процессора). Для этого можно воспользоваться функцией getrlimit, которая принимает в качестве аргумента константу, определяющую ресурс.
import resource
import signal
# Получаем значения нежёсткого и жёсткого лимитов для RLIMIT_CPU
soft_limit, hard_limit = resource.getrlimit(resource.RLIMIT_CPU)
Затем необходимо установить лимит, используя некоторое число секунд (или другую единицу измерения времени) в качестве аргумента. Для этого можно воспользоваться функцией setrlimit, передав ей новое значение нежёсткого лимита и ранее полученное значение жёсткого лимита.
# Устанавливаем лимит использования процессора в 5 секунд
resource.setrlimit(resource.RLIMIT_CPU, (5, hard_limit))
Далее необходимо зарегистрировать обработчик сигнала, который будет инициировать процедуру выхода из программы при превышении установленного лимита. Для этого можно воспользоваться модулем signal и функцией signal.signal, указав соответствующий сигнал (например, signal.SIGXCPU для превышения времени процессора).
# Регистрируем обработчик сигнала SIGXCPU
def signal_handler(signum, frame):
print("Превышено время использования процессора")
exit()
signal.signal(signal.SIGXCPU, signal_handler)
Для ограничения использования памяти можно также использовать модуль resource. После получения значений нежёсткого и жёсткого лимитов для нужного ресурса (например, RLIMIT_AS для ограничения использования памяти), необходимо установить ограничение с помощью функции setrlimit, передав размер ограничения и значение жёсткого лимита.
# Получаем значения нежёсткого и жёсткого лимитов для RLIMIT_AS
soft_limit, hard_limit = resource.getrlimit(resource.RLIMIT_AS)
# Устанавливаем лимит использования памяти в 100 МБ
resource.setrlimit(resource.RLIMIT_AS, (100 * 1024 * 1024, hard_limit))
Другие уроки курса "Python"
- Распаковка аргументов в Python
- Группировка элементов в словарь
- Генерация ключей RSA
- Отладка в Python
- Фильтрация списка чисел
- Работа с каталогами в Python
- Создание namedtuple из словаря
- Генераторы в Python
- Наследование в программировании
- Сравнение def и lambda функций в Python
- ROT13 Шифр Цезаря в Python
- Работа с контекстными менеджерами
- Antigravity модуль
- Вычисление фазы комплексного числа
- Создание словарей с defaultdict
- Псевдонимы в Python
- Непрерывная проверка в Python
- Преобразование в float
- Исключение NotImplementedError
- Итерация по итерируемым объектам
- Python Calendar Usage
- Операторы сравнения в Python
- Отрицательные индексы списков
- Создание функций с произвольным количеством аргументов
- Метод ne для сравнения объектов
- Метод ior для битовых операций
- Форматирование строк в Python
- Необязательные аргументы в Python
- Ограничение ресурсов в Python
- Синтаксис переменных цикла в Python
- Оператор is в Python
- Измерение времени выполнения в Python
- Метод get для словаря
- Ветвление выражения в Python
- Сложение матриц в NumPy
- Выключение компьютера с помощью Python
- Создание словарей в Python
- Итерация по копии коллекции
- Flask: создание веб-приложений
- Обработка исключений
- Работа с аргументами командной строки в Python
- Создание комплексных чисел
- Функции-генераторы в Python
- Логирование с Loguru
