diff --git a/neo/pt.py b/neo/pt.py index 547dd0653a4b2a8f30bdd02b6c5778564a144623..d6bcd3d208aadb6bd9ba488aeff70f0b045c5369 100644 --- a/neo/pt.py +++ b/neo/pt.py @@ -136,11 +136,11 @@ class PartitionTable(object): return [] def getCellListForTID(self, tid, readable=False, writable=False): - return self.getCellList(self._getPartitionFromIndex(u64(tid)), + return self.getCellList(self.getPartitionFromIndex(tid), readable, writable) def getCellListForOID(self, oid, readable=False, writable=False): - return self.getCellList(self._getPartitionFromIndex(u64(oid)), + return self.getCellList(self.getPartitionFromIndex(oid), readable, writable) def isAssigned(self, oid, uuid): @@ -150,6 +150,9 @@ class PartitionTable(object): return True return False + def getPartitionFromIndex(self, index): + return self._getPartitionFromIndex(u64(index)) + def _getPartitionFromIndex(self, index): return index % self.np