зеркало из https://github.com/openwpm/OpenWPM.git
81 строка
2.3 KiB
Python
81 строка
2.3 KiB
Python
from typing import Iterable, List
|
|
|
|
import yaml
|
|
|
|
with open("environment-unpinned.yaml", "r") as fp:
|
|
env_unpinned = yaml.load(
|
|
fp.read(),
|
|
Loader=yaml.SafeLoader,
|
|
)
|
|
with open("environment-unpinned-dev.yaml", "r") as fp:
|
|
env_unpinned_dev = yaml.load(
|
|
fp.read(),
|
|
Loader=yaml.SafeLoader,
|
|
)
|
|
with open("../environment.yaml", "r") as fp:
|
|
env_pinned = yaml.load(
|
|
fp.read(),
|
|
Loader=yaml.SafeLoader,
|
|
)
|
|
|
|
# Only pin explicit dependencies
|
|
|
|
|
|
def iterate_deps(
|
|
pinned_deps: Iterable[str], unpinned_deps: Iterable[str], accumulator: List[str]
|
|
) -> None:
|
|
for pinned_dep in pinned_deps:
|
|
for unpinned_dep in unpinned_deps:
|
|
for sep in ["=", "<", ">"]:
|
|
if pinned_dep.split("=")[0] == unpinned_dep.split(sep)[0]:
|
|
accumulator.append(pinned_dep)
|
|
break # break out of the sep loop to avoid duplicate entries
|
|
|
|
|
|
deps_not_pip: List[str] = []
|
|
deps_pip: List[str] = []
|
|
|
|
env_unpinned_contains_pip = "pip" in env_unpinned["dependencies"][-1]
|
|
env_unpinned_dev_contains_pip = "pip" in env_unpinned_dev["dependencies"][-1]
|
|
iterate_deps(
|
|
env_pinned["dependencies"][:-1]
|
|
if env_unpinned_contains_pip or env_unpinned_dev_contains_pip
|
|
else env_pinned["dependencies"],
|
|
(
|
|
env_unpinned["dependencies"][:-1]
|
|
if env_unpinned_contains_pip
|
|
else env_unpinned["dependencies"]
|
|
)
|
|
+ (
|
|
env_unpinned_dev["dependencies"][:-1]
|
|
if env_unpinned_dev_contains_pip
|
|
else env_unpinned_dev["dependencies"]
|
|
),
|
|
deps_not_pip,
|
|
)
|
|
|
|
deps_pip_unpinned = (
|
|
env_unpinned["dependencies"][-1]["pip"] if env_unpinned_contains_pip else []
|
|
)
|
|
deps_pip_unpinned_dev = (
|
|
env_unpinned_dev["dependencies"][-1]["pip"] if env_unpinned_dev_contains_pip else []
|
|
)
|
|
if env_unpinned_contains_pip or env_unpinned_dev_contains_pip:
|
|
iterate_deps(
|
|
env_pinned["dependencies"][-1]["pip"],
|
|
deps_pip_unpinned_dev + deps_pip_unpinned,
|
|
deps_pip,
|
|
)
|
|
pruned_dependencies = [
|
|
*sorted(list(set(deps_not_pip))),
|
|
{"pip": sorted(list(set(deps_pip)))},
|
|
]
|
|
|
|
# Update env_pinned with pruned dependencies
|
|
env_pinned["dependencies"] = pruned_dependencies
|
|
# We don't want the prefix line
|
|
env_pinned.pop("prefix")
|
|
|
|
with open("../environment.yaml", "w") as f:
|
|
f.write(yaml.dump(env_pinned))
|