RustPython/vm/src/stdlib/time_module.rs at function_attributes · casey/RustPython

38 lines (33 loc) · 1.2 KB

//! The python `time` module.

use crate::obj::objfloat;

use crate::pyobject::{PyContext, PyFuncArgs, PyObjectRef, PyResult, TypeProtocol};

use crate::VirtualMachine;

use std::thread;

use std::time::{Duration, SystemTime, UNIX_EPOCH};

fn time_sleep(vm: &mut VirtualMachine, args: PyFuncArgs) -> PyResult {

arg_check!(vm, args, required = [(seconds, Some(vm.ctx.float_type()))]);

let seconds = objfloat::get_value(seconds);

let secs: u64 = seconds.trunc() as u64;

let nanos: u32 = (seconds.fract() * 1e9) as u32;

let duration = Duration::new(secs, nanos);

thread::sleep(duration);

Ok(vm.get_none())

}

fn duration_to_f64(d: Duration) -> f64 {

(d.as_secs() as f64) + (f64::from(d.subsec_nanos()) / 1e9)

}

fn time_time(vm: &mut VirtualMachine, args: PyFuncArgs) -> PyResult {

arg_check!(vm, args);

let x = match SystemTime::now().duration_since(UNIX_EPOCH) {

Ok(v) => duration_to_f64(v),

Err(err) => panic!("Error: {:?}", err),

};

let value = vm.ctx.new_float(x);

Ok(value)

}

pub fn mk_module(ctx: &PyContext) -> PyObjectRef {

py_module!(ctx, "time", {

"sleep" => ctx.new_rustfunc(time_sleep),

"time" => ctx.new_rustfunc(time_time)

})

}

Provide feedback