1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
|
#!/usr/bin/env python
# Authors: The MNE-Python contributors.
# License: BSD-3-Clause
# Copyright the MNE-Python contributors.
import re
from pathlib import Path
import tomllib
repo_root = Path(__file__).resolve().parents[2]
with open(repo_root / "pyproject.toml", "rb") as fid:
pyproj = tomllib.load(fid)
# Get our "full" dependences from `pyproject.toml`, but actually ignore the
# "full" section as it's just "full-noqt" plus PyQt6, and for conda we need PySide
ignore = ("dev", "doc", "test", "test_extra", "full", "full-pyqt6")
deps = set(pyproj["project"]["dependencies"])
for section, section_deps in pyproj["project"]["optional-dependencies"].items():
if section not in ignore:
deps |= set(section_deps)
recursive_deps = set(d for d in deps if d.startswith("mne["))
deps -= recursive_deps
deps |= {"pip"}
def remove_spaces(version_spec):
"""Remove spaces in version specs (conda is stricter than pip about this).
https://docs.conda.io/projects/conda/en/latest/user-guide/concepts/pkg-specs.html#package-match-specifications
"""
return "".join(version_spec.split())
def split_dep(dep):
"""Separate package name from version spec."""
pattern = re.compile(r"([^!=<>]+)?([!=<>].*)?")
groups = list(pattern.match(dep).groups())
groups[1] = "" if groups[1] is None else remove_spaces(groups[1])
return tuple(map(str.strip, groups))
# python version
req_python = remove_spaces(pyproj["project"]["requires-python"])
# split package name from version spec
translations = dict(neo="python-neo")
pip_deps = set()
conda_deps = set()
for dep in deps:
package_name, version_spec = split_dep(dep)
# handle package name differences
package_name = translations.get(package_name, package_name)
# PySide6==6.7.0 only exists on PyPI, not conda-forge, so excluding it in
# `environment.yaml` breaks the solver
if package_name == "PySide6":
version_spec = version_spec.replace("!=6.7.0,", "")
# rstrip output line in case `version_spec` == ""
line = f" - {package_name} {version_spec}".rstrip()
# use pip for packages needing e.g. `platform_system` or `python_version` triaging
if ";" in version_spec:
pip_deps.add(f" {line}")
else:
conda_deps.add(line)
# TODO: temporary workaround while we wait for a release containing the fix for
# https://github.com/mamba-org/mamba/issues/3467
pip_deps.remove(" - pyobjc-framework-Cocoa >=5.2.0;platform_system=='Darwin'")
# prepare the pip dependencies section
newline = "\n" # python < 3.12 forbids backslash in {} part of f-string
pip_section = f"""\
- pip:
{newline.join(sorted(pip_deps, key=str.casefold))}
"""
pip_section = pip_section if len(pip_deps) else ""
# prepare the env file
env = f"""\
# THIS FILE IS AUTO-GENERATED BY {'/'.join(Path(__file__).parts[-3:])} AND WILL BE OVERWRITTEN
name: mne
channels:
- conda-forge
dependencies:
- python {req_python}
{newline.join(sorted(conda_deps, key=str.casefold))}
{pip_section}""" # noqa: E501
(repo_root / "environment.yml").write_text(env)
|
