copymap.rs
115 lines
| 3.0 KiB
| application/rls-services+xml
|
RustLexer
|
r42999 | // copymap.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 `hg::dirstate::dirstate_map::CopyMap` provided by the | ||||
//! `hg-core` package. | ||||
use cpython::{PyBytes, PyClone, PyDict, PyObject, PyResult, Python}; | ||||
use std::cell::RefCell; | ||||
use crate::dirstate::dirstate_map::{DirstateMap, DirstateMapLeakedRef}; | ||||
|
r43158 | use hg::CopyMapIter; | ||
|
r42999 | |||
py_class!(pub class CopyMap |py| { | ||||
data dirstate_map: DirstateMap; | ||||
def __getitem__(&self, key: PyObject) -> PyResult<PyBytes> { | ||||
(*self.dirstate_map(py)).copymapgetitem(py, key) | ||||
} | ||||
def __len__(&self) -> PyResult<usize> { | ||||
self.dirstate_map(py).copymaplen(py) | ||||
} | ||||
def __contains__(&self, key: PyObject) -> PyResult<bool> { | ||||
self.dirstate_map(py).copymapcontains(py, key) | ||||
} | ||||
def get( | ||||
&self, | ||||
key: PyObject, | ||||
default: Option<PyObject> = None | ||||
) -> PyResult<Option<PyObject>> { | ||||
self.dirstate_map(py).copymapget(py, key, default) | ||||
} | ||||
def pop( | ||||
&self, | ||||
key: PyObject, | ||||
default: Option<PyObject> = None | ||||
) -> PyResult<Option<PyObject>> { | ||||
self.dirstate_map(py).copymappop(py, key, default) | ||||
} | ||||
def __iter__(&self) -> PyResult<CopyMapKeysIterator> { | ||||
self.dirstate_map(py).copymapiter(py) | ||||
} | ||||
// Python's `dict()` builtin works with either a subclass of dict | ||||
// or an abstract mapping. Said mapping needs to implement `__getitem__` | ||||
// and `keys`. | ||||
def keys(&self) -> PyResult<CopyMapKeysIterator> { | ||||
self.dirstate_map(py).copymapiter(py) | ||||
} | ||||
def items(&self) -> PyResult<CopyMapItemsIterator> { | ||||
self.dirstate_map(py).copymapitemsiter(py) | ||||
} | ||||
def iteritems(&self) -> PyResult<CopyMapItemsIterator> { | ||||
self.dirstate_map(py).copymapitemsiter(py) | ||||
} | ||||
def __setitem__( | ||||
&self, | ||||
key: PyObject, | ||||
item: PyObject | ||||
) -> PyResult<()> { | ||||
self.dirstate_map(py).copymapsetitem(py, key, item)?; | ||||
Ok(()) | ||||
} | ||||
def copy(&self) -> PyResult<PyDict> { | ||||
self.dirstate_map(py).copymapcopy(py) | ||||
} | ||||
}); | ||||
impl CopyMap { | ||||
pub fn from_inner(py: Python, dm: DirstateMap) -> PyResult<Self> { | ||||
Self::create_instance(py, dm) | ||||
} | ||||
fn translate_key( | ||||
py: Python, | ||||
res: (&Vec<u8>, &Vec<u8>), | ||||
) -> PyResult<Option<PyBytes>> { | ||||
Ok(Some(PyBytes::new(py, res.0))) | ||||
} | ||||
fn translate_key_value( | ||||
py: Python, | ||||
res: (&Vec<u8>, &Vec<u8>), | ||||
) -> PyResult<Option<(PyBytes, PyBytes)>> { | ||||
let (k, v) = res; | ||||
Ok(Some((PyBytes::new(py, k), PyBytes::new(py, v)))) | ||||
} | ||||
} | ||||
|
r43159 | py_shared_iterator!( | ||
|
r42999 | CopyMapKeysIterator, | ||
DirstateMapLeakedRef, | ||||
|
r43158 | CopyMapIter<'static>, | ||
|
r42999 | CopyMap::translate_key, | ||
Option<PyBytes> | ||||
); | ||||
|
r43159 | py_shared_iterator!( | ||
|
r42999 | CopyMapItemsIterator, | ||
DirstateMapLeakedRef, | ||||
|
r43158 | CopyMapIter<'static>, | ||
|
r42999 | CopyMap::translate_key_value, | ||
Option<(PyBytes, PyBytes)> | ||||
); | ||||