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
use colored::*;
mod cmd;
mod error;
mod execv;
mod lock;
mod python_info;
mod venv_manager;
pub use crate::cmd::Command;
use crate::cmd::SubCommand;
pub use crate::cmd::{print_error, print_info_1, print_info_2};
pub use crate::error::Error;
use crate::python_info::PythonInfo;
use crate::venv_manager::InstallOptions;
use crate::venv_manager::VenvManager;
pub use crate::venv_manager::LOCK_FILE_NAME;
pub fn run(cmd: Command) -> Result<(), Error> {
let working_dir = if let Some(cwd) = cmd.working_dir {
std::path::PathBuf::from(cwd)
} else {
let maybe_cwd = std::env::current_dir();
if let Err(e) = maybe_cwd {
return Err(Error::Other {
message: format!("Could not get current directory: {}", e),
});
}
maybe_cwd.unwrap()
};
if let SubCommand::Run { ref cmd, .. } = cmd.sub_cmd {
if cmd.is_empty() {
return Err(Error::Other {
message: format!("Missing argument after '{}'", "run".green()),
});
}
}
let mut python_info = PythonInfo::new(&cmd.python_binary)?;
if std::env::var("DMENV_NO_VENV_STDLIB").is_ok() {
python_info.venv_from_stdlib = false;
}
let venv_manager = VenvManager::new(working_dir, python_info)?;
match &cmd.sub_cmd {
SubCommand::Install {
no_develop,
no_upgrade_pip,
} => {
let mut install_options = InstallOptions::default();
install_options.develop = !no_develop;
install_options.upgrade_pip = !no_upgrade_pip;
venv_manager.install(&install_options)
}
SubCommand::Clean {} => venv_manager.clean(),
SubCommand::Develop {} => venv_manager.develop(),
SubCommand::Init {
name,
version,
author,
} => venv_manager.init(&name, &version, author),
SubCommand::Lock {} => venv_manager.lock(),
SubCommand::BumpInLock { name, version, git } => {
venv_manager.bump_in_lock(name, version, *git)?;
Ok(())
}
SubCommand::Run { ref cmd, no_exec } => {
if *no_exec {
venv_manager.run_no_exec(cmd)
} else {
venv_manager.run(cmd)
}
}
SubCommand::ShowDeps {} => venv_manager.show_deps(),
SubCommand::ShowVenvPath {} => venv_manager.show_venv_path(),
SubCommand::UpgradePip {} => venv_manager.upgrade_pip(),
}
}