diff --git a/bootstrapvz/base/fs/partitionmaps/abstract.py b/bootstrapvz/base/fs/partitionmaps/abstract.py index d1e08ec..15cfb91 100644 --- a/bootstrapvz/base/fs/partitionmaps/abstract.py +++ b/bootstrapvz/base/fs/partitionmaps/abstract.py @@ -38,7 +38,7 @@ class AbstractPartitionMap(FSMProxy): """Returns the total size the partitions occupy :return: The size of all partitions - :rtype: Bytes + :rtype: Sectors """ # We just need the endpoint of the last partition return self.partitions[-1].get_end() diff --git a/bootstrapvz/base/fs/partitions/abstract.py b/bootstrapvz/base/fs/partitions/abstract.py index 038e393..10e66e4 100644 --- a/bootstrapvz/base/fs/partitions/abstract.py +++ b/bootstrapvz/base/fs/partitions/abstract.py @@ -53,7 +53,7 @@ class AbstractPartition(FSMProxy): """Gets the end of the partition :return: The end of the partition - :rtype: Bytes + :rtype: Sectors """ return self.get_start() + self.size diff --git a/bootstrapvz/base/fs/partitions/base.py b/bootstrapvz/base/fs/partitions/base.py index 019218c..23af1b8 100644 --- a/bootstrapvz/base/fs/partitions/base.py +++ b/bootstrapvz/base/fs/partitions/base.py @@ -59,7 +59,7 @@ class BasePartition(AbstractPartition): """Gets the starting byte of this partition :return: The starting byte of this partition - :rtype: Bytes + :rtype: Sectors """ if self.previous is None: # If there is no previous partition, this partition begins at the offset diff --git a/bootstrapvz/base/fs/partitions/single.py b/bootstrapvz/base/fs/partitions/single.py index 80683bc..e890493 100644 --- a/bootstrapvz/base/fs/partitions/single.py +++ b/bootstrapvz/base/fs/partitions/single.py @@ -9,8 +9,8 @@ class SinglePartition(AbstractPartition): """Gets the starting byte of this partition :return: The starting byte of this partition - :rtype: Bytes + :rtype: Sectors """ - from bootstrapvz.common.bytes import Bytes + from bootstrapvz.common.sectors import Sectors # On an unpartitioned volume there is no offset and no previous partition - return Bytes(0) + return Sectors(0, self.size.sector_size)