Mercurial > hg
view rust/hg-cpython/src/dirstate/non_normal_entries.rs @ 44441:8cdd0b9629e3
merge with stable
author | Martin von Zweigbergk <martinvonz@google.com> |
---|---|
date | Thu, 05 Mar 2020 10:46:31 -0800 |
parents | 8ac5726d695d |
children | 26114bd6ec60 |
line wrap: on
line source
// non_normal_other_parent_entries.rs // // Copyright 2020 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. use cpython::{ exc::NotImplementedError, CompareOp, ObjectProtocol, PyBytes, PyClone, PyErr, PyList, PyObject, PyResult, PyString, Python, PythonObject, ToPyObject, UnsafePyLeaked, }; use crate::dirstate::DirstateMap; use hg::utils::hg_path::HgPathBuf; use std::cell::RefCell; use std::collections::hash_set; py_class!(pub class NonNormalEntries |py| { data dmap: DirstateMap; def __contains__(&self, key: PyObject) -> PyResult<bool> { self.dmap(py).non_normal_entries_contains(py, key) } def remove(&self, key: PyObject) -> PyResult<PyObject> { self.dmap(py).non_normal_entries_remove(py, key) } def union(&self, other: PyObject) -> PyResult<PyList> { self.dmap(py).non_normal_entries_union(py, other) } def __richcmp__(&self, other: PyObject, op: CompareOp) -> PyResult<bool> { match op { CompareOp::Eq => self.is_equal_to(py, other), CompareOp::Ne => Ok(!self.is_equal_to(py, other)?), _ => Err(PyErr::new::<NotImplementedError, _>(py, "")) } } def __repr__(&self) -> PyResult<PyString> { self.dmap(py).non_normal_entries_display(py) } def __iter__(&self) -> PyResult<NonNormalEntriesIterator> { self.dmap(py).non_normal_entries_iter(py) } }); impl NonNormalEntries { pub fn from_inner(py: Python, dm: DirstateMap) -> PyResult<Self> { Self::create_instance(py, dm) } fn is_equal_to(&self, py: Python, other: PyObject) -> PyResult<bool> { for item in other.iter(py)? { if !self.dmap(py).non_normal_entries_contains(py, item?)? { return Ok(false); } } Ok(true) } fn translate_key( py: Python, key: &HgPathBuf, ) -> PyResult<Option<PyBytes>> { Ok(Some(PyBytes::new(py, key.as_ref()))) } } type NonNormalEntriesIter<'a> = hash_set::Iter<'a, HgPathBuf>; py_shared_iterator!( NonNormalEntriesIterator, UnsafePyLeaked<NonNormalEntriesIter<'static>>, NonNormalEntries::translate_key, Option<PyBytes> );