Comment définir la classe getitem pour gérer à la fois les index simples et le découpage?
Les objets Slice sont automatiquement créés lorsque vous utilisez un signe deux-points dans la notation en indice - et c'est ce à quoi est passé __getitem__
. Utilisez isinstance
pour vérifier si vous avez un objet tranche:
from __future__ import print_function
class Sliceable(object):
def __getitem__(self, subscript):
if isinstance(subscript, slice):
# do your handling for a slice object:
print(subscript.start, subscript.stop, subscript.step)
else:
# Do your handling for a plain index
print(subscript)
Supposons que nous utilisions un objet range, mais que nous souhaitons que les tranches renvoient des listes au lieu de nouveaux objets range (comme c'est le cas):
>>> range(1,100, 4)[::-1]
range(97, -3, -4)
Nous ne pouvons pas sous-classer range en raison de limitations internes, mais nous pouvons lui déléguer:
class Range:
"""like builtin range, but when sliced gives a list"""
__slots__ = "_range"
def __init__(self, *args):
self._range = range(*args) # takes no keyword arguments.
def __getattr__(self, name):
return getattr(self._range, name)
def __getitem__(self, subscript):
result = self._range.__getitem__(subscript)
if isinstance(subscript, slice):
return list(result)
else:
return result
r = Range(100)
Nous n'avons pas d'objet Range parfaitement remplaçable, mais c'est assez proche:
>>> r[1:3]
[1, 2]
>>> r[1]
1
>>> 2 in r
True
>>> r.count(3)
1
Pour mieux comprendre la notation de tranche, voici un exemple d'utilisation de Sliceable:
>>> sliceme = Sliceable()
>>> sliceme[1]
1
>>> sliceme[2]
2
>>> sliceme[:]
None None None
>>> sliceme[1:]
1 None None
>>> sliceme[1:2]
1 2 None
>>> sliceme[1:2:3]
1 2 3
>>> sliceme[:2:3]
None 2 3
>>> sliceme[::3]
None None 3
>>> sliceme[::]
None None None
>>> sliceme[:]
None None None
Python 2, soyez conscient:
Dans Python 2, il existe une méthode obsolète que vous devrez peut-être remplacer lors du sous-classement de certains types intégrés.
À partir de la documentation du modèle de données :
object.__getslice__(self, i, j)
Obsolète depuis la version 2.0: prend en charge les objets de tranche en tant que paramètres de la __getitem__()
méthode. (Cependant, les types intégrés dans CPython sont toujours implémentés __getslice__()
. Par conséquent, vous devez le remplacer dans les classes dérivées lors de l'implémentation du découpage.)
Cela a disparu dans Python 3.
__getslice__
versions de python 2.X. voir docs.python.org/2/reference/datamodel.html#object.__getslice__