oamg · MichalHe · Feb 12, 2025 · Feb 12, 2025 · Jan 16, 2025 · pirat89
diff --git a/leapp/__init__.py b/leapp/__init__.py
@@ -1,6 +1,4 @@
 from leapp.utils.workarounds import apply_workarounds
-
-VERSION = "0.18.0"
-FULL_VERSION = VERSION
+from leapp.__version__ import VERSION, FULL_VERSION
 
 apply_workarounds()
diff --git a/leapp/__version__.py b/leapp/__version__.py
@@ -0,0 +1,2 @@
+VERSION = "0.18.0"
+FULL_VERSION = VERSION
diff --git a/leapp/exceptions.py b/leapp/exceptions.py
@@ -4,6 +4,10 @@ def __init__(self, message):
         self.message = message
 
 
+class FrameworkInitializationError(LeappError):
+    pass
+
+
 class RepositoryConfigurationError(LeappError):
     pass
 

diff --git a/leapp/utils/workarounds/__init__.py b/leapp/utils/workarounds/__init__.py
@@ -1,7 +1,9 @@
 import leapp.utils.workarounds.mp
 import leapp.utils.workarounds.fqdn
+import leapp.utils.workarounds.resources
 
 
 def apply_workarounds():
     leapp.utils.workarounds.mp.apply_workaround()
     leapp.utils.workarounds.fqdn.apply_workaround()
+    leapp.utils.workarounds.resources.apply_workaround()
diff --git a/leapp/utils/workarounds/resources.py b/leapp/utils/workarounds/resources.py
@@ -0,0 +1,40 @@
+import resource
+from leapp.exceptions import FrameworkInitializationError
+
+
+def apply_workaround():
+    """
+    Set resource limits for the maximum number of open file descriptors and the maximum writable file size.
+
+    :raises: `CommandError` if the resource limits cannot be set
+    """
+
+    def set_resource_limit(resource_type, soft, hard):
+        rtype_string = (
+            'open file descriptors' if resource_type == resource.RLIMIT_NOFILE
+            else 'writable file size' if resource_type == resource.RLIMIT_FSIZE
+            else 'unknown resource'
+        )
+        try:
+            resource.setrlimit(resource_type, (soft, hard))
+        except ValueError as err:
+            raise FrameworkInitializationError(
+                'Unable to set the required soft limit of resource due to system restrictions. '
+                'This limit is necessary to prevent the program from crashin. '
-                'This limit is necessary to prevent the program from crashin. '
+                'This limit is necessary to prevent the program from crashing. '
-                'This limit is necessary to prevent the program from crashin. '
+                'This limit is necessary to prevent the program from crashing. '
+                'Resource type: {}, error: {}'.format(rtype_string, err)
+            )
+        except OSError as err:
+            raise FrameworkInitializationError(
+                'Failed to set resource limit. Resource type: {}, error: {}'.format(rtype_string, err)
+            )
+
+    soft_nofile, _ = resource.getrlimit(resource.RLIMIT_NOFILE)
+    soft_fsize, _ = resource.getrlimit(resource.RLIMIT_FSIZE)
+    nofile_limit = 1024 * 16
+    fsize_limit = resource.RLIM_INFINITY
+
+    if soft_nofile < nofile_limit:
+        set_resource_limit(resource.RLIMIT_NOFILE, nofile_limit, nofile_limit)
+
+    if soft_fsize != fsize_limit:
+        set_resource_limit(resource.RLIMIT_FSIZE, fsize_limit, fsize_limit)
diff --git a/setup.py b/setup.py
@@ -4,7 +4,7 @@
 from distutils.util import convert_path
 
 main_ns = {}
-ver_path = convert_path('leapp/__init__.py')
+ver_path = convert_path('leapp/__version__.py')
 with open(ver_path) as ver_file:
     exec(ver_file.read(), main_ns)
-Original file line number
+Diff line change
@@ Expand Up / @@ -4,6 +4,10 @@ def __init__(self, message): @@
             self.message = message
+    class FrameworkInitializationError(LeappError):
+        pass
     class RepositoryConfigurationError(LeappError):
         pass
@@ Expand Down @@