Como definir a classe getitem para lidar com índices simples e fatiamento?
Objetos de fatia são criados automaticamente quando você usa dois-pontos na notação de subscrito - e é para isso que é passado __getitem__
. Use isinstance
para verificar se você tem um objeto de fatia:
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)
Digamos que estivéssemos usando um objeto de intervalo, mas queremos que as fatias retornem listas em vez de novos objetos de intervalo (como acontece):
>>> range(1,100, 4)[::-1]
range(97, -3, -4)
Não podemos subclassificar o intervalo devido a limitações internas, mas podemos delegar a ele:
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)
Não temos um objeto Range perfeitamente substituível, mas é bastante próximo:
>>> r[1:3]
[1, 2]
>>> r[1]
1
>>> 2 in r
True
>>> r.count(3)
1
Para entender melhor a notação de fatia, aqui está um exemplo de uso 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, esteja ciente:
No Python 2, há um método obsoleto que você pode precisar substituir ao criar subclasses de alguns tipos integrados.
Da documentação do modelo de dados :
object.__getslice__(self, i, j)
Obsoleto desde a versão 2.0: Suporta objetos de fatia como parâmetros do __getitem__()
método. (No entanto, os tipos integrados no CPython atualmente ainda implementam __getslice__()
. Portanto, você deve substituí-los em classes derivadas ao implementar o fatiamento.)
Isso acabou no Python 3.
__getslice__
para versões 2.X do python. consulte docs.python.org/2/reference/datamodel.html#object.__getslice__