kiwi-el8/kiwi/partitioner/__init__.py
Marcus Schäfer 1a5257a0e8 Prevent use of project relative import statements
For details on the motivation of this change please visit:
https://wiki.python.org/moin/FutureProofPython
2017-03-07 11:23:37 +01:00

76 lines
2.5 KiB
Python

# Copyright (c) 2015 SUSE Linux GmbH. All rights reserved.
#
# This file is part of kiwi.
#
# kiwi is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# kiwi is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with kiwi. If not, see <http://www.gnu.org/licenses/>
#
import platform
# project
from kiwi.partitioner.gpt import PartitionerGpt
from kiwi.partitioner.msdos import PartitionerMsDos
from kiwi.partitioner.dasd import PartitionerDasd
from kiwi.exceptions import (
KiwiPartitionerSetupError
)
class Partitioner(object):
"""
Partitioner factory
Attributes
* :attr:`table_type`
Table type name
* :attr:`storage_provider`
Instance of class based on DeviceProvider
"""
def __new__(self, table_type, storage_provider):
host_architecture = platform.machine()
if host_architecture == 'x86_64':
if table_type == 'gpt':
return PartitionerGpt(storage_provider)
elif table_type == 'msdos':
return PartitionerMsDos(storage_provider)
elif host_architecture == 'i686' or host_architecture == 'i586':
if table_type == 'msdos':
return PartitionerMsDos(storage_provider)
elif 'ppc64' in host_architecture:
if table_type == 'gpt':
return PartitionerGpt(storage_provider)
elif table_type == 'msdos':
return PartitionerMsDos(storage_provider)
elif 's390' in host_architecture:
if table_type == 'dasd':
return PartitionerDasd(storage_provider)
elif table_type == 'msdos':
return PartitionerMsDos(storage_provider)
elif 'arm' in host_architecture or host_architecture == 'aarch64':
if table_type == 'gpt':
return PartitionerGpt(storage_provider)
elif table_type == 'msdos':
return PartitionerMsDos(storage_provider)
raise KiwiPartitionerSetupError(
'Support for partitioner on %s architecture not implemented' %
host_architecture
)