forked from Rust-related/RustPython
47 lines
1.3 KiB
Rust
47 lines
1.3 KiB
Rust
pub(crate) use _uuid::make_module;
|
|
|
|
#[pymodule]
|
|
mod _uuid {
|
|
use crate::{builtins::PyNone, vm::VirtualMachine};
|
|
use mac_address::get_mac_address;
|
|
use once_cell::sync::OnceCell;
|
|
use rand::Rng;
|
|
use std::time::{Duration, SystemTime};
|
|
use uuid::{
|
|
v1::{Context, Timestamp},
|
|
Uuid,
|
|
};
|
|
|
|
fn get_node_id() -> [u8; 6] {
|
|
match get_mac_address() {
|
|
Ok(Some(_ma)) => get_mac_address().unwrap().unwrap().bytes(),
|
|
_ => rand::thread_rng().gen::<[u8; 6]>(),
|
|
}
|
|
}
|
|
|
|
pub fn now_unix_duration() -> Duration {
|
|
use std::time::UNIX_EPOCH;
|
|
|
|
let now = SystemTime::now();
|
|
now.duration_since(UNIX_EPOCH)
|
|
.expect("SystemTime before UNIX EPOCH!")
|
|
}
|
|
|
|
#[pyfunction]
|
|
fn generate_time_safe() -> (Vec<u8>, PyNone) {
|
|
static CONTEXT: Context = Context::new(0);
|
|
let now = now_unix_duration();
|
|
let ts = Timestamp::from_unix(&CONTEXT, now.as_secs(), now.subsec_nanos());
|
|
|
|
static NODE_ID: OnceCell<[u8; 6]> = OnceCell::new();
|
|
let unique_node_id = NODE_ID.get_or_init(get_node_id);
|
|
|
|
(Uuid::new_v1(ts, unique_node_id).as_bytes().to_vec(), PyNone)
|
|
}
|
|
|
|
#[pyattr]
|
|
fn has_uuid_generate_time_safe(_vm: &VirtualMachine) -> u32 {
|
|
0
|
|
}
|
|
}
|