Курс 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"
- Применение команды break
- Приближение чисел в Python
- Транспонирование 2D-массива с помощью zip
- Импорт модуля из другого каталога
- Работа с zip()
- Работа с модулем random
- Сортировка данных с лямбда-функциями
- Работа с zip-архивами в Python
- Сортировка и разворот списка
- Метод __complex__ в Python
- Работа с массивами в Python
- Enum в Python
- Проверка списка: any() и all()
- Метод pop() списка
- Фильтрация списков с itertools
- Руководство по Pymorphy2
- Проверка наличия элемента в списке
- Встроенные функции Python
- Избегайте пустого списка
- Метод pos в Python
- Оператор is в Python
- Обработка ошибок ввода данных
- Работа с модулем bisect
- Создание объекта timedelta
- Форматирование кода на Python
- Работа со временем в Python
- Декоратор защиты анонимных пользователей
- Тестирование модели в PyTorch
- Извлечение чисел из текста
- Метод join для наборов
- Циклы for в Python
- Работа с изменяемыми коллекциями
- Бесконечная проверка в Python
- Использование метода lower()
- Решение переменной Шредингера
- Переменные класса и экземпляра
- Необязательные аргументы в Python
- Преобразование чисел в восьмеричную строку
- Поиск кода
- Освоение Python
- Работа с словарями в Python
- Переворот строки с использованием цикла
- Определение объема памяти объекта
- Отправка поздравлений по дню рождения
- Отображение графиков в Jupyter с Matplotlib
- Обмен переменными в Jupyter
- Цикл for в Python
