summaryrefslogtreecommitdiff
path: root/src/python.rs
blob: 2b851f8df3cbd7cf94379e3d7b3c886a389047ca (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
44
45
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> {
        debug!("considering {} as a Python project", dirname.display());
        let setup_py = dirname.join("setup.py");
        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 {
                info!("Looks like a Python project: {}", dirname.display());
                Ok(Self { version_pys: files })
            }
        } else {
            debug!("{} does not exist", setup_py.display());
            Err(BumperError::UnknownProjectKind(dirname.to_path_buf()))
        }
    }

    pub fn set_version(&mut self, version: &str) -> Result<(), 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(())
    }
}

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()
}