pyo3/src/token.rs

// Copyright (c) 2017-present PyO3 Project and Contributors

use std::marker::PhantomData;

use pointers::Py;
use python::Python;
use typeob::{PyTypeInfo, PyObjectAlloc};


pub struct PythonToken<T>(PhantomData<T>);

impl<T> PythonToken<T> {
    pub fn token<'p>(&'p self) -> Python<'p> {
        unsafe { Python::assume_gil_acquired() }
    }
}

#[inline]
pub fn with_token<'p, T, F>(py: Python<'p>, f: F) -> Py<'p, T>
    where F: FnOnce(PythonToken<T>) -> T,
          T: PyTypeInfo + PyObjectAlloc<Type=T>
{
    let value = f(PythonToken(PhantomData));
    if let Ok(ob) = Py::new(py, value) {
        ob
    } else {
        ::err::panic_after_error()
    }
}


pub trait PythonObjectWithGilToken<'p> : Sized {
    fn gil(&self) -> Python<'p>;
}

pub trait PythonObjectWithToken : Sized {
    fn token<'p>(&'p self) -> Python<'p>;
}