mb-dev 0.1.12

my dev cli tools
Documentation 
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
90
91
92

use std::{env, path::Path};

use clap::{crate_version, Arg, Command};

use mb_dev::{exit, shell, shell_capture};

fn process_install(packages: Vec<&str>) {
    if env::var_os("VIRTUAL_ENV").is_none() {
        exit("venv is not activated")
    }

    if !packages.is_empty() {
        shell(&format!("pip install {}", packages.join(" ")))
    } else if Path::new("setup.py").exists() {
        shell("pip install -Ue .[dev]")
    } else {
        shell("pip install -Ur requirements.txt")
    }
}

fn process_venv() {
    if env::var_os("VIRTUAL_ENV").is_some() {
        exit("venv is activated already")
    }
    if Path::new(".venv").exists() {
        exit(".venv exists already")
    }
    shell("python3 -m venv .venv");
    shell(".venv/bin/pip install -U pip setuptools");
}

fn process_uninstall() {
    if env::var_os("VIRTUAL_ENV").is_none() {
        exit("venv is not activated")
    }
    shell("pip list --format freeze -e | xargs pip uninstall -y");
    shell("pip freeze | xargs pip uninstall -y");
}

fn process_kill() {
    for p in find_python_processes() {
        shell(&format!("kill -9 {}", p))
    }
}

fn find_python_processes() -> Vec<u32> {
    let mut result: Vec<u32> = vec![];
    let res = shell_capture(r#"echo "$(ps -o pid,command -ax)""#);
    for line in res.split('\n') {
        if line.contains("python")
            && (line.contains("uvicorn")
                || line.contains("multiprocessing.resource_tracker")
                || line.contains("multiprocessing.spawn"))
        {
            if let Some(number) = line.split_ascii_whitespace().next() {
                result.push(number.parse::<u32>().unwrap());
            }
        }
    }
    result
}

fn main() {
    let matches = Command::new("python helper")
        .version(crate_version!())
        .subcommand_required(true)
        .arg_required_else_help(true)
        .subcommand(Command::new("pip-list-outdated").alias("o").about("pip list -o"))
        .subcommand(Command::new("pip-list").alias("l").about("pip list"))
        .subcommand(Command::new("pip-update").alias("u").about("pip install -U pip setuptools"))
        .subcommand(
            Command::new("install")
                .alias("i")
                .about("install packages or project (setup.py or requirements.txt)")
                .arg(Arg::new("packages").required(false).multiple_values(true)),
        )
        .subcommand(Command::new("venv").alias("v").about("create .venv"))
        .subcommand(Command::new("uninstall").alias("d").about("uninstall all packages(+editable) from venv"))
        .subcommand(Command::new("kill").alias("k").about("kill python processes"))
        .get_matches();

    match matches.subcommand() {
        Some(("pip-list-outdated", _)) => shell("pip list -o"),
        Some(("pip-list", _)) => shell("pip list"),
        Some(("pip-update", _)) => shell("pip install -U pip setuptools"),
        Some(("install", m)) => process_install(m.values_of("packages").unwrap_or_default().collect::<Vec<&str>>()),
        Some(("venv", _)) => process_venv(),
        Some(("uninstall", _)) => process_uninstall(),
        Some(("kill", _)) => process_kill(),
        _ => println!("unsupported command"),
    }
}