Remove only insertion of YAML libs to safety

This commit is contained in:
Oz N Tiram
2022-07-25 21:18:38 +02:00
parent 53dc092d38
commit e532fb5328
2 changed files with 1 additions and 7 deletions
-3
View File
@@ -44,8 +44,5 @@ def insert_pipenv_dirs():
if __name__ == "__main__": # pragma: no cover
insert_pipenv_dirs()
yaml_lib = "pipenv.patched.yaml{0}".format(sys.version_info[0])
locals()[yaml_lib] = __import__(yaml_lib)
sys.modules["yaml"] = sys.modules[yaml_lib]
from pipenv.patched.safety.cli import cli
cli(prog_name="safety")
@@ -2,7 +2,7 @@ diff --git a/pipenv/patched/safety/__main__.py b/pipenv/patched/safety/__main__.
index d9a0bdab..f905408a 100644
--- a/pipenv/patched/safety/__main__.py
+++ b/pipenv/patched/safety/__main__.py
@@ -1,8 +1,51 @@
@@ -1,8 +1,48 @@
"""Allow safety to be executable through `python -m safety`."""
from __future__ import absolute_import
@@ -50,8 +50,5 @@ index d9a0bdab..f905408a 100644
if __name__ == "__main__": # pragma: no cover
+ insert_pipenv_dirs()
+ yaml_lib = "pipenv.patched.yaml{0}".format(sys.version_info[0])
+ locals()[yaml_lib] = __import__(yaml_lib)
+ sys.modules["yaml"] = sys.modules[yaml_lib]
+ from safety.cli import cli
cli(prog_name="safety")