Как запретить импорт определённого модуля в python-скрипте?

Question

[Alexey Al]

Пример:

Если в любом месте кода написать `import uuid` или `from uuid import <...>`, скрипт не запустится.
При этом библиотеки могут использовать `import uuid` свободно.

Существует ли способ такое сделать?

Comments

[Saboteur]

а зачем???

[Alexey Al]

Saboteur, в нашем сервисе используется библиотека fastuuid вместо uuid,
и я целый день потратил на поиск ошибки, когда у меня два set() с ууидами не пересекаются, хотя и должны.
В итоге выяснилось, что в одном множестве лежат uuid.UUID, а в другом fastuuid.UUID,
поэтому эти множества никогда не пересекутся.
Мы подумали, а нельзя ли как-то запретить импорт uuid, раз уж нужно использовать fastuuid, потому что по-любому, кто-нибудь в следующий раз про это забудет и снова потратит кучу времени.

При этом некоторые пакеты в requirements.txt используют uuid, это ок.

Пробовал разобраться, можно ли как-то приспособить под это ast, но так и недоразобрался.

[Everything_is_bad]

Alexey Al, ну так эту проблему надо было сразу и озвучить в вопросе, а решается она не запретами, а созданием абстракции, которая и будет выдать "правильные" UUID, причем даже лучше чтобы ее вызов был автоматический, а не явный ручной.

Answer 1

[angeloffy]

import sys
import uuid
module_name = "uuid"
if sys.modules.get(module_name):
    raise ImportError(f"Запрет на модуль {module_name}")

В ответе: ImportError: Запрет на модуль uuid

Answer 2

[Vindicar]

Как ты отличишь импорт из скрипта от импорта из библиотеки?
Можно отреагировать на импорт в момент, когда он происходит - но понять, кто его выполнил, тяжело.
И даже если ты сумеешь через анализ текущего фрейма стека понять, кто вызвал импорт - импортированный модуль будет закэширован, и скрипт потом сможет обратиться к нему через sys.modules. Так что либо запрещать импорт модуля в принципе (например, через аудит хук), либо разрешать его во всём скрипте.