summaryrefslogtreecommitdiff
path: root/src/python.rs
blob: 4cc043e69620090dc9d04c63c189c189331d198d (plain)
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
use crate::errors::BumperError;
use glob::glob;
use log::{debug, info};
use std::path::{Path, PathBuf};

pub struct Python {
    version_pys: Vec<PathBuf>,
}

impl Python {
    pub fn new(dirname: &Path) -> Result<Self, BumperError> {
        let setup_py = dirname.join("setup.py");
        debug!("does {} exist? {}", setup_py.display(), setup_py.exists());
        if setup_py.exists() {
            let files = find_version_py_files(dirname);
            if files.is_empty() {
                debug!("no version.py files in {}", dirname.display());
                Err(BumperError::NoVersionPy(dirname.to_path_buf()))
            } else {
                Ok(Self { version_pys: files })
            }
        } else {
            Err(BumperError::UnknownProjectKind(dirname.to_path_buf()))
        }
    }

    pub fn set_version(&mut self, version: &str) -> Result<String, BumperError> {
        for filename in self.version_pys.iter() {
            info!("writing Python version to {}", filename.display());
            std::fs::write(filename, format!("__version__ = \"{}\"\n", version))
                .map_err(|err| BumperError::PythonWrite(filename.to_path_buf(), err))?;
        }
        Ok(version.to_string())
    }
}

fn find_version_py_files(dirname: &Path) -> Vec<PathBuf> {
    glob(&dirname.join("*/version.py").display().to_string())
        .expect("glob pattern error")
        .filter(|x| x.is_ok())
        .map(|x| x.unwrap())
        .collect()
}