Mercurial > hg
view rust/hg-cpython/src/dirstate/dirs_multiset.rs @ 43177:5cb8867c9e2b
rust-cpython: move $leaked struct out of macro
It wasn't easy to hack the $leaked struct since errors in macro would
generate lots of compile errors. Let's make it a plain struct so we can
easily extend it.
PyLeakedRef keeps a more generic PyObject instead of the $name struct
since it no longer has to call any specific methods implemented by
the $name class. $leaked parameter in py_shared_iterator!() is kept
for future change.
author | Yuya Nishihara <yuya@tcha.org> |
---|---|
date | Sun, 08 Sep 2019 20:26:55 +0900 |
parents | 070a38737334 |
children | 1b2200bd06b6 |
line wrap: on
line source
// dirs_multiset.rs // // Copyright 2019 Raphaël Gomès <rgomes@octobus.net> // // This software may be used and distributed according to the terms of the // GNU General Public License version 2 or any later version. //! Bindings for the `hg::dirstate::dirs_multiset` file provided by the //! `hg-core` package. use std::cell::RefCell; use std::convert::TryInto; use cpython::{ exc, ObjectProtocol, PyBytes, PyClone, PyDict, PyErr, PyObject, PyResult, Python, }; use crate::dirstate::extract_dirstate; use crate::ref_sharing::{PyLeakedRef, PySharedRefCell}; use hg::{ utils::hg_path::{HgPath, HgPathBuf}, DirsMultiset, DirsMultisetIter, DirstateMapError, DirstateParseError, EntryState, }; py_class!(pub class Dirs |py| { data inner: PySharedRefCell<DirsMultiset>; // `map` is either a `dict` or a flat iterator (usually a `set`, sometimes // a `list`) def __new__( _cls, map: PyObject, skip: Option<PyObject> = None ) -> PyResult<Self> { let mut skip_state: Option<EntryState> = None; if let Some(skip) = skip { skip_state = Some( skip.extract::<PyBytes>(py)?.data(py)[0] .try_into() .map_err(|e: DirstateParseError| { PyErr::new::<exc::ValueError, _>(py, e.to_string()) })?, ); } let inner = if let Ok(map) = map.cast_as::<PyDict>(py) { let dirstate = extract_dirstate(py, &map)?; DirsMultiset::from_dirstate(&dirstate, skip_state) } else { let map: Result<Vec<HgPathBuf>, PyErr> = map .iter(py)? .map(|o| { Ok(HgPathBuf::from_bytes( o?.extract::<PyBytes>(py)?.data(py), )) }) .collect(); DirsMultiset::from_manifest(&map?) }; Self::create_instance( py, PySharedRefCell::new(inner), ) } def addpath(&self, path: PyObject) -> PyResult<PyObject> { self.borrow_mut(py)?.add_path( HgPath::new(path.extract::<PyBytes>(py)?.data(py)), ); Ok(py.None()) } def delpath(&self, path: PyObject) -> PyResult<PyObject> { self.borrow_mut(py)?.delete_path( HgPath::new(path.extract::<PyBytes>(py)?.data(py)), ) .and(Ok(py.None())) .or_else(|e| { match e { DirstateMapError::PathNotFound(_p) => { Err(PyErr::new::<exc::ValueError, _>( py, "expected a value, found none".to_string(), )) } DirstateMapError::EmptyPath => { Ok(py.None()) } } }) } def __iter__(&self) -> PyResult<DirsMultisetKeysIterator> { let (leak_handle, leaked_ref) = unsafe { self.leak_immutable(py)? }; DirsMultisetKeysIterator::from_inner( py, leak_handle, leaked_ref.iter(), ) } def __contains__(&self, item: PyObject) -> PyResult<bool> { Ok(self.inner(py).borrow().contains(HgPath::new( item.extract::<PyBytes>(py)?.data(py).as_ref(), ))) } }); py_shared_ref!(Dirs, DirsMultiset, inner); impl Dirs { pub fn from_inner(py: Python, d: DirsMultiset) -> PyResult<Self> { Self::create_instance(py, PySharedRefCell::new(d)) } fn translate_key( py: Python, res: &HgPathBuf, ) -> PyResult<Option<PyBytes>> { Ok(Some(PyBytes::new(py, res.as_ref()))) } } py_shared_iterator!( DirsMultisetKeysIterator, PyLeakedRef, DirsMultisetIter<'static>, Dirs::translate_key, Option<PyBytes> );