| 46 | |
| 47 | #[pymethod] |
| 48 | fn seed(&self, n: OptionalOption<PyObjectRef>, vm: &VirtualMachine) -> PyResult<()> { |
| 49 | *self.rng.lock() = match n.flatten() { |
| 50 | Some(n) => { |
| 51 | // Fallback to using hash if object isn't Int-like. |
| 52 | let (_, mut key) = match n.downcast::<PyInt>() { |
| 53 | Ok(n) => n.as_bigint().abs(), |
| 54 | Err(obj) => BigInt::from(obj.hash(vm)?).abs(), |
| 55 | } |
| 56 | .to_u32_digits(); |
| 57 | if cfg!(target_endian = "big") { |
| 58 | key.reverse(); |
| 59 | } |
| 60 | let key = if key.is_empty() { &[0] } else { key.as_slice() }; |
| 61 | MT19937::new_with_slice_seed(key) |
| 62 | } |
| 63 | None => MT19937::try_from_os_rng() |
| 64 | .map_err(|e| std::io::Error::from(e).to_pyexception(vm))?, |
| 65 | }; |
| 66 | Ok(()) |
| 67 | } |
| 68 | |
| 69 | #[pymethod] |
| 70 | fn getrandbits(&self, k: PyObjectRef, vm: &VirtualMachine) -> PyResult<BigInt> { |