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