Nuitka-winsvc 1.8__cp311-cp311-win_amd64.whl → 2.5.1__cp311-cp311-win_amd64.whl

nuitka/code_generation/Contexts.py

@@ -1,20 +1,6 @@
-#     Copyright 2023, Kay Hayen, mailto:kay.hayen@gmail.com
-#
-#     Part of "Nuitka", an optimizing Python compiler that is compatible and
-#     integrates with CPython, but also works on its own.
-#
-#     Licensed under the Apache License, Version 2.0 (the "License");
-#     you may not use this file except in compliance with the License.
-#     You may obtain a copy of the License at
-#
-#        http://www.apache.org/licenses/LICENSE-2.0
-#
-#     Unless required by applicable law or agreed to in writing, software
-#     distributed under the License is distributed on an "AS IS" BASIS,
-#     WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-#     See the License for the specific language governing permissions and
-#     limitations under the License.
-#
+#     Copyright 2024, Kay Hayen, mailto:kay.hayen@gmail.com find license text at end of file
+
+
 """ Code generation contexts.
 
 """
@@ -24,7 +10,7 @@ from abc import abstractmethod
 from contextlib import contextmanager
 
 from nuitka import Options
-from nuitka.__past__ import getMetaClassBase, iterItems
+from nuitka.__past__ import iterItems
 from nuitka.Constants import isMutable
 from nuitka.PythonVersions import python_version
 from nuitka.Serialization import ConstantAccessor
@@ -34,6 +20,7 @@ from nuitka.utils.InstanceCounters import (
     counted_init,
     isCountingInstances,
 )
+from nuitka.utils.SlotMetaClasses import getMetaClassBase
 
 from .VariableDeclarations import VariableDeclaration, VariableStorage
 
@@ -44,6 +31,9 @@ class TempMixin(object):
     # Lots of details, everything gets to store bits here, to indicate
     # code generation states, and there are many, pylint: disable=too-many-instance-attributes
 
+    # Mixins are not allowed to specify slots, pylint: disable=assigning-non-slot
+    __slots__ = ()
+
     def __init__(self):
         self.tmp_names = {}
 
@@ -59,7 +49,7 @@ class TempMixin(object):
         self.false_target = None
 
         self.keeper_variable_count = 0
-        self.exception_keepers = (None, None, None, None)
+        self.exception_keepers = (None, None)
 
         self.preserver_variable_declaration = {}
 
@@ -75,6 +65,9 @@ class TempMixin(object):
         # We might be hard coding too many details for special temps
         # here, pylint: disable=too-many-branches
 
+        # TODO: This should be achieved.
+        # assert not base_name.startswith("tmp_"), base_name
+
         if unique:
             number = None
         else:
@@ -179,24 +172,14 @@ class TempMixin(object):
         debug = Options.is_debug and python_version >= 0x300
 
         if debug:
-            keeper_obj_init = "NULL"
+            keeper_obj_init = "Empty_Nuitka_ExceptionPreservationItem"
         else:
             keeper_obj_init = None
 
         return (
             self.variable_storage.addVariableDeclarationTop(
-                "PyObject *",
-                "exception_keeper_type_%d" % self.keeper_variable_count,
-                keeper_obj_init,
-            ),
-            self.variable_storage.addVariableDeclarationTop(
-                "PyObject *",
-                "exception_keeper_value_%d" % self.keeper_variable_count,
-                keeper_obj_init,
-            ),
-            self.variable_storage.addVariableDeclarationTop(
-                "PyTracebackObject *",
-                "exception_keeper_tb_%d" % self.keeper_variable_count,
+                "struct Nuitka_ExceptionPreservationItem",
+                "exception_keeper_name_%d" % self.keeper_variable_count,
                 keeper_obj_init,
             ),
             self.variable_storage.addVariableDeclarationTop(
@@ -212,6 +195,7 @@ class TempMixin(object):
     def setExceptionKeeperVariables(self, keeper_vars):
         result = self.exception_keepers
         self.exception_keepers = tuple(keeper_vars)
+        assert len(self.exception_keepers) == 2
         return result
 
     def addExceptionPreserverVariables(self, preserver_id):
@@ -225,12 +209,12 @@ class TempMixin(object):
             else:
                 preserver_obj_init = None
 
-            self.preserver_variable_declaration[
-                preserver_id
-            ] = self.variable_storage.addVariableDeclarationTop(
-                "struct Nuitka_ExceptionStackItem",
-                "exception_preserved_%d" % preserver_id,
-                preserver_obj_init,
+            self.preserver_variable_declaration[preserver_id] = (
+                self.variable_storage.addVariableDeclarationTop(
+                    "struct Nuitka_ExceptionStackItem",
+                    "exception_preserved_%d" % preserver_id,
+                    preserver_obj_init,
+                )
             )
 
         return self.preserver_variable_declaration[preserver_id]
@@ -301,6 +285,9 @@ CodeObjectHandle = collections.namedtuple(
 
 
 class CodeObjectsMixin(object):
+    # Mixins are not allowed to specify slots, pylint: disable=assigning-non-slot
+    __slots__ = ()
+
     def __init__(self):
         # Code objects needed made unique by a key.
         self.code_objects = {}
@@ -328,7 +315,7 @@ class CodeObjectsMixin(object):
         )
 
         if key not in self.code_objects:
-            self.code_objects[key] = "codeobj_%s" % self._calcHash(key)
+            self.code_objects[key] = "code_objects_%s" % self._calcHash(key)
 
         return self.code_objects[key]
 
@@ -336,7 +323,9 @@ class CodeObjectsMixin(object):
         return getStringHash("-".join(str(s) for s in key))
 
 
-class PythonContextBase(getMetaClassBase("Context")):
+class PythonContextBase(getMetaClassBase("Context", require_slots=True)):
+    __slots__ = ("source_ref", "current_source_ref")
+
     @counted_init
     def __init__(self):
         self.source_ref = None
@@ -370,6 +359,10 @@ class PythonContextBase(getMetaClassBase("Context")):
     def getConstantCode(self, constant, deep_check=False):
         pass
 
+    @abstractmethod
+    def addModuleInitCode(self, code):
+        pass
+
     @abstractmethod
     def getModuleCodeName(self):
         pass
@@ -522,6 +515,8 @@ class PythonContextBase(getMetaClassBase("Context")):
 class PythonChildContextBase(PythonContextBase):
     # Base classes can be abstract, pylint: disable=I0021,abstract-method
 
+    __slots__ = ("parent",)
+
     def __init__(self, parent):
         PythonContextBase.__init__(self)
 
@@ -530,6 +525,9 @@ class PythonChildContextBase(PythonContextBase):
     def getConstantCode(self, constant, deep_check=False):
         return self.parent.getConstantCode(constant, deep_check=deep_check)
 
+    def addModuleInitCode(self, code):
+        self.parent.addModuleInitCode(code)
+
     def getModuleCodeName(self):
         return self.parent.getModuleCodeName()
 
@@ -566,8 +564,14 @@ class PythonChildContextBase(PythonContextBase):
     def addFunctionCreationInfo(self, creation_info):
         return self.parent.addFunctionCreationInfo(creation_info)
 
+    def setModuleVariableAccessorCaching(self, variable_name, caching):
+        self.parent.setModuleVariableAccessorCaching(variable_name, caching)
+
 
 class FrameDeclarationsMixin(object):
+    # Mixins are not allowed to specify slots, pylint: disable=assigning-non-slot
+    __slots__ = ()
+
     def __init__(self):
         # Frame is active or not, default not.
         self.frame_variables_stack = [""]
@@ -599,7 +603,7 @@ class FrameDeclarationsMixin(object):
             )
 
         else:
-            frame_handle = code_identifier.replace("codeobj_", "frame_")
+            frame_handle = code_identifier.replace("code_objects_", "frame_")
 
             if self.frames_used > 1:
                 frame_handle += "_%d" % self.frames_used
@@ -699,6 +703,9 @@ class FrameDeclarationsMixin(object):
 
 
 class ReturnReleaseModeMixin(object):
+    # Mixins are not allowed to specify slots, pylint: disable=assigning-non-slot
+    __slots__ = ()
+
     def __init__(self):
         self.return_release_mode = False
 
@@ -722,6 +729,9 @@ class ReturnReleaseModeMixin(object):
 
 
 class ReturnValueNameMixin(object):
+    # Mixins are not allowed to specify slots, pylint: disable=assigning-non-slot
+    __slots__ = ()
+
     def __init__(self):
         self.return_name = None
 
@@ -738,9 +748,9 @@ class ReturnValueNameMixin(object):
 
 
 class PythonModuleContext(
+    FrameDeclarationsMixin,
     TempMixin,
     CodeObjectsMixin,
-    FrameDeclarationsMixin,
     ReturnReleaseModeMixin,
     ReturnValueNameMixin,
     PythonContextBase,
@@ -748,6 +758,46 @@ class PythonModuleContext(
     # Plenty of attributes, because it's storing so many different things.
     # pylint: disable=too-many-instance-attributes
 
+    __slots__ = (
+        "module",
+        "name",
+        "code_name",
+        "declaration_codes",
+        "helper_codes",
+        "frame_handle",
+        "variable_storage",
+        "function_table_entries",
+        "constant_accessor",
+        "module_init_codes",
+        "module_variable_caching",
+        # FrameDeclarationsMixin
+        "frame_variables_stack",
+        "frame_type_descriptions",
+        "frame_variable_types",
+        "frames_used",
+        "frame_stack",
+        "locals_dict_names",
+        # TempMixin:
+        "tmp_names",
+        "labels",
+        "exception_escape",
+        "loop_continue",
+        "loop_break",
+        "true_target",
+        "false_target",
+        "keeper_variable_count",
+        "exception_keepers",
+        "preserver_variable_declaration",
+        "cleanup_names",
+        # CodeObjectsMixin
+        "code_objects",
+        # ReturnReleaseModeMixin
+        "return_release_mode",
+        "return_exit",
+        # ReturnValueNameMixin
+        "return_name",
+    )
+
     def __init__(self, module, data_filename):
         PythonContextBase.__init__(self)
 
@@ -776,6 +826,10 @@ class PythonModuleContext(
             top_level_name="mod_consts", data_filename=data_filename
         )
 
+        self.module_init_codes = []
+
+        self.module_variable_caching = {}
+
     def __repr__(self):
         return "<PythonModuleContext instance for module %s>" % self.name
 
@@ -785,6 +839,18 @@ class PythonModuleContext(
     def getEntryPoint(self):
         return self.module
 
+    def setModuleVariableAccessorCaching(self, variable_name, caching):
+        if caching:
+            self.module_variable_caching[variable_name] = True
+        elif variable_name not in self.module_variable_caching:
+            self.module_variable_caching[variable_name] = False
+
+    def isModuleVariableAccessorCaching(self, variable_name):
+        return self.module_variable_caching.get(variable_name, False)
+
+    def getModuleVariableAccessors(self):
+        return self.module_variable_caching
+
     def isCompiledPythonModule(self):
         return True
 
@@ -835,6 +901,12 @@ class PythonModuleContext(
     def getConstantsCount(self):
         return self.constant_accessor.getConstantsCount()
 
+    def getModuleInitCodes(self):
+        return self.module_init_codes
+
+    def addModuleInitCode(self, code):
+        self.module_init_codes.append(code)
+
     def addFunctionCreationInfo(self, creation_info):
         self.function_table_entries.append(creation_info)
 
@@ -857,6 +929,36 @@ class PythonFunctionContext(
     ReturnValueNameMixin,
     PythonChildContextBase,
 ):
+    __slots__ = (
+        "function",
+        "frame_handle",
+        "variable_storage",
+        # FrameDeclarationsMixin
+        "frame_variables_stack",
+        "frame_type_descriptions",
+        "frame_variable_types",
+        "frames_used",
+        "frame_stack",
+        "locals_dict_names",
+        # TempMixin:
+        "tmp_names",
+        "labels",
+        "exception_escape",
+        "loop_continue",
+        "loop_break",
+        "true_target",
+        "false_target",
+        "keeper_variable_count",
+        "exception_keepers",
+        "preserver_variable_declaration",
+        "cleanup_names",
+        # ReturnReleaseModeMixin
+        "return_release_mode",
+        "return_exit",
+        # ReturnValueNameMixin
+        "return_name",
+    )
+
     def __init__(self, parent, function):
         PythonChildContextBase.__init__(self, parent=parent)
 
@@ -902,6 +1004,8 @@ class PythonFunctionContext(
 
 
 class PythonFunctionDirectContext(PythonFunctionContext):
+    __slots__ = ()
+
     @staticmethod
     def getContextObjectName():
         return None
@@ -916,6 +1020,8 @@ class PythonFunctionDirectContext(PythonFunctionContext):
 
 
 class PythonGeneratorObjectContext(PythonFunctionContext):
+    __slots__ = ()
+
     def _makeVariableStorage(self):
         return VariableStorage(heap_name="%s_heap" % self.getContextObjectName())
 
@@ -939,18 +1045,24 @@ class PythonGeneratorObjectContext(PythonFunctionContext):
 
 
 class PythonCoroutineObjectContext(PythonGeneratorObjectContext):
+    __slots__ = ()
+
     @staticmethod
     def getContextObjectName():
         return "coroutine"
 
 
 class PythonAsyncgenObjectContext(PythonGeneratorObjectContext):
+    __slots__ = ()
+
     @staticmethod
     def getContextObjectName():
         return "asyncgen"
 
 
 class PythonFunctionCreatedContext(PythonFunctionContext):
+    __slots__ = ()
+
     @staticmethod
     def getContextObjectName():
         return None
@@ -967,6 +1079,16 @@ class PythonFunctionCreatedContext(PythonFunctionContext):
 class PythonFunctionOutlineContext(
     ReturnReleaseModeMixin, ReturnValueNameMixin, PythonChildContextBase
 ):
+    __slots__ = (
+        "outline",
+        "variable_storage",
+        # ReturnReleaseModeMixin
+        "return_release_mode",
+        "return_exit",
+        # ReturnValueNameMixin
+        "return_name",
+    )
+
     def __init__(self, parent, outline):
         PythonChildContextBase.__init__(self, parent=parent)
 
@@ -1095,3 +1217,19 @@ class PythonFunctionOutlineContext(
 
     def getContextObjectName(self):
         return self.parent.getContextObjectName()
+
+
+#     Part of "Nuitka", an optimizing Python compiler that is compatible and
+#     integrates with CPython, but also works on its own.
+#
+#     Licensed under the Apache License, Version 2.0 (the "License");
+#     you may not use this file except in compliance with the License.
+#     You may obtain a copy of the License at
+#
+#        http://www.apache.org/licenses/LICENSE-2.0
+#
+#     Unless required by applicable law or agreed to in writing, software
+#     distributed under the License is distributed on an "AS IS" BASIS,
+#     WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+#     See the License for the specific language governing permissions and
+#     limitations under the License.

nuitka/code_generation/CoroutineCodes.py

@@ -1,20 +1,6 @@
-#     Copyright 2023, Kay Hayen, mailto:kay.hayen@gmail.com
-#
-#     Part of "Nuitka", an optimizing Python compiler that is compatible and
-#     integrates with CPython, but also works on its own.
-#
-#     Licensed under the Apache License, Version 2.0 (the "License");
-#     you may not use this file except in compliance with the License.
-#     You may obtain a copy of the License at
-#
-#        http://www.apache.org/licenses/LICENSE-2.0
-#
-#     Unless required by applicable law or agreed to in writing, software
-#     distributed under the License is distributed on an "AS IS" BASIS,
-#     WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-#     See the License for the specific language governing permissions and
-#     limitations under the License.
-#
+#     Copyright 2024, Kay Hayen, mailto:kay.hayen@gmail.com find license text at end of file
+
+
 """ Code to generate and interact with compiled coroutine objects.
 
 """
@@ -101,17 +87,13 @@ def getCoroutineObjectCode(
 
     if needs_exception_exit:
         (
-            exception_type,
-            exception_value,
-            exception_tb,
+            exception_state_name,
             _exception_lineno,
         ) = context.variable_storage.getExceptionVariableDescriptions()
 
         generator_exit = template_coroutine_exception_exit % {
             "function_cleanup": indented(function_cleanup),
-            "exception_type": exception_type,
-            "exception_value": exception_value,
-            "exception_tb": exception_tb,
+            "exception_state_name": exception_state_name,
         }
     else:
         generator_exit = template_coroutine_noexception_exit % {
@@ -254,3 +236,19 @@ def generateAsyncNextCode(to_name, expression, emit, context):
         )
 
         context.addCleanupTempName(result_name)
+
+
+#     Part of "Nuitka", an optimizing Python compiler that is compatible and
+#     integrates with CPython, but also works on its own.
+#
+#     Licensed under the Apache License, Version 2.0 (the "License");
+#     you may not use this file except in compliance with the License.
+#     You may obtain a copy of the License at
+#
+#        http://www.apache.org/licenses/LICENSE-2.0
+#
+#     Unless required by applicable law or agreed to in writing, software
+#     distributed under the License is distributed on an "AS IS" BASIS,
+#     WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+#     See the License for the specific language governing permissions and
+#     limitations under the License.

nuitka/code_generation/CtypesCodes.py

@@ -1,20 +1,6 @@
-#     Copyright 2023, Kay Hayen, mailto:kay.hayen@gmail.com
-#
-#     Part of "Nuitka", an optimizing Python compiler that is compatible and
-#     integrates with CPython, but also works on its own.
-#
-#     Licensed under the Apache License, Version 2.0 (the "License");
-#     you may not use this file except in compliance with the License.
-#     You may obtain a copy of the License at
-#
-#        http://www.apache.org/licenses/LICENSE-2.0
-#
-#     Unless required by applicable law or agreed to in writing, software
-#     distributed under the License is distributed on an "AS IS" BASIS,
-#     WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-#     See the License for the specific language governing permissions and
-#     limitations under the License.
-#
+#     Copyright 2024, Kay Hayen, mailto:kay.hayen@gmail.com find license text at end of file
+
+
 """ Code generation for ctypes module stuff. """
 
 from .BuiltinCodes import getBuiltinCallViaSpecCode
@@ -42,3 +28,19 @@ def generateCtypesCdllCallCode(to_name, expression, emit, context):
         emit=emit,
         context=context,
     )
+
+
+#     Part of "Nuitka", an optimizing Python compiler that is compatible and
+#     integrates with CPython, but also works on its own.
+#
+#     Licensed under the Apache License, Version 2.0 (the "License");
+#     you may not use this file except in compliance with the License.
+#     You may obtain a copy of the License at
+#
+#        http://www.apache.org/licenses/LICENSE-2.0
+#
+#     Unless required by applicable law or agreed to in writing, software
+#     distributed under the License is distributed on an "AS IS" BASIS,
+#     WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+#     See the License for the specific language governing permissions and
+#     limitations under the License.

nuitka/code_generation/DictCodes.py

@@ -1,20 +1,6 @@
-#     Copyright 2023, Kay Hayen, mailto:kay.hayen@gmail.com
-#
-#     Part of "Nuitka", an optimizing Python compiler that is compatible and
-#     integrates with CPython, but also works on its own.
-#
-#     Licensed under the Apache License, Version 2.0 (the "License");
-#     you may not use this file except in compliance with the License.
-#     You may obtain a copy of the License at
-#
-#        http://www.apache.org/licenses/LICENSE-2.0
-#
-#     Unless required by applicable law or agreed to in writing, software
-#     distributed under the License is distributed on an "AS IS" BASIS,
-#     WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-#     See the License for the specific language governing permissions and
-#     limitations under the License.
-#
+#     Copyright 2024, Kay Hayen, mailto:kay.hayen@gmail.com find license text at end of file
+
+
 """ Code generation for dictionaries.
 
 """
@@ -279,13 +265,14 @@ def generateDictOperationGet2Code(to_name, expression, emit, context):
             renderTemplateFromString(
                 r"""
 {% if expression.known_hashable_key %}
-%(value_name)s = DICT_GET_ITEM0(tstate, %(dict_name)s, %(key_name)s);
+%(value_name)s = DICT_GET_ITEM1(tstate, %(dict_name)s, %(key_name)s);
 if (%(value_name)s == NULL) {
 {% else %}
-%(value_name)s = DICT_GET_ITEM_WITH_HASH_ERROR0(tstate, %(dict_name)s, %(key_name)s);
+%(value_name)s = DICT_GET_ITEM_WITH_HASH_ERROR1(tstate, %(dict_name)s, %(key_name)s);
 if (%(value_name)s == NULL && !HAS_ERROR_OCCURRED(tstate)) {
 {% endif %}
     %(value_name)s = Py_None;
+    Py_INCREF_IMMORTAL(%(value_name)s);
 }
 """,
                 expression=expression,
@@ -305,15 +292,14 @@ if (%(value_name)s == NULL && !HAS_ERROR_OCCURRED(tstate)) {
             context=context,
         )
 
+        context.addCleanupTempName(value_name)
+
 
 def generateDictOperationGet3Code(to_name, expression, emit, context):
     dict_name, key_name, default_name = generateChildExpressionsCode(
         expression=expression, emit=emit, context=context
     )
 
-    # TODO: This code could actually make it dependent on default taking
-    # a reference or not, and then use DICT_GET_ITEM0/DICT_GET_ITEM_WITH_HASH_ERROR0 if not.
-
     with withObjectCodeTemporaryAssignment(
         to_name, "dict_value", expression, emit, context
     ) as value_name:
@@ -555,7 +541,7 @@ def generateDictOperationCopyCode(to_name, expression, emit, context):
     generateCAPIObjectCode(
         to_name=to_name,
         capi="DICT_COPY",
-        tstate=False,
+        tstate=True,
         arg_desc=(("dict_arg", expression.subnode_dict_arg),),
         may_raise=False,
         conversion_check=decideConversionCheckNeeded(to_name, expression),
@@ -600,7 +586,7 @@ def generateDictOperationIteritemsCode(to_name, expression, emit, context):
     generateCAPIObjectCode(
         to_name=to_name,
         capi="DICT_ITERITEMS",
-        tstate=False,
+        tstate=True,
         arg_desc=(("dict_arg", expression.subnode_dict_arg),),
         may_raise=expression.mayRaiseException(BaseException),
         conversion_check=decideConversionCheckNeeded(to_name, expression),
@@ -642,7 +628,7 @@ def generateDictOperationIterkeysCode(to_name, expression, emit, context):
     generateCAPIObjectCode(
         to_name=to_name,
         capi="DICT_ITERKEYS",
-        tstate=False,
+        tstate=True,
         arg_desc=(("dict_arg", expression.subnode_dict_arg),),
         may_raise=False,
         conversion_check=decideConversionCheckNeeded(to_name, expression),
@@ -729,7 +715,7 @@ def generateDictOperationItervaluesCode(to_name, expression, emit, context):
     generateCAPIObjectCode(
         to_name=to_name,
         capi="DICT_ITERVALUES",
-        tstate=False,
+        tstate=True,
         arg_desc=(("dict_arg", expression.subnode_dict_arg),),
         may_raise=False,
         conversion_check=decideConversionCheckNeeded(to_name, expression),
@@ -914,3 +900,19 @@ def generateDictOperationRemoveCode(statement, emit, context):
             emit=emit,
             context=context,
         )
+
+
+#     Part of "Nuitka", an optimizing Python compiler that is compatible and
+#     integrates with CPython, but also works on its own.
+#
+#     Licensed under the Apache License, Version 2.0 (the "License");
+#     you may not use this file except in compliance with the License.
+#     You may obtain a copy of the License at
+#
+#        http://www.apache.org/licenses/LICENSE-2.0
+#
+#     Unless required by applicable law or agreed to in writing, software
+#     distributed under the License is distributed on an "AS IS" BASIS,
+#     WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+#     See the License for the specific language governing permissions and
+#     limitations under the License.