# 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 logging
# project
from kiwi.system.size import SystemSize
from kiwi.defaults import Defaults
from kiwi.xml_state import XMLState
log = logging.getLogger('kiwi')
[docs]
class FileSystemSetup:
"""
**Implement filesystem setup methods**
Methods from this class provides information
from the root directory required before building a
filesystem image
:param object xml_state: Instance of XMLState
:param string root_dir: root directory path
"""
def __init__(self, xml_state: XMLState, root_dir: str):
self.configured_size = xml_state.get_build_type_size(
include_unpartitioned=True
)
if xml_state.get_build_type_unpartitioned_bytes() > 0:
log.warning(
'Unpartitoned size attribute is ignored for filesystem images'
)
self.size = SystemSize(root_dir)
self.requested_image_type = xml_state.get_build_type_name()
if self.requested_image_type in Defaults.get_filesystem_image_types():
self.requested_filesystem = self.requested_image_type
else:
self.requested_filesystem = xml_state.build_type.get_filesystem()
[docs]
def get_size_mbytes(self, filesystem: str = None) -> int:
"""
Precalculate the requires size in mbytes to store all data
from the root directory in the requested filesystem. Return
the configured value if present, if not return the calculated
result
:param string filesystem: name
:return: mbytes
:rtype: int
"""
fstype = self.requested_filesystem if not filesystem else filesystem
root_dir_mbytes = self.size.accumulate_mbyte_file_sizes()
filesystem_mbytes = self.size.customize(
root_dir_mbytes, fstype
)
if not self.configured_size:
log.info(
'Using calculated size: %d MB',
filesystem_mbytes
)
return filesystem_mbytes
elif self.configured_size.additive:
result_filesystem_mbytes = \
self.configured_size.mbytes + filesystem_mbytes
log.info(
'Using configured size: %d MB + %d MB calculated = %d MB',
self.configured_size.mbytes,
filesystem_mbytes,
result_filesystem_mbytes
)
return result_filesystem_mbytes
else:
log.info(
'Using configured size: %d MB',
self.configured_size.mbytes
)
if self.configured_size.mbytes < filesystem_mbytes:
log.warning(
'--> Configured size smaller than calculated size: %d MB',
filesystem_mbytes
)
return self.configured_size.mbytes