forked from fd-sturniolo/AntTracker
-
Notifications
You must be signed in to change notification settings - Fork 0
/
check_env.py
22 lines (21 loc) · 990 Bytes
/
check_env.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
from typing import Literal
def check_env(module: Literal['tracker', 'labeler']):
import sys
frozen = getattr(sys, 'frozen', False) and hasattr(sys, '_MEIPASS')
if not frozen:
from pathlib import Path
for file in (Path("../.env_info"), Path(".env_info")):
env_info_file = file
if env_info_file.exists():
break
else:
raise ValueError("Debe generar un conda env con create-env.ps1")
needed_env = [line.split(':')[1] for line in env_info_file.read_text().split("\n") if line.startswith(module)]
if not needed_env:
raise ValueError("Debe generar un conda env con create-env.ps1")
needed_env = needed_env[0]
import os
current_env = os.environ['CONDA_DEFAULT_ENV']
if needed_env != current_env:
raise ValueError(f"Sólo ejecutar este archivo en el conda-env "
f"generado por create-env.ps1 ({needed_env})")