flodl-cli 0.5.0

libtorch manager and GPU diagnostic tool for Rust deep learning
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

//! Archive extraction (zip files).
//!
//! Shells out to `unzip` (Linux/macOS) or `PowerShell Expand-Archive` (Windows).

use std::path::Path;
use std::process::{Command, Stdio};

/// Extract a zip file to `dest_dir`.
///
/// Creates `dest_dir` if it doesn't exist.
pub fn extract_zip(zip_path: &Path, dest_dir: &Path) -> Result<(), String> {
    std::fs::create_dir_all(dest_dir)
        .map_err(|e| format!("cannot create {}: {}", dest_dir.display(), e))?;

    let zip_str = zip_path
        .to_str()
        .ok_or_else(|| "zip path is not valid UTF-8".to_string())?;
    let dest_str = dest_dir
        .to_str()
        .ok_or_else(|| "destination path is not valid UTF-8".to_string())?;

    let status = if cfg!(target_os = "windows") {
        Command::new("powershell")
            .args([
                "-NoProfile",
                "-Command",
                &format!(
                    "Expand-Archive -Force -Path '{}' -DestinationPath '{}'",
                    zip_str, dest_str
                ),
            ])
            .stdout(Stdio::null())
            .stderr(Stdio::inherit())
            .status()
    } else {
        // Check for unzip
        if Command::new("unzip")
            .arg("--help")
            .stdout(Stdio::null())
            .stderr(Stdio::null())
            .status()
            .is_err()
        {
            return Err(
                "unzip is required but not installed.\n\
                 \n\
                 \x20 Ubuntu/Debian:  sudo apt install unzip\n\
                 \x20 Fedora/RHEL:    sudo dnf install unzip\n\
                 \x20 macOS:          available by default"
                    .into(),
            );
        }
        Command::new("unzip")
            .args(["-q", "-o", zip_str, "-d", dest_str])
            .stdout(Stdio::null())
            .stderr(Stdio::inherit())
            .status()
    };

    match status {
        Ok(s) if s.success() => Ok(()),
        Ok(s) => Err(format!(
            "extraction failed (exit code {})\n  Archive: {}",
            s.code().unwrap_or(-1),
            zip_path.display()
        )),
        Err(e) => Err(format!("failed to run extraction command: {}", e)),
    }
}