2761 lines
		
	
	
		
			80 KiB
		
	
	
	
		
			Python
		
	
	
	
	
	
			
		
		
	
	
			2761 lines
		
	
	
		
			80 KiB
		
	
	
	
		
			Python
		
	
	
	
	
	
| # ##### BEGIN GPL LICENSE BLOCK #####
 | |
| #
 | |
| #  This program 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 2
 | |
| #  of the License, or (at your option) any later version.
 | |
| #
 | |
| #  This program 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 this program; if not, write to the Free Software Foundation,
 | |
| #  Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
 | |
| #
 | |
| # ##### END GPL LICENSE BLOCK #####
 | |
| 
 | |
| # <pep8 compliant>
 | |
| 
 | |
| import bpy
 | |
| from bpy.types import (
 | |
|     Menu,
 | |
|     Operator,
 | |
|     OperatorFileListElement
 | |
| )
 | |
| from bpy.props import (
 | |
|     BoolProperty,
 | |
|     EnumProperty,
 | |
|     FloatProperty,
 | |
|     IntProperty,
 | |
|     StringProperty,
 | |
|     CollectionProperty,
 | |
| )
 | |
| 
 | |
| from bpy.app.translations import pgettext_tip as tip_
 | |
| 
 | |
| 
 | |
| rna_path_prop = StringProperty(
 | |
|     name="Context Attributes",
 | |
|     description="RNA context string",
 | |
|     maxlen=1024,
 | |
| )
 | |
| 
 | |
| rna_reverse_prop = BoolProperty(
 | |
|     name="Reverse",
 | |
|     description="Cycle backwards",
 | |
|     default=False,
 | |
| )
 | |
| 
 | |
| rna_wrap_prop = BoolProperty(
 | |
|     name="Wrap",
 | |
|     description="Wrap back to the first/last values",
 | |
|     default=False,
 | |
| )
 | |
| 
 | |
| rna_relative_prop = BoolProperty(
 | |
|     name="Relative",
 | |
|     description="Apply relative to the current value (delta)",
 | |
|     default=False,
 | |
| )
 | |
| 
 | |
| rna_space_type_prop = EnumProperty(
 | |
|     name="Type",
 | |
|     items=tuple(
 | |
|         (e.identifier, e.name, "", e. value)
 | |
|         for e in bpy.types.Space.bl_rna.properties["type"].enum_items
 | |
|     ),
 | |
|     default='EMPTY',
 | |
| )
 | |
| 
 | |
| 
 | |
| def context_path_validate(context, data_path):
 | |
|     try:
 | |
|         value = eval("context.%s" % data_path) if data_path else Ellipsis
 | |
|     except AttributeError as ex:
 | |
|         if str(ex).startswith("'NoneType'"):
 | |
|             # One of the items in the rna path is None, just ignore this
 | |
|             value = Ellipsis
 | |
|         else:
 | |
|             # We have a real error in the rna path, don't ignore that
 | |
|             raise
 | |
| 
 | |
|     return value
 | |
| 
 | |
| 
 | |
| def operator_value_is_undo(value):
 | |
|     if value in {None, Ellipsis}:
 | |
|         return False
 | |
| 
 | |
|     # typical properties or objects
 | |
|     id_data = getattr(value, "id_data", Ellipsis)
 | |
| 
 | |
|     if id_data is None:
 | |
|         return False
 | |
|     elif id_data is Ellipsis:
 | |
|         # handle mathutils types
 | |
|         id_data = getattr(getattr(value, "owner", None), "id_data", None)
 | |
| 
 | |
|         if id_data is None:
 | |
|             return False
 | |
| 
 | |
|     # return True if its a non window ID type
 | |
|     return (isinstance(id_data, bpy.types.ID) and
 | |
|             (not isinstance(id_data, (bpy.types.WindowManager,
 | |
|                                       bpy.types.Screen,
 | |
|                                       bpy.types.Brush,
 | |
|                                       ))))
 | |
| 
 | |
| 
 | |
| def operator_path_is_undo(context, data_path):
 | |
|     # note that if we have data paths that use strings this could fail
 | |
|     # luckily we don't do this!
 | |
|     #
 | |
|     # When we can't find the data owner assume no undo is needed.
 | |
|     data_path_head = data_path.rpartition(".")[0]
 | |
| 
 | |
|     if not data_path_head:
 | |
|         return False
 | |
| 
 | |
|     value = context_path_validate(context, data_path_head)
 | |
| 
 | |
|     return operator_value_is_undo(value)
 | |
| 
 | |
| 
 | |
| def operator_path_undo_return(context, data_path):
 | |
|     return {'FINISHED'} if operator_path_is_undo(context, data_path) else {'CANCELLED'}
 | |
| 
 | |
| 
 | |
| def operator_value_undo_return(value):
 | |
|     return {'FINISHED'} if operator_value_is_undo(value) else {'CANCELLED'}
 | |
| 
 | |
| 
 | |
| def execute_context_assign(self, context):
 | |
|     data_path = self.data_path
 | |
|     if context_path_validate(context, data_path) is Ellipsis:
 | |
|         return {'PASS_THROUGH'}
 | |
| 
 | |
|     if getattr(self, "relative", False):
 | |
|         exec("context.%s += self.value" % data_path)
 | |
|     else:
 | |
|         exec("context.%s = self.value" % data_path)
 | |
| 
 | |
|     return operator_path_undo_return(context, data_path)
 | |
| 
 | |
| 
 | |
| def module_filesystem_remove(path_base, module_name):
 | |
|     import os
 | |
|     module_name = os.path.splitext(module_name)[0]
 | |
|     for f in os.listdir(path_base):
 | |
|         f_base = os.path.splitext(f)[0]
 | |
|         if f_base == module_name:
 | |
|             f_full = os.path.join(path_base, f)
 | |
| 
 | |
|             if os.path.isdir(f_full):
 | |
|                 os.rmdir(f_full)
 | |
|             else:
 | |
|                 os.remove(f_full)
 | |
| 
 | |
| 
 | |
| class BRUSH_OT_active_index_set(Operator):
 | |
|     """Set active sculpt/paint brush from it's number"""
 | |
|     bl_idname = "brush.active_index_set"
 | |
|     bl_label = "Set Brush Number"
 | |
| 
 | |
|     mode: StringProperty(
 | |
|         name="Mode",
 | |
|         description="Paint mode to set brush for",
 | |
|         maxlen=1024,
 | |
|     )
 | |
|     index: IntProperty(
 | |
|         name="Number",
 | |
|         description="Brush number",
 | |
|     )
 | |
| 
 | |
|     _attr_dict = {
 | |
|         "sculpt": "use_paint_sculpt",
 | |
|         "vertex_paint": "use_paint_vertex",
 | |
|         "weight_paint": "use_paint_weight",
 | |
|         "image_paint": "use_paint_image",
 | |
|     }
 | |
| 
 | |
|     def execute(self, context):
 | |
|         attr = self._attr_dict.get(self.mode)
 | |
|         if attr is None:
 | |
|             return {'CANCELLED'}
 | |
| 
 | |
|         toolsettings = context.tool_settings
 | |
|         for i, brush in enumerate((cur for cur in bpy.data.brushes if getattr(cur, attr))):
 | |
|             if i == self.index:
 | |
|                 getattr(toolsettings, self.mode).brush = brush
 | |
|                 return {'FINISHED'}
 | |
| 
 | |
|         return {'CANCELLED'}
 | |
| 
 | |
| 
 | |
| class WM_OT_context_set_boolean(Operator):
 | |
|     """Set a context value"""
 | |
|     bl_idname = "wm.context_set_boolean"
 | |
|     bl_label = "Context Set Boolean"
 | |
|     bl_options = {'UNDO', 'INTERNAL'}
 | |
| 
 | |
|     data_path: rna_path_prop
 | |
|     value: BoolProperty(
 | |
|         name="Value",
 | |
|         description="Assignment value",
 | |
|         default=True,
 | |
|     )
 | |
| 
 | |
|     execute = execute_context_assign
 | |
| 
 | |
| 
 | |
| class WM_OT_context_set_int(Operator):  # same as enum
 | |
|     """Set a context value"""
 | |
|     bl_idname = "wm.context_set_int"
 | |
|     bl_label = "Context Set"
 | |
|     bl_options = {'UNDO', 'INTERNAL'}
 | |
| 
 | |
|     data_path: rna_path_prop
 | |
|     value: IntProperty(
 | |
|         name="Value",
 | |
|         description="Assign value",
 | |
|         default=0,
 | |
|     )
 | |
|     relative: rna_relative_prop
 | |
| 
 | |
|     execute = execute_context_assign
 | |
| 
 | |
| 
 | |
| class WM_OT_context_scale_float(Operator):
 | |
|     """Scale a float context value"""
 | |
|     bl_idname = "wm.context_scale_float"
 | |
|     bl_label = "Context Scale Float"
 | |
|     bl_options = {'UNDO', 'INTERNAL'}
 | |
| 
 | |
|     data_path: rna_path_prop
 | |
|     value: FloatProperty(
 | |
|         name="Value",
 | |
|         description="Assign value",
 | |
|         default=1.0,
 | |
|     )
 | |
| 
 | |
|     def execute(self, context):
 | |
|         data_path = self.data_path
 | |
|         if context_path_validate(context, data_path) is Ellipsis:
 | |
|             return {'PASS_THROUGH'}
 | |
| 
 | |
|         value = self.value
 | |
| 
 | |
|         if value == 1.0:  # nothing to do
 | |
|             return {'CANCELLED'}
 | |
| 
 | |
|         exec("context.%s *= value" % data_path)
 | |
| 
 | |
|         return operator_path_undo_return(context, data_path)
 | |
| 
 | |
| 
 | |
| class WM_OT_context_scale_int(Operator):
 | |
|     """Scale an int context value"""
 | |
|     bl_idname = "wm.context_scale_int"
 | |
|     bl_label = "Context Scale Int"
 | |
|     bl_options = {'UNDO', 'INTERNAL'}
 | |
| 
 | |
|     data_path: rna_path_prop
 | |
|     value: FloatProperty(
 | |
|         name="Value",
 | |
|         description="Assign value",
 | |
|         default=1.0,
 | |
|     )
 | |
|     always_step: BoolProperty(
 | |
|         name="Always Step",
 | |
|         description="Always adjust the value by a minimum of 1 when 'value' is not 1.0",
 | |
|         default=True,
 | |
|     )
 | |
| 
 | |
|     def execute(self, context):
 | |
|         data_path = self.data_path
 | |
|         if context_path_validate(context, data_path) is Ellipsis:
 | |
|             return {'PASS_THROUGH'}
 | |
| 
 | |
|         value = self.value
 | |
| 
 | |
|         if value == 1.0:  # nothing to do
 | |
|             return {'CANCELLED'}
 | |
| 
 | |
|         if getattr(self, "always_step", False):
 | |
|             if value > 1.0:
 | |
|                 add = "1"
 | |
|                 func = "max"
 | |
|             else:
 | |
|                 add = "-1"
 | |
|                 func = "min"
 | |
|             exec("context.%s = %s(round(context.%s * value), context.%s + %s)" %
 | |
|                  (data_path, func, data_path, data_path, add))
 | |
|         else:
 | |
|             exec("context.%s *= value" % data_path)
 | |
| 
 | |
|         return operator_path_undo_return(context, data_path)
 | |
| 
 | |
| 
 | |
| class WM_OT_context_set_float(Operator):  # same as enum
 | |
|     """Set a context value"""
 | |
|     bl_idname = "wm.context_set_float"
 | |
|     bl_label = "Context Set Float"
 | |
|     bl_options = {'UNDO', 'INTERNAL'}
 | |
| 
 | |
|     data_path: rna_path_prop
 | |
|     value: FloatProperty(
 | |
|         name="Value",
 | |
|         description="Assignment value",
 | |
|         default=0.0,
 | |
|     )
 | |
|     relative: rna_relative_prop
 | |
| 
 | |
|     execute = execute_context_assign
 | |
| 
 | |
| 
 | |
| class WM_OT_context_set_string(Operator):  # same as enum
 | |
|     """Set a context value"""
 | |
|     bl_idname = "wm.context_set_string"
 | |
|     bl_label = "Context Set String"
 | |
|     bl_options = {'UNDO', 'INTERNAL'}
 | |
| 
 | |
|     data_path: rna_path_prop
 | |
|     value: StringProperty(
 | |
|         name="Value",
 | |
|         description="Assign value",
 | |
|         maxlen=1024,
 | |
|     )
 | |
| 
 | |
|     execute = execute_context_assign
 | |
| 
 | |
| 
 | |
| class WM_OT_context_set_enum(Operator):
 | |
|     """Set a context value"""
 | |
|     bl_idname = "wm.context_set_enum"
 | |
|     bl_label = "Context Set Enum"
 | |
|     bl_options = {'UNDO', 'INTERNAL'}
 | |
| 
 | |
|     data_path: rna_path_prop
 | |
|     value: StringProperty(
 | |
|         name="Value",
 | |
|         description="Assignment value (as a string)",
 | |
|         maxlen=1024,
 | |
|     )
 | |
| 
 | |
|     execute = execute_context_assign
 | |
| 
 | |
| 
 | |
| class WM_OT_context_set_value(Operator):
 | |
|     """Set a context value"""
 | |
|     bl_idname = "wm.context_set_value"
 | |
|     bl_label = "Context Set Value"
 | |
|     bl_options = {'UNDO', 'INTERNAL'}
 | |
| 
 | |
|     data_path: rna_path_prop
 | |
|     value: StringProperty(
 | |
|         name="Value",
 | |
|         description="Assignment value (as a string)",
 | |
|         maxlen=1024,
 | |
|     )
 | |
| 
 | |
|     def execute(self, context):
 | |
|         data_path = self.data_path
 | |
|         if context_path_validate(context, data_path) is Ellipsis:
 | |
|             return {'PASS_THROUGH'}
 | |
|         exec("context.%s = %s" % (data_path, self.value))
 | |
|         return operator_path_undo_return(context, data_path)
 | |
| 
 | |
| 
 | |
| class WM_OT_context_toggle(Operator):
 | |
|     """Toggle a context value"""
 | |
|     bl_idname = "wm.context_toggle"
 | |
|     bl_label = "Context Toggle"
 | |
|     bl_options = {'UNDO', 'INTERNAL'}
 | |
| 
 | |
|     data_path: rna_path_prop
 | |
| 
 | |
|     def execute(self, context):
 | |
|         data_path = self.data_path
 | |
| 
 | |
|         if context_path_validate(context, data_path) is Ellipsis:
 | |
|             return {'PASS_THROUGH'}
 | |
| 
 | |
|         exec("context.%s = not (context.%s)" % (data_path, data_path))
 | |
| 
 | |
|         return operator_path_undo_return(context, data_path)
 | |
| 
 | |
| 
 | |
| class WM_OT_context_toggle_enum(Operator):
 | |
|     """Toggle a context value"""
 | |
|     bl_idname = "wm.context_toggle_enum"
 | |
|     bl_label = "Context Toggle Values"
 | |
|     bl_options = {'UNDO', 'INTERNAL'}
 | |
| 
 | |
|     data_path: rna_path_prop
 | |
|     value_1: StringProperty(
 | |
|         name="Value",
 | |
|         description="Toggle enum",
 | |
|         maxlen=1024,
 | |
|     )
 | |
|     value_2: StringProperty(
 | |
|         name="Value",
 | |
|         description="Toggle enum",
 | |
|         maxlen=1024,
 | |
|     )
 | |
| 
 | |
|     def execute(self, context):
 | |
|         data_path = self.data_path
 | |
| 
 | |
|         if context_path_validate(context, data_path) is Ellipsis:
 | |
|             return {'PASS_THROUGH'}
 | |
| 
 | |
|         # failing silently is not ideal, but we don't want errors for shortcut
 | |
|         # keys that some values that are only available in a particular context
 | |
|         try:
 | |
|             exec("context.%s = ('%s', '%s')[context.%s != '%s']" %
 | |
|                  (data_path, self.value_1,
 | |
|                   self.value_2, data_path,
 | |
|                   self.value_2,
 | |
|                   ))
 | |
|         except:
 | |
|             return {'PASS_THROUGH'}
 | |
| 
 | |
|         return operator_path_undo_return(context, data_path)
 | |
| 
 | |
| 
 | |
| class WM_OT_context_cycle_int(Operator):
 | |
|     """Set a context value (useful for cycling active material, """ \
 | |
|         """vertex keys, groups, etc.)"""
 | |
|     bl_idname = "wm.context_cycle_int"
 | |
|     bl_label = "Context Int Cycle"
 | |
|     bl_options = {'UNDO', 'INTERNAL'}
 | |
| 
 | |
|     data_path: rna_path_prop
 | |
|     reverse: rna_reverse_prop
 | |
|     wrap: rna_wrap_prop
 | |
| 
 | |
|     def execute(self, context):
 | |
|         data_path = self.data_path
 | |
|         value = context_path_validate(context, data_path)
 | |
|         if value is Ellipsis:
 | |
|             return {'PASS_THROUGH'}
 | |
| 
 | |
|         if self.reverse:
 | |
|             value -= 1
 | |
|         else:
 | |
|             value += 1
 | |
| 
 | |
|         exec("context.%s = value" % data_path)
 | |
| 
 | |
|         if self.wrap:
 | |
|             if value != eval("context.%s" % data_path):
 | |
|                 # relies on rna clamping integers out of the range
 | |
|                 if self.reverse:
 | |
|                     value = (1 << 31) - 1
 | |
|                 else:
 | |
|                     value = -1 << 31
 | |
| 
 | |
|                 exec("context.%s = value" % data_path)
 | |
| 
 | |
|         return operator_path_undo_return(context, data_path)
 | |
| 
 | |
| 
 | |
| class WM_OT_context_cycle_enum(Operator):
 | |
|     """Toggle a context value"""
 | |
|     bl_idname = "wm.context_cycle_enum"
 | |
|     bl_label = "Context Enum Cycle"
 | |
|     bl_options = {'UNDO', 'INTERNAL'}
 | |
| 
 | |
|     data_path: rna_path_prop
 | |
|     reverse: rna_reverse_prop
 | |
|     wrap: rna_wrap_prop
 | |
| 
 | |
|     def execute(self, context):
 | |
|         data_path = self.data_path
 | |
|         value = context_path_validate(context, data_path)
 | |
|         if value is Ellipsis:
 | |
|             return {'PASS_THROUGH'}
 | |
| 
 | |
|         orig_value = value
 | |
| 
 | |
|         # Have to get rna enum values
 | |
|         rna_struct_str, rna_prop_str = data_path.rsplit('.', 1)
 | |
|         i = rna_prop_str.find('[')
 | |
| 
 | |
|         # just in case we get "context.foo.bar[0]"
 | |
|         if i != -1:
 | |
|             rna_prop_str = rna_prop_str[0:i]
 | |
| 
 | |
|         rna_struct = eval("context.%s.rna_type" % rna_struct_str)
 | |
| 
 | |
|         rna_prop = rna_struct.properties[rna_prop_str]
 | |
| 
 | |
|         if type(rna_prop) != bpy.types.EnumProperty:
 | |
|             raise Exception("expected an enum property")
 | |
| 
 | |
|         enums = rna_struct.properties[rna_prop_str].enum_items.keys()
 | |
|         orig_index = enums.index(orig_value)
 | |
| 
 | |
|         # Have the info we need, advance to the next item.
 | |
|         #
 | |
|         # When wrap's disabled we may set the value to its self,
 | |
|         # this is done to ensure update callbacks run.
 | |
|         if self.reverse:
 | |
|             if orig_index == 0:
 | |
|                 advance_enum = enums[-1] if self.wrap else enums[0]
 | |
|             else:
 | |
|                 advance_enum = enums[orig_index - 1]
 | |
|         else:
 | |
|             if orig_index == len(enums) - 1:
 | |
|                 advance_enum = enums[0] if self.wrap else enums[-1]
 | |
|             else:
 | |
|                 advance_enum = enums[orig_index + 1]
 | |
| 
 | |
|         # set the new value
 | |
|         exec("context.%s = advance_enum" % data_path)
 | |
|         return operator_path_undo_return(context, data_path)
 | |
| 
 | |
| 
 | |
| class WM_OT_context_cycle_array(Operator):
 | |
|     """Set a context array value """ \
 | |
|         """(useful for cycling the active mesh edit mode)"""
 | |
|     bl_idname = "wm.context_cycle_array"
 | |
|     bl_label = "Context Array Cycle"
 | |
|     bl_options = {'UNDO', 'INTERNAL'}
 | |
| 
 | |
|     data_path: rna_path_prop
 | |
|     reverse: rna_reverse_prop
 | |
| 
 | |
|     def execute(self, context):
 | |
|         data_path = self.data_path
 | |
|         value = context_path_validate(context, data_path)
 | |
|         if value is Ellipsis:
 | |
|             return {'PASS_THROUGH'}
 | |
| 
 | |
|         def cycle(array):
 | |
|             if self.reverse:
 | |
|                 array.insert(0, array.pop())
 | |
|             else:
 | |
|                 array.append(array.pop(0))
 | |
|             return array
 | |
| 
 | |
|         exec("context.%s = cycle(context.%s[:])" % (data_path, data_path))
 | |
| 
 | |
|         return operator_path_undo_return(context, data_path)
 | |
| 
 | |
| 
 | |
| class WM_OT_context_menu_enum(Operator):
 | |
|     bl_idname = "wm.context_menu_enum"
 | |
|     bl_label = "Context Enum Menu"
 | |
|     bl_options = {'UNDO', 'INTERNAL'}
 | |
| 
 | |
|     data_path: rna_path_prop
 | |
| 
 | |
|     def execute(self, context):
 | |
|         data_path = self.data_path
 | |
|         value = context_path_validate(context, data_path)
 | |
| 
 | |
|         if value is Ellipsis:
 | |
|             return {'PASS_THROUGH'}
 | |
| 
 | |
|         base_path, prop_string = data_path.rsplit(".", 1)
 | |
|         value_base = context_path_validate(context, base_path)
 | |
|         prop = value_base.bl_rna.properties[prop_string]
 | |
| 
 | |
|         def draw_cb(self, context):
 | |
|             layout = self.layout
 | |
|             layout.prop(value_base, prop_string, expand=True)
 | |
| 
 | |
|         context.window_manager.popup_menu(draw_func=draw_cb, title=prop.name, icon=prop.icon)
 | |
| 
 | |
|         return {'FINISHED'}
 | |
| 
 | |
| 
 | |
| class WM_OT_context_pie_enum(Operator):
 | |
|     bl_idname = "wm.context_pie_enum"
 | |
|     bl_label = "Context Enum Pie"
 | |
|     bl_options = {'UNDO', 'INTERNAL'}
 | |
| 
 | |
|     data_path: rna_path_prop
 | |
| 
 | |
|     def invoke(self, context, event):
 | |
|         wm = context.window_manager
 | |
|         data_path = self.data_path
 | |
|         value = context_path_validate(context, data_path)
 | |
| 
 | |
|         if value is Ellipsis:
 | |
|             return {'PASS_THROUGH'}
 | |
| 
 | |
|         base_path, prop_string = data_path.rsplit(".", 1)
 | |
|         value_base = context_path_validate(context, base_path)
 | |
|         prop = value_base.bl_rna.properties[prop_string]
 | |
| 
 | |
|         def draw_cb(self, context):
 | |
|             layout = self.layout
 | |
|             layout.prop(value_base, prop_string, expand=True)
 | |
| 
 | |
|         wm.popup_menu_pie(draw_func=draw_cb, title=prop.name, icon=prop.icon, event=event)
 | |
| 
 | |
|         return {'FINISHED'}
 | |
| 
 | |
| 
 | |
| class WM_OT_operator_pie_enum(Operator):
 | |
|     bl_idname = "wm.operator_pie_enum"
 | |
|     bl_label = "Operator Enum Pie"
 | |
|     bl_options = {'UNDO', 'INTERNAL'}
 | |
| 
 | |
|     data_path: StringProperty(
 | |
|         name="Operator",
 | |
|         description="Operator name (in python as string)",
 | |
|         maxlen=1024,
 | |
|     )
 | |
|     prop_string: StringProperty(
 | |
|         name="Property",
 | |
|         description="Property name (as a string)",
 | |
|         maxlen=1024,
 | |
|     )
 | |
| 
 | |
|     def invoke(self, context, event):
 | |
|         wm = context.window_manager
 | |
| 
 | |
|         data_path = self.data_path
 | |
|         prop_string = self.prop_string
 | |
| 
 | |
|         # same as eval("bpy.ops." + data_path)
 | |
|         op_mod_str, ob_id_str = data_path.split(".", 1)
 | |
|         op = getattr(getattr(bpy.ops, op_mod_str), ob_id_str)
 | |
|         del op_mod_str, ob_id_str
 | |
| 
 | |
|         try:
 | |
|             op_rna = op.get_rna_type()
 | |
|         except KeyError:
 | |
|             self.report({'ERROR'}, "Operator not found: bpy.ops.%s" % data_path)
 | |
|             return {'CANCELLED'}
 | |
| 
 | |
|         def draw_cb(self, context):
 | |
|             layout = self.layout
 | |
|             pie = layout.menu_pie()
 | |
|             pie.operator_enum(data_path, prop_string)
 | |
| 
 | |
|         wm.popup_menu_pie(draw_func=draw_cb, title=op_rna.name, event=event)
 | |
| 
 | |
|         return {'FINISHED'}
 | |
| 
 | |
| 
 | |
| class WM_OT_context_set_id(Operator):
 | |
|     """Set a context value to an ID data-block"""
 | |
|     bl_idname = "wm.context_set_id"
 | |
|     bl_label = "Set Library ID"
 | |
|     bl_options = {'UNDO', 'INTERNAL'}
 | |
| 
 | |
|     data_path: rna_path_prop
 | |
|     value: StringProperty(
 | |
|         name="Value",
 | |
|         description="Assign value",
 | |
|         maxlen=1024,
 | |
|     )
 | |
| 
 | |
|     def execute(self, context):
 | |
|         value = self.value
 | |
|         data_path = self.data_path
 | |
| 
 | |
|         # match the pointer type from the target property to bpy.data.*
 | |
|         # so we lookup the correct list.
 | |
|         data_path_base, data_path_prop = data_path.rsplit(".", 1)
 | |
|         data_prop_rna = eval("context.%s" % data_path_base).rna_type.properties[data_path_prop]
 | |
|         data_prop_rna_type = data_prop_rna.fixed_type
 | |
| 
 | |
|         id_iter = None
 | |
| 
 | |
|         for prop in bpy.data.rna_type.properties:
 | |
|             if prop.rna_type.identifier == "CollectionProperty":
 | |
|                 if prop.fixed_type == data_prop_rna_type:
 | |
|                     id_iter = prop.identifier
 | |
|                     break
 | |
| 
 | |
|         if id_iter:
 | |
|             value_id = getattr(bpy.data, id_iter).get(value)
 | |
|             exec("context.%s = value_id" % data_path)
 | |
| 
 | |
|         return operator_path_undo_return(context, data_path)
 | |
| 
 | |
| 
 | |
| doc_id = StringProperty(
 | |
|     name="Doc ID",
 | |
|     maxlen=1024,
 | |
|     options={'HIDDEN'},
 | |
| )
 | |
| 
 | |
| data_path_iter = StringProperty(
 | |
|     description="The data path relative to the context, must point to an iterable")
 | |
| 
 | |
| data_path_item = StringProperty(
 | |
|     description="The data path from each iterable to the value (int or float)")
 | |
| 
 | |
| 
 | |
| class WM_OT_context_collection_boolean_set(Operator):
 | |
|     """Set boolean values for a collection of items"""
 | |
|     bl_idname = "wm.context_collection_boolean_set"
 | |
|     bl_label = "Context Collection Boolean Set"
 | |
|     bl_options = {'UNDO', 'REGISTER', 'INTERNAL'}
 | |
| 
 | |
|     data_path_iter: data_path_iter
 | |
|     data_path_item: data_path_item
 | |
| 
 | |
|     type: EnumProperty(
 | |
|         name="Type",
 | |
|         items=(('TOGGLE', "Toggle", ""),
 | |
|                ('ENABLE', "Enable", ""),
 | |
|                ('DISABLE', "Disable", ""),
 | |
|                ),
 | |
|     )
 | |
| 
 | |
|     def execute(self, context):
 | |
|         data_path_iter = self.data_path_iter
 | |
|         data_path_item = self.data_path_item
 | |
| 
 | |
|         items = list(getattr(context, data_path_iter))
 | |
|         items_ok = []
 | |
|         is_set = False
 | |
|         for item in items:
 | |
|             try:
 | |
|                 value_orig = eval("item." + data_path_item)
 | |
|             except:
 | |
|                 continue
 | |
| 
 | |
|             if value_orig is True:
 | |
|                 is_set = True
 | |
|             elif value_orig is False:
 | |
|                 pass
 | |
|             else:
 | |
|                 self.report({'WARNING'}, "Non boolean value found: %s[ ].%s" %
 | |
|                             (data_path_iter, data_path_item))
 | |
|                 return {'CANCELLED'}
 | |
| 
 | |
|             items_ok.append(item)
 | |
| 
 | |
|         # avoid undo push when nothing to do
 | |
|         if not items_ok:
 | |
|             return {'CANCELLED'}
 | |
| 
 | |
|         if self.type == 'ENABLE':
 | |
|             is_set = True
 | |
|         elif self.type == 'DISABLE':
 | |
|             is_set = False
 | |
|         else:
 | |
|             is_set = not is_set
 | |
| 
 | |
|         exec_str = "item.%s = %s" % (data_path_item, is_set)
 | |
|         for item in items_ok:
 | |
|             exec(exec_str)
 | |
| 
 | |
|         return operator_value_undo_return(item)
 | |
| 
 | |
| 
 | |
| class WM_OT_context_modal_mouse(Operator):
 | |
|     """Adjust arbitrary values with mouse input"""
 | |
|     bl_idname = "wm.context_modal_mouse"
 | |
|     bl_label = "Context Modal Mouse"
 | |
|     bl_options = {'GRAB_CURSOR', 'BLOCKING', 'UNDO', 'INTERNAL'}
 | |
| 
 | |
|     data_path_iter: data_path_iter
 | |
|     data_path_item: data_path_item
 | |
|     header_text: StringProperty(
 | |
|         name="Header Text",
 | |
|         description="Text to display in header during scale",
 | |
|     )
 | |
| 
 | |
|     input_scale: FloatProperty(
 | |
|         description="Scale the mouse movement by this value before applying the delta",
 | |
|         default=0.01,
 | |
|     )
 | |
|     invert: BoolProperty(
 | |
|         description="Invert the mouse input",
 | |
|         default=False,
 | |
|     )
 | |
|     initial_x: IntProperty(options={'HIDDEN'})
 | |
| 
 | |
|     def _values_store(self, context):
 | |
|         data_path_iter = self.data_path_iter
 | |
|         data_path_item = self.data_path_item
 | |
| 
 | |
|         self._values = values = {}
 | |
| 
 | |
|         for item in getattr(context, data_path_iter):
 | |
|             try:
 | |
|                 value_orig = eval("item." + data_path_item)
 | |
|             except:
 | |
|                 continue
 | |
| 
 | |
|             # check this can be set, maybe this is library data.
 | |
|             try:
 | |
|                 exec("item.%s = %s" % (data_path_item, value_orig))
 | |
|             except:
 | |
|                 continue
 | |
| 
 | |
|             values[item] = value_orig
 | |
| 
 | |
|     def _values_delta(self, delta):
 | |
|         delta *= self.input_scale
 | |
|         if self.invert:
 | |
|             delta = - delta
 | |
| 
 | |
|         data_path_item = self.data_path_item
 | |
|         for item, value_orig in self._values.items():
 | |
|             if type(value_orig) == int:
 | |
|                 exec("item.%s = int(%d)" % (data_path_item, round(value_orig + delta)))
 | |
|             else:
 | |
|                 exec("item.%s = %f" % (data_path_item, value_orig + delta))
 | |
| 
 | |
|     def _values_restore(self):
 | |
|         data_path_item = self.data_path_item
 | |
|         for item, value_orig in self._values.items():
 | |
|             exec("item.%s = %s" % (data_path_item, value_orig))
 | |
| 
 | |
|         self._values.clear()
 | |
| 
 | |
|     def _values_clear(self):
 | |
|         self._values.clear()
 | |
| 
 | |
|     def modal(self, context, event):
 | |
|         event_type = event.type
 | |
| 
 | |
|         if event_type == 'MOUSEMOVE':
 | |
|             delta = event.mouse_x - self.initial_x
 | |
|             self._values_delta(delta)
 | |
|             header_text = self.header_text
 | |
|             if header_text:
 | |
|                 if len(self._values) == 1:
 | |
|                     (item, ) = self._values.keys()
 | |
|                     header_text = header_text % eval("item.%s" % self.data_path_item)
 | |
|                 else:
 | |
|                     header_text = (self.header_text % delta) + " (delta)"
 | |
|                 context.area.header_text_set(header_text)
 | |
| 
 | |
|         elif 'LEFTMOUSE' == event_type:
 | |
|             item = next(iter(self._values.keys()))
 | |
|             self._values_clear()
 | |
|             context.area.header_text_set(None)
 | |
|             return operator_value_undo_return(item)
 | |
| 
 | |
|         elif event_type in {'RIGHTMOUSE', 'ESC'}:
 | |
|             self._values_restore()
 | |
|             context.area.header_text_set(None)
 | |
|             return {'CANCELLED'}
 | |
| 
 | |
|         return {'RUNNING_MODAL'}
 | |
| 
 | |
|     def invoke(self, context, event):
 | |
|         self._values_store(context)
 | |
| 
 | |
|         if not self._values:
 | |
|             self.report({'WARNING'}, "Nothing to operate on: %s[ ].%s" %
 | |
|                         (self.data_path_iter, self.data_path_item))
 | |
| 
 | |
|             return {'CANCELLED'}
 | |
|         else:
 | |
|             self.initial_x = event.mouse_x
 | |
| 
 | |
|             context.window_manager.modal_handler_add(self)
 | |
|             return {'RUNNING_MODAL'}
 | |
| 
 | |
| 
 | |
| class WM_OT_url_open(Operator):
 | |
|     """Open a website in the web-browser"""
 | |
|     bl_idname = "wm.url_open"
 | |
|     bl_label = ""
 | |
|     bl_options = {'INTERNAL'}
 | |
| 
 | |
|     url: StringProperty(
 | |
|         name="URL",
 | |
|         description="URL to open",
 | |
|     )
 | |
| 
 | |
|     def execute(self, context):
 | |
|         import webbrowser
 | |
|         webbrowser.open(self.url)
 | |
|         return {'FINISHED'}
 | |
| 
 | |
| 
 | |
| class WM_OT_path_open(Operator):
 | |
|     """Open a path in a file browser"""
 | |
|     bl_idname = "wm.path_open"
 | |
|     bl_label = ""
 | |
|     bl_options = {'INTERNAL'}
 | |
| 
 | |
|     filepath: StringProperty(
 | |
|         subtype='FILE_PATH',
 | |
|         options={'SKIP_SAVE'},
 | |
|     )
 | |
| 
 | |
|     def execute(self, context):
 | |
|         import sys
 | |
|         import os
 | |
|         import subprocess
 | |
| 
 | |
|         filepath = self.filepath
 | |
| 
 | |
|         if not filepath:
 | |
|             self.report({'ERROR'}, "File path was not set")
 | |
|             return {'CANCELLED'}
 | |
| 
 | |
|         filepath = bpy.path.abspath(filepath)
 | |
|         filepath = os.path.normpath(filepath)
 | |
| 
 | |
|         if not os.path.exists(filepath):
 | |
|             self.report({'ERROR'}, "File '%s' not found" % filepath)
 | |
|             return {'CANCELLED'}
 | |
| 
 | |
|         if sys.platform[:3] == "win":
 | |
|             os.startfile(filepath)
 | |
|         elif sys.platform == "darwin":
 | |
|             subprocess.check_call(["open", filepath])
 | |
|         else:
 | |
|             try:
 | |
|                 subprocess.check_call(["xdg-open", filepath])
 | |
|             except:
 | |
|                 # xdg-open *should* be supported by recent Gnome, KDE, Xfce
 | |
|                 import traceback
 | |
|                 traceback.print_exc()
 | |
| 
 | |
|         return {'FINISHED'}
 | |
| 
 | |
| 
 | |
| def _wm_doc_get_id(doc_id, do_url=True, url_prefix=""):
 | |
| 
 | |
|     def operator_exists_pair(a, b):
 | |
|         # Not fast, this is only for docs.
 | |
|         return b in dir(getattr(bpy.ops, a))
 | |
| 
 | |
|     def operator_exists_single(a):
 | |
|         a, b = a.partition("_OT_")[::2]
 | |
|         return operator_exists_pair(a.lower(), b)
 | |
| 
 | |
|     id_split = doc_id.split(".")
 | |
|     url = rna = None
 | |
| 
 | |
|     if len(id_split) == 1:  # rna, class
 | |
|         if do_url:
 | |
|             url = "%s/bpy.types.%s.html" % (url_prefix, id_split[0])
 | |
|         else:
 | |
|             rna = "bpy.types.%s" % id_split[0]
 | |
| 
 | |
|     elif len(id_split) == 2:  # rna, class.prop
 | |
|         class_name, class_prop = id_split
 | |
| 
 | |
|         # an operator (common case - just button referencing an op)
 | |
|         if operator_exists_pair(class_name, class_prop):
 | |
|             if do_url:
 | |
|                 url = (
 | |
|                     "%s/bpy.ops.%s.html#bpy.ops.%s.%s" %
 | |
|                     (url_prefix, class_name, class_name, class_prop)
 | |
|                 )
 | |
|             else:
 | |
|                 rna = "bpy.ops.%s.%s" % (class_name, class_prop)
 | |
|         elif operator_exists_single(class_name):
 | |
|             # note: ignore the prop name since we don't have a way to link into it
 | |
|             class_name, class_prop = class_name.split("_OT_", 1)
 | |
|             class_name = class_name.lower()
 | |
|             if do_url:
 | |
|                 url = (
 | |
|                     "%s/bpy.ops.%s.html#bpy.ops.%s.%s" %
 | |
|                     (url_prefix, class_name, class_name, class_prop)
 | |
|                 )
 | |
|             else:
 | |
|                 rna = "bpy.ops.%s.%s" % (class_name, class_prop)
 | |
|         else:
 | |
|             # an RNA setting, common case
 | |
|             rna_class = getattr(bpy.types, class_name)
 | |
| 
 | |
|             # detect if this is a inherited member and use that name instead
 | |
|             rna_parent = rna_class.bl_rna
 | |
|             rna_prop = rna_parent.properties.get(class_prop)
 | |
|             if rna_prop:
 | |
|                 rna_parent = rna_parent.base
 | |
|                 while rna_parent and rna_prop == rna_parent.properties.get(class_prop):
 | |
|                     class_name = rna_parent.identifier
 | |
|                     rna_parent = rna_parent.base
 | |
| 
 | |
|                 if do_url:
 | |
|                     url = (
 | |
|                         "%s/bpy.types.%s.html#bpy.types.%s.%s" %
 | |
|                         (url_prefix, class_name, class_name, class_prop)
 | |
|                     )
 | |
|                 else:
 | |
|                     rna = "bpy.types.%s.%s" % (class_name, class_prop)
 | |
|             else:
 | |
|                 # We assume this is custom property, only try to generate generic url/rna_id...
 | |
|                 if do_url:
 | |
|                     url = ("%s/bpy.types.bpy_struct.html#bpy.types.bpy_struct.items" % (url_prefix,))
 | |
|                 else:
 | |
|                     rna = "bpy.types.bpy_struct"
 | |
| 
 | |
|     return url if do_url else rna
 | |
| 
 | |
| 
 | |
| class WM_OT_doc_view_manual(Operator):
 | |
|     """Load online manual"""
 | |
|     bl_idname = "wm.doc_view_manual"
 | |
|     bl_label = "View Manual"
 | |
| 
 | |
|     doc_id: doc_id
 | |
| 
 | |
|     @staticmethod
 | |
|     def _find_reference(rna_id, url_mapping, verbose=True):
 | |
|         if verbose:
 | |
|             print("online manual check for: '%s'... " % rna_id)
 | |
|         from fnmatch import fnmatchcase
 | |
|         # XXX, for some reason all RNA ID's are stored lowercase
 | |
|         # Adding case into all ID's isn't worth the hassle so force lowercase.
 | |
|         rna_id = rna_id.lower()
 | |
|         for pattern, url_suffix in url_mapping:
 | |
|             if fnmatchcase(rna_id, pattern):
 | |
|                 if verbose:
 | |
|                     print("            match found: '%s' --> '%s'" % (pattern, url_suffix))
 | |
|                 return url_suffix
 | |
|         if verbose:
 | |
|             print("match not found")
 | |
|         return None
 | |
| 
 | |
|     @staticmethod
 | |
|     def _lookup_rna_url(rna_id, verbose=True):
 | |
|         for prefix, url_manual_mapping in bpy.utils.manual_map():
 | |
|             rna_ref = WM_OT_doc_view_manual._find_reference(rna_id, url_manual_mapping, verbose=verbose)
 | |
|             if rna_ref is not None:
 | |
|                 url = prefix + rna_ref
 | |
|                 return url
 | |
| 
 | |
|     def execute(self, context):
 | |
|         rna_id = _wm_doc_get_id(self.doc_id, do_url=False)
 | |
|         if rna_id is None:
 | |
|             return {'PASS_THROUGH'}
 | |
| 
 | |
|         url = self._lookup_rna_url(rna_id)
 | |
| 
 | |
|         if url is None:
 | |
|             self.report(
 | |
|                 {'WARNING'},
 | |
|                 "No reference available %r, "
 | |
|                 "Update info in 'rna_manual_reference.py' "
 | |
|                 "or callback to bpy.utils.manual_map()" %
 | |
|                 self.doc_id
 | |
|             )
 | |
|             return {'CANCELLED'}
 | |
|         else:
 | |
|             import webbrowser
 | |
|             webbrowser.open(url)
 | |
|             return {'FINISHED'}
 | |
| 
 | |
| 
 | |
| class WM_OT_doc_view(Operator):
 | |
|     """Load online reference docs"""
 | |
|     bl_idname = "wm.doc_view"
 | |
|     bl_label = "View Documentation"
 | |
| 
 | |
|     doc_id: doc_id
 | |
|     if bpy.app.version_cycle == "release":
 | |
|         _prefix = ("https://docs.blender.org/api/blender_python_api_current")
 | |
|     else:
 | |
|         _prefix = ("https://docs.blender.org/api/blender_python_api_master")
 | |
| 
 | |
|     def execute(self, context):
 | |
|         url = _wm_doc_get_id(self.doc_id, do_url=True, url_prefix=self._prefix)
 | |
|         if url is None:
 | |
|             return {'PASS_THROUGH'}
 | |
| 
 | |
|         import webbrowser
 | |
|         webbrowser.open(url)
 | |
| 
 | |
|         return {'FINISHED'}
 | |
| 
 | |
| 
 | |
| rna_path = StringProperty(
 | |
|     name="Property Edit",
 | |
|     description="Property data_path edit",
 | |
|     maxlen=1024,
 | |
|     options={'HIDDEN'},
 | |
| )
 | |
| 
 | |
| rna_value = StringProperty(
 | |
|     name="Property Value",
 | |
|     description="Property value edit",
 | |
|     maxlen=1024,
 | |
| )
 | |
| 
 | |
| rna_property = StringProperty(
 | |
|     name="Property Name",
 | |
|     description="Property name edit",
 | |
|     maxlen=1024,
 | |
| )
 | |
| 
 | |
| rna_min = FloatProperty(
 | |
|     name="Min",
 | |
|     default=-10000.0,
 | |
|     precision=3,
 | |
| )
 | |
| 
 | |
| rna_max = FloatProperty(
 | |
|     name="Max",
 | |
|     default=10000.0,
 | |
|     precision=3,
 | |
| )
 | |
| 
 | |
| rna_use_soft_limits = BoolProperty(
 | |
|     name="Use Soft Limits",
 | |
| )
 | |
| 
 | |
| rna_is_overridable_static = BoolProperty(
 | |
|     name="Is Statically Overridable",
 | |
|     default=False,
 | |
| )
 | |
| 
 | |
| 
 | |
| class WM_OT_properties_edit(Operator):
 | |
|     bl_idname = "wm.properties_edit"
 | |
|     bl_label = "Edit Property"
 | |
|     # register only because invoke_props_popup requires.
 | |
|     bl_options = {'REGISTER', 'INTERNAL'}
 | |
| 
 | |
|     data_path: rna_path
 | |
|     property: rna_property
 | |
|     value: rna_value
 | |
|     min: rna_min
 | |
|     max: rna_max
 | |
|     use_soft_limits: rna_use_soft_limits
 | |
|     is_overridable_static: rna_is_overridable_static
 | |
|     soft_min: rna_min
 | |
|     soft_max: rna_max
 | |
|     description: StringProperty(
 | |
|         name="Tooltip",
 | |
|     )
 | |
| 
 | |
|     def _cmp_props_get(self):
 | |
|         # Changing these properties will refresh the UI
 | |
|         return {
 | |
|             "use_soft_limits": self.use_soft_limits,
 | |
|             "soft_range": (self.soft_min, self.soft_max),
 | |
|             "hard_range": (self.min, self.max),
 | |
|         }
 | |
| 
 | |
|     def execute(self, context):
 | |
|         from rna_prop_ui import (
 | |
|             rna_idprop_ui_prop_get,
 | |
|             rna_idprop_ui_prop_clear,
 | |
|             rna_idprop_ui_prop_update,
 | |
|         )
 | |
| 
 | |
|         data_path = self.data_path
 | |
|         value = self.value
 | |
|         prop = self.property
 | |
| 
 | |
|         prop_old = getattr(self, "_last_prop", [None])[0]
 | |
| 
 | |
|         if prop_old is None:
 | |
|             self.report({'ERROR'}, "Direct execution not supported")
 | |
|             return {'CANCELLED'}
 | |
| 
 | |
|         try:
 | |
|             value_eval = eval(value)
 | |
|             # assert else None -> None, not "None", see [#33431]
 | |
|             assert(type(value_eval) in {str, float, int, bool, tuple, list})
 | |
|         except:
 | |
|             value_eval = value
 | |
| 
 | |
|         # First remove
 | |
|         item = eval("context.%s" % data_path)
 | |
|         prop_type_old = type(item[prop_old])
 | |
| 
 | |
|         rna_idprop_ui_prop_clear(item, prop_old)
 | |
|         exec_str = "del item[%r]" % prop_old
 | |
|         # print(exec_str)
 | |
|         exec(exec_str)
 | |
| 
 | |
|         # Reassign
 | |
|         exec_str = "item[%r] = %s" % (prop, repr(value_eval))
 | |
|         # print(exec_str)
 | |
|         exec(exec_str)
 | |
| 
 | |
|         exec_str = "item.property_overridable_static_set('[\"%s\"]', %s)" % (prop, self.is_overridable_static)
 | |
|         exec(exec_str)
 | |
| 
 | |
|         rna_idprop_ui_prop_update(item, prop)
 | |
| 
 | |
|         self._last_prop[:] = [prop]
 | |
| 
 | |
|         prop_type = type(item[prop])
 | |
| 
 | |
|         prop_ui = rna_idprop_ui_prop_get(item, prop)
 | |
| 
 | |
|         if prop_type in {float, int}:
 | |
|             prop_ui["min"] = prop_type(self.min)
 | |
|             prop_ui["max"] = prop_type(self.max)
 | |
| 
 | |
|             if self.use_soft_limits:
 | |
|                 prop_ui["soft_min"] = prop_type(self.soft_min)
 | |
|                 prop_ui["soft_max"] = prop_type(self.soft_max)
 | |
|             else:
 | |
|                 prop_ui["soft_min"] = prop_type(self.min)
 | |
|                 prop_ui["soft_max"] = prop_type(self.max)
 | |
| 
 | |
|         prop_ui["description"] = self.description
 | |
| 
 | |
|         # If we have changed the type of the property, update its potential anim curves!
 | |
|         if prop_type_old != prop_type:
 | |
|             data_path = '["%s"]' % bpy.utils.escape_identifier(prop)
 | |
|             done = set()
 | |
| 
 | |
|             def _update(fcurves):
 | |
|                 for fcu in fcurves:
 | |
|                     if fcu not in done and fcu.data_path == data_path:
 | |
|                         fcu.update_autoflags(item)
 | |
|                         done.add(fcu)
 | |
| 
 | |
|             def _update_strips(strips):
 | |
|                 for st in strips:
 | |
|                     if st.type == 'CLIP' and st.action:
 | |
|                         _update(st.action.fcurves)
 | |
|                     elif st.type == 'META':
 | |
|                         _update_strips(st.strips)
 | |
| 
 | |
|             adt = getattr(item, "animation_data", None)
 | |
|             if adt is not None:
 | |
|                 if adt.action:
 | |
|                     _update(adt.action.fcurves)
 | |
|                 if adt.drivers:
 | |
|                     _update(adt.drivers)
 | |
|                 if adt.nla_tracks:
 | |
|                     for nt in adt.nla_tracks:
 | |
|                         _update_strips(nt.strips)
 | |
| 
 | |
|         # otherwise existing buttons which reference freed
 | |
|         # memory may crash blender [#26510]
 | |
|         # context.area.tag_redraw()
 | |
|         for win in context.window_manager.windows:
 | |
|             for area in win.screen.areas:
 | |
|                 area.tag_redraw()
 | |
| 
 | |
|         return {'FINISHED'}
 | |
| 
 | |
|     def invoke(self, context, event):
 | |
|         from rna_prop_ui import rna_idprop_ui_prop_get
 | |
| 
 | |
|         data_path = self.data_path
 | |
| 
 | |
|         if not data_path:
 | |
|             self.report({'ERROR'}, "Data path not set")
 | |
|             return {'CANCELLED'}
 | |
| 
 | |
|         self._last_prop = [self.property]
 | |
| 
 | |
|         item = eval("context.%s" % data_path)
 | |
| 
 | |
|         # setup defaults
 | |
|         prop_ui = rna_idprop_ui_prop_get(item, self.property, False)  # don't create
 | |
|         if prop_ui:
 | |
|             self.min = prop_ui.get("min", -1000000000)
 | |
|             self.max = prop_ui.get("max", 1000000000)
 | |
|             self.description = prop_ui.get("description", "")
 | |
| 
 | |
|             self.soft_min = prop_ui.get("soft_min", self.min)
 | |
|             self.soft_max = prop_ui.get("soft_max", self.max)
 | |
|             self.use_soft_limits = (
 | |
|                 self.min != self.soft_min or
 | |
|                 self.max != self.soft_max
 | |
|             )
 | |
| 
 | |
|         # store for comparison
 | |
|         self._cmp_props = self._cmp_props_get()
 | |
| 
 | |
|         wm = context.window_manager
 | |
|         return wm.invoke_props_dialog(self)
 | |
| 
 | |
|     def check(self, context):
 | |
|         cmp_props = self._cmp_props_get()
 | |
|         changed = False
 | |
|         if self._cmp_props != cmp_props:
 | |
|             if cmp_props["use_soft_limits"]:
 | |
|                 if cmp_props["soft_range"] != self._cmp_props["soft_range"]:
 | |
|                     self.min = min(self.min, self.soft_min)
 | |
|                     self.max = max(self.max, self.soft_max)
 | |
|                     changed = True
 | |
|                 if cmp_props["hard_range"] != self._cmp_props["hard_range"]:
 | |
|                     self.soft_min = max(self.min, self.soft_min)
 | |
|                     self.soft_max = min(self.max, self.soft_max)
 | |
|                     changed = True
 | |
|             else:
 | |
|                 if cmp_props["soft_range"] != cmp_props["hard_range"]:
 | |
|                     self.soft_min = self.min
 | |
|                     self.soft_max = self.max
 | |
|                     changed = True
 | |
| 
 | |
|             changed |= (cmp_props["use_soft_limits"] != self._cmp_props["use_soft_limits"])
 | |
| 
 | |
|             if changed:
 | |
|                 cmp_props = self._cmp_props_get()
 | |
| 
 | |
|             self._cmp_props = cmp_props
 | |
| 
 | |
|         return changed
 | |
| 
 | |
|     def draw(self, context):
 | |
|         layout = self.layout
 | |
|         layout.prop(self, "property")
 | |
|         layout.prop(self, "value")
 | |
|         row = layout.row(align=True)
 | |
|         row.prop(self, "min")
 | |
|         row.prop(self, "max")
 | |
| 
 | |
|         row = layout.row()
 | |
|         row.prop(self, "use_soft_limits")
 | |
|         row.prop(self, "is_overridable_static")
 | |
| 
 | |
|         row = layout.row(align=True)
 | |
|         row.enabled = self.use_soft_limits
 | |
|         row.prop(self, "soft_min", text="Soft Min")
 | |
|         row.prop(self, "soft_max", text="Soft Max")
 | |
|         layout.prop(self, "description")
 | |
| 
 | |
| 
 | |
| class WM_OT_properties_add(Operator):
 | |
|     bl_idname = "wm.properties_add"
 | |
|     bl_label = "Add Property"
 | |
|     bl_options = {'UNDO', 'INTERNAL'}
 | |
| 
 | |
|     data_path: rna_path
 | |
| 
 | |
|     def execute(self, context):
 | |
|         from rna_prop_ui import (
 | |
|             rna_idprop_ui_prop_get,
 | |
|             rna_idprop_ui_prop_update,
 | |
|         )
 | |
| 
 | |
|         data_path = self.data_path
 | |
|         item = eval("context.%s" % data_path)
 | |
| 
 | |
|         def unique_name(names):
 | |
|             prop = "prop"
 | |
|             prop_new = prop
 | |
|             i = 1
 | |
|             while prop_new in names:
 | |
|                 prop_new = prop + str(i)
 | |
|                 i += 1
 | |
| 
 | |
|             return prop_new
 | |
| 
 | |
|         prop = unique_name({
 | |
|             *item.keys(),
 | |
|             *type(item).bl_rna.properties.keys(),
 | |
|         })
 | |
| 
 | |
|         item[prop] = 1.0
 | |
|         rna_idprop_ui_prop_update(item, prop)
 | |
| 
 | |
|         # not essential, but without this we get [#31661]
 | |
|         prop_ui = rna_idprop_ui_prop_get(item, prop)
 | |
|         prop_ui["soft_min"] = prop_ui["min"] = 0.0
 | |
|         prop_ui["soft_max"] = prop_ui["max"] = 1.0
 | |
| 
 | |
|         return {'FINISHED'}
 | |
| 
 | |
| 
 | |
| class WM_OT_properties_context_change(Operator):
 | |
|     """Jump to a different tab inside the properties editor"""
 | |
|     bl_idname = "wm.properties_context_change"
 | |
|     bl_label = ""
 | |
|     bl_options = {'INTERNAL'}
 | |
| 
 | |
|     context: StringProperty(
 | |
|         name="Context",
 | |
|         maxlen=64,
 | |
|     )
 | |
| 
 | |
|     def execute(self, context):
 | |
|         context.space_data.context = self.context
 | |
|         return {'FINISHED'}
 | |
| 
 | |
| 
 | |
| class WM_OT_properties_remove(Operator):
 | |
|     """Internal use (edit a property data_path)"""
 | |
|     bl_idname = "wm.properties_remove"
 | |
|     bl_label = "Remove Property"
 | |
|     bl_options = {'UNDO', 'INTERNAL'}
 | |
| 
 | |
|     data_path: rna_path
 | |
|     property: rna_property
 | |
| 
 | |
|     def execute(self, context):
 | |
|         from rna_prop_ui import (
 | |
|             rna_idprop_ui_prop_clear,
 | |
|             rna_idprop_ui_prop_update,
 | |
|         )
 | |
|         data_path = self.data_path
 | |
|         item = eval("context.%s" % data_path)
 | |
|         prop = self.property
 | |
|         rna_idprop_ui_prop_update(item, prop)
 | |
|         del item[prop]
 | |
|         rna_idprop_ui_prop_clear(item, prop)
 | |
| 
 | |
|         return {'FINISHED'}
 | |
| 
 | |
| 
 | |
| class WM_OT_keyconfig_activate(Operator):
 | |
|     bl_idname = "wm.keyconfig_activate"
 | |
|     bl_label = "Activate Keyconfig"
 | |
| 
 | |
|     filepath: StringProperty(
 | |
|         subtype='FILE_PATH',
 | |
|     )
 | |
| 
 | |
|     def execute(self, context):
 | |
|         if bpy.utils.keyconfig_set(self.filepath, report=self.report):
 | |
|             return {'FINISHED'}
 | |
|         else:
 | |
|             return {'CANCELLED'}
 | |
| 
 | |
| 
 | |
| class WM_OT_appconfig_default(Operator):
 | |
|     bl_idname = "wm.appconfig_default"
 | |
|     bl_label = "Default Application Configuration"
 | |
| 
 | |
|     def execute(self, context):
 | |
|         import os
 | |
| 
 | |
|         context.window_manager.keyconfigs.active = context.window_manager.keyconfigs.default
 | |
| 
 | |
|         filepath = os.path.join(bpy.utils.preset_paths("interaction")[0], "blender.py")
 | |
| 
 | |
|         if os.path.exists(filepath):
 | |
|             bpy.ops.script.execute_preset(
 | |
|                 filepath=filepath,
 | |
|                 menu_idname="USERPREF_MT_interaction_presets",
 | |
|             )
 | |
| 
 | |
|         return {'FINISHED'}
 | |
| 
 | |
| 
 | |
| class WM_OT_appconfig_activate(Operator):
 | |
|     bl_idname = "wm.appconfig_activate"
 | |
|     bl_label = "Activate Application Configuration"
 | |
| 
 | |
|     filepath: StringProperty(
 | |
|         subtype='FILE_PATH',
 | |
|     )
 | |
| 
 | |
|     def execute(self, context):
 | |
|         import os
 | |
|         filepath = self.filepath
 | |
|         bpy.utils.keyconfig_set(filepath)
 | |
|         dirname, filename = os.path.split(filepath)
 | |
|         filepath = os.path.normpath(os.path.join(dirname, os.pardir, "interaction", filename))
 | |
|         if os.path.exists(filepath):
 | |
|             bpy.ops.script.execute_preset(
 | |
|                 filepath=filepath,
 | |
|                 menu_idname="USERPREF_MT_interaction_presets",
 | |
|             )
 | |
| 
 | |
|         return {'FINISHED'}
 | |
| 
 | |
| 
 | |
| class WM_OT_sysinfo(Operator):
 | |
|     """Generate system information, saved into a text file"""
 | |
| 
 | |
|     bl_idname = "wm.sysinfo"
 | |
|     bl_label = "Save System Info"
 | |
| 
 | |
|     filepath: StringProperty(
 | |
|         subtype='FILE_PATH',
 | |
|         options={'SKIP_SAVE'},
 | |
|     )
 | |
| 
 | |
|     def execute(self, context):
 | |
|         import sys_info
 | |
|         sys_info.write_sysinfo(self.filepath)
 | |
|         return {'FINISHED'}
 | |
| 
 | |
|     def invoke(self, context, event):
 | |
|         import os
 | |
| 
 | |
|         if not self.filepath:
 | |
|             self.filepath = os.path.join(
 | |
|                 os.path.expanduser("~"), "system-info.txt")
 | |
| 
 | |
|         wm = context.window_manager
 | |
|         wm.fileselect_add(self)
 | |
|         return {'RUNNING_MODAL'}
 | |
| 
 | |
| 
 | |
| class WM_OT_copy_prev_settings(Operator):
 | |
|     """Copy settings from previous version"""
 | |
|     bl_idname = "wm.copy_prev_settings"
 | |
|     bl_label = "Copy Previous Settings"
 | |
| 
 | |
|     @staticmethod
 | |
|     def previous_version():
 | |
|         ver = bpy.app.version
 | |
|         ver_old = ((ver[0] * 100) + ver[1]) - 1
 | |
|         return ver_old // 100, ver_old % 100
 | |
| 
 | |
|     @staticmethod
 | |
|     def _old_path():
 | |
|         ver = bpy.app.version
 | |
|         ver_old = ((ver[0] * 100) + ver[1]) - 1
 | |
|         return bpy.utils.resource_path('USER', ver_old // 100, ver_old % 100)
 | |
| 
 | |
|     @staticmethod
 | |
|     def _new_path():
 | |
|         return bpy.utils.resource_path('USER')
 | |
| 
 | |
|     @classmethod
 | |
|     def poll(cls, context):
 | |
|         import os
 | |
| 
 | |
|         old = cls._old_path()
 | |
|         new = cls._new_path()
 | |
|         if os.path.isdir(old) and not os.path.isdir(new):
 | |
|             return True
 | |
| 
 | |
|         old_userpref = os.path.join(old, "config", "userpref.blend")
 | |
|         new_userpref = os.path.join(new, "config", "userpref.blend")
 | |
|         return os.path.isfile(old_userpref) and not os.path.isfile(new_userpref)
 | |
| 
 | |
|     def execute(self, context):
 | |
|         import shutil
 | |
| 
 | |
|         shutil.copytree(self._old_path(), self._new_path(), symlinks=True)
 | |
| 
 | |
|         # reload recent-files.txt
 | |
|         bpy.ops.wm.read_history()
 | |
| 
 | |
|         # don't loose users work if they open the splash later.
 | |
|         if bpy.data.is_saved is bpy.data.is_dirty is False:
 | |
|             bpy.ops.wm.read_homefile()
 | |
|         else:
 | |
|             self.report({'INFO'}, "Reload Start-Up file to restore settings")
 | |
| 
 | |
|         return {'FINISHED'}
 | |
| 
 | |
| 
 | |
| class WM_OT_keyconfig_test(Operator):
 | |
|     """Test key-config for conflicts"""
 | |
|     bl_idname = "wm.keyconfig_test"
 | |
|     bl_label = "Test Key Configuration for Conflicts"
 | |
| 
 | |
|     def execute(self, context):
 | |
|         from bpy_extras import keyconfig_utils
 | |
| 
 | |
|         wm = context.window_manager
 | |
|         kc = wm.keyconfigs.default
 | |
| 
 | |
|         if keyconfig_utils.keyconfig_test(kc):
 | |
|             print("CONFLICT")
 | |
| 
 | |
|         return {'FINISHED'}
 | |
| 
 | |
| 
 | |
| class WM_OT_keyconfig_import(Operator):
 | |
|     """Import key configuration from a python script"""
 | |
|     bl_idname = "wm.keyconfig_import"
 | |
|     bl_label = "Import Key Configuration..."
 | |
| 
 | |
|     filepath: StringProperty(
 | |
|         subtype='FILE_PATH',
 | |
|         default="keymap.py",
 | |
|     )
 | |
|     filter_folder: BoolProperty(
 | |
|         name="Filter folders",
 | |
|         default=True,
 | |
|         options={'HIDDEN'},
 | |
|     )
 | |
|     filter_text: BoolProperty(
 | |
|         name="Filter text",
 | |
|         default=True,
 | |
|         options={'HIDDEN'},
 | |
|     )
 | |
|     filter_python: BoolProperty(
 | |
|         name="Filter python",
 | |
|         default=True,
 | |
|         options={'HIDDEN'},
 | |
|     )
 | |
|     keep_original: BoolProperty(
 | |
|         name="Keep original",
 | |
|         description="Keep original file after copying to configuration folder",
 | |
|         default=True,
 | |
|     )
 | |
| 
 | |
|     def execute(self, context):
 | |
|         import os
 | |
|         from os.path import basename
 | |
|         import shutil
 | |
| 
 | |
|         if not self.filepath:
 | |
|             self.report({'ERROR'}, "Filepath not set")
 | |
|             return {'CANCELLED'}
 | |
| 
 | |
|         config_name = basename(self.filepath)
 | |
| 
 | |
|         path = bpy.utils.user_resource('SCRIPTS', os.path.join("presets", "keyconfig"), create=True)
 | |
|         path = os.path.join(path, config_name)
 | |
| 
 | |
|         try:
 | |
|             if self.keep_original:
 | |
|                 shutil.copy(self.filepath, path)
 | |
|             else:
 | |
|                 shutil.move(self.filepath, path)
 | |
|         except Exception as ex:
 | |
|             self.report({'ERROR'}, "Installing keymap failed: %s" % ex)
 | |
|             return {'CANCELLED'}
 | |
| 
 | |
|         # sneaky way to check we're actually running the code.
 | |
|         if bpy.utils.keyconfig_set(path, report=self.report):
 | |
|             return {'FINISHED'}
 | |
|         else:
 | |
|             return {'CANCELLED'}
 | |
| 
 | |
|     def invoke(self, context, event):
 | |
|         wm = context.window_manager
 | |
|         wm.fileselect_add(self)
 | |
|         return {'RUNNING_MODAL'}
 | |
| 
 | |
| # This operator is also used by interaction presets saving - AddPresetBase
 | |
| 
 | |
| 
 | |
| class WM_OT_keyconfig_export(Operator):
 | |
|     """Export key configuration to a python script"""
 | |
|     bl_idname = "wm.keyconfig_export"
 | |
|     bl_label = "Export Key Configuration..."
 | |
| 
 | |
|     all: BoolProperty(
 | |
|         name="All Keymaps",
 | |
|         default=False,
 | |
|         description="Write all keymaps (not just user modified)",
 | |
|     )
 | |
|     filepath: StringProperty(
 | |
|         subtype='FILE_PATH',
 | |
|         default="keymap.py",
 | |
|     )
 | |
|     filter_folder: BoolProperty(
 | |
|         name="Filter folders",
 | |
|         default=True,
 | |
|         options={'HIDDEN'},
 | |
|     )
 | |
|     filter_text: BoolProperty(
 | |
|         name="Filter text",
 | |
|         default=True,
 | |
|         options={'HIDDEN'},
 | |
|     )
 | |
|     filter_python: BoolProperty(
 | |
|         name="Filter python",
 | |
|         default=True,
 | |
|         options={'HIDDEN'},
 | |
|     )
 | |
| 
 | |
|     def execute(self, context):
 | |
|         from bpy_extras import keyconfig_utils
 | |
| 
 | |
|         if not self.filepath:
 | |
|             raise Exception("Filepath not set")
 | |
| 
 | |
|         if not self.filepath.endswith(".py"):
 | |
|             self.filepath += ".py"
 | |
| 
 | |
|         wm = context.window_manager
 | |
| 
 | |
|         keyconfig_utils.keyconfig_export_as_data(
 | |
|             wm,
 | |
|             wm.keyconfigs.active,
 | |
|             self.filepath,
 | |
|             all_keymaps=self.all,
 | |
|         )
 | |
| 
 | |
|         return {'FINISHED'}
 | |
| 
 | |
|     def invoke(self, context, event):
 | |
|         wm = context.window_manager
 | |
|         wm.fileselect_add(self)
 | |
|         return {'RUNNING_MODAL'}
 | |
| 
 | |
| 
 | |
| class WM_OT_keymap_restore(Operator):
 | |
|     """Restore key map(s)"""
 | |
|     bl_idname = "wm.keymap_restore"
 | |
|     bl_label = "Restore Key Map(s)"
 | |
| 
 | |
|     all: BoolProperty(
 | |
|         name="All Keymaps",
 | |
|         description="Restore all keymaps to default",
 | |
|     )
 | |
| 
 | |
|     def execute(self, context):
 | |
|         wm = context.window_manager
 | |
| 
 | |
|         if self.all:
 | |
|             for km in wm.keyconfigs.user.keymaps:
 | |
|                 km.restore_to_default()
 | |
|         else:
 | |
|             km = context.keymap
 | |
|             km.restore_to_default()
 | |
| 
 | |
|         return {'FINISHED'}
 | |
| 
 | |
| 
 | |
| class WM_OT_keyitem_restore(Operator):
 | |
|     """Restore key map item"""
 | |
|     bl_idname = "wm.keyitem_restore"
 | |
|     bl_label = "Restore Key Map Item"
 | |
| 
 | |
|     item_id: IntProperty(
 | |
|         name="Item Identifier",
 | |
|         description="Identifier of the item to remove",
 | |
|     )
 | |
| 
 | |
|     @classmethod
 | |
|     def poll(cls, context):
 | |
|         keymap = getattr(context, "keymap", None)
 | |
|         return keymap
 | |
| 
 | |
|     def execute(self, context):
 | |
|         km = context.keymap
 | |
|         kmi = km.keymap_items.from_id(self.item_id)
 | |
| 
 | |
|         if (not kmi.is_user_defined) and kmi.is_user_modified:
 | |
|             km.restore_item_to_default(kmi)
 | |
| 
 | |
|         return {'FINISHED'}
 | |
| 
 | |
| 
 | |
| class WM_OT_keyitem_add(Operator):
 | |
|     """Add key map item"""
 | |
|     bl_idname = "wm.keyitem_add"
 | |
|     bl_label = "Add Key Map Item"
 | |
| 
 | |
|     def execute(self, context):
 | |
|         km = context.keymap
 | |
| 
 | |
|         if km.is_modal:
 | |
|             km.keymap_items.new_modal("", 'A', 'PRESS')
 | |
|         else:
 | |
|             km.keymap_items.new("none", 'A', 'PRESS')
 | |
| 
 | |
|         # clear filter and expand keymap so we can see the newly added item
 | |
|         if context.space_data.filter_text != "":
 | |
|             context.space_data.filter_text = ""
 | |
|             km.show_expanded_items = True
 | |
|             km.show_expanded_children = True
 | |
| 
 | |
|         return {'FINISHED'}
 | |
| 
 | |
| 
 | |
| class WM_OT_keyitem_remove(Operator):
 | |
|     """Remove key map item"""
 | |
|     bl_idname = "wm.keyitem_remove"
 | |
|     bl_label = "Remove Key Map Item"
 | |
| 
 | |
|     item_id: IntProperty(
 | |
|         name="Item Identifier",
 | |
|         description="Identifier of the item to remove",
 | |
|     )
 | |
| 
 | |
|     @classmethod
 | |
|     def poll(cls, context):
 | |
|         return hasattr(context, "keymap")
 | |
| 
 | |
|     def execute(self, context):
 | |
|         km = context.keymap
 | |
|         kmi = km.keymap_items.from_id(self.item_id)
 | |
|         km.keymap_items.remove(kmi)
 | |
|         return {'FINISHED'}
 | |
| 
 | |
| 
 | |
| class WM_OT_keyconfig_remove(Operator):
 | |
|     """Remove key config"""
 | |
|     bl_idname = "wm.keyconfig_remove"
 | |
|     bl_label = "Remove Key Config"
 | |
| 
 | |
|     @classmethod
 | |
|     def poll(cls, context):
 | |
|         wm = context.window_manager
 | |
|         keyconf = wm.keyconfigs.active
 | |
|         return keyconf and keyconf.is_user_defined
 | |
| 
 | |
|     def execute(self, context):
 | |
|         wm = context.window_manager
 | |
|         keyconfig = wm.keyconfigs.active
 | |
|         wm.keyconfigs.remove(keyconfig)
 | |
|         return {'FINISHED'}
 | |
| 
 | |
| 
 | |
| class WM_OT_operator_cheat_sheet(Operator):
 | |
|     """List all the Operators in a text-block, useful for scripting"""
 | |
|     bl_idname = "wm.operator_cheat_sheet"
 | |
|     bl_label = "Operator Cheat Sheet"
 | |
| 
 | |
|     def execute(self, context):
 | |
|         op_strings = []
 | |
|         tot = 0
 | |
|         for op_module_name in dir(bpy.ops):
 | |
|             op_module = getattr(bpy.ops, op_module_name)
 | |
|             for op_submodule_name in dir(op_module):
 | |
|                 op = getattr(op_module, op_submodule_name)
 | |
|                 text = repr(op)
 | |
|                 if text.split("\n")[-1].startswith("bpy.ops."):
 | |
|                     op_strings.append(text)
 | |
|                     tot += 1
 | |
| 
 | |
|             op_strings.append('')
 | |
| 
 | |
|         textblock = bpy.data.texts.new("OperatorList.txt")
 | |
|         textblock.write('# %d Operators\n\n' % tot)
 | |
|         textblock.write('\n'.join(op_strings))
 | |
|         self.report({'INFO'}, "See OperatorList.txt textblock")
 | |
|         return {'FINISHED'}
 | |
| 
 | |
| 
 | |
| # -----------------------------------------------------------------------------
 | |
| # Add-on Operators
 | |
| 
 | |
| class WM_OT_addon_enable(Operator):
 | |
|     """Enable an add-on"""
 | |
|     bl_idname = "wm.addon_enable"
 | |
|     bl_label = "Enable Add-on"
 | |
| 
 | |
|     module: StringProperty(
 | |
|         name="Module",
 | |
|         description="Module name of the add-on to enable",
 | |
|     )
 | |
| 
 | |
|     def execute(self, context):
 | |
|         import addon_utils
 | |
| 
 | |
|         err_str = ""
 | |
| 
 | |
|         def err_cb(ex):
 | |
|             import traceback
 | |
|             nonlocal err_str
 | |
|             err_str = traceback.format_exc()
 | |
|             print(err_str)
 | |
| 
 | |
|         mod = addon_utils.enable(self.module, default_set=True, handle_error=err_cb)
 | |
| 
 | |
|         if mod:
 | |
|             info = addon_utils.module_bl_info(mod)
 | |
| 
 | |
|             info_ver = info.get("blender", (0, 0, 0))
 | |
| 
 | |
|             if info_ver > bpy.app.version:
 | |
|                 self.report(
 | |
|                     {'WARNING'},
 | |
|                     "This script was written Blender "
 | |
|                     "version %d.%d.%d and might not "
 | |
|                     "function (correctly), "
 | |
|                     "though it is enabled" %
 | |
|                     info_ver
 | |
|                 )
 | |
|             return {'FINISHED'}
 | |
|         else:
 | |
| 
 | |
|             if err_str:
 | |
|                 self.report({'ERROR'}, err_str)
 | |
| 
 | |
|             return {'CANCELLED'}
 | |
| 
 | |
| 
 | |
| class WM_OT_addon_disable(Operator):
 | |
|     """Disable an add-on"""
 | |
|     bl_idname = "wm.addon_disable"
 | |
|     bl_label = "Disable Add-on"
 | |
| 
 | |
|     module: StringProperty(
 | |
|         name="Module",
 | |
|         description="Module name of the add-on to disable",
 | |
|     )
 | |
| 
 | |
|     def execute(self, context):
 | |
|         import addon_utils
 | |
| 
 | |
|         err_str = ""
 | |
| 
 | |
|         def err_cb(ex):
 | |
|             import traceback
 | |
|             nonlocal err_str
 | |
|             err_str = traceback.format_exc()
 | |
|             print(err_str)
 | |
| 
 | |
|         addon_utils.disable(self.module, default_set=True, handle_error=err_cb)
 | |
| 
 | |
|         if err_str:
 | |
|             self.report({'ERROR'}, err_str)
 | |
| 
 | |
|         return {'FINISHED'}
 | |
| 
 | |
| 
 | |
| class WM_OT_owner_enable(Operator):
 | |
|     """Enable workspace owner ID"""
 | |
|     bl_idname = "wm.owner_enable"
 | |
|     bl_label = "Enable Add-on"
 | |
| 
 | |
|     owner_id: StringProperty(
 | |
|         name="UI Tag",
 | |
|     )
 | |
| 
 | |
|     def execute(self, context):
 | |
|         workspace = context.workspace
 | |
|         workspace.owner_ids.new(self.owner_id)
 | |
|         return {'FINISHED'}
 | |
| 
 | |
| 
 | |
| class WM_OT_owner_disable(Operator):
 | |
|     """Enable workspace owner ID"""
 | |
|     bl_idname = "wm.owner_disable"
 | |
|     bl_label = "Disable UI Tag"
 | |
| 
 | |
|     owner_id: StringProperty(
 | |
|         name="UI Tag",
 | |
|     )
 | |
| 
 | |
|     def execute(self, context):
 | |
|         workspace = context.workspace
 | |
|         owner_id = workspace.owner_ids[self.owner_id]
 | |
|         workspace.owner_ids.remove(owner_id)
 | |
|         return {'FINISHED'}
 | |
| 
 | |
| 
 | |
| class WM_OT_theme_install(Operator):
 | |
|     """Load and apply a Blender XML theme file"""
 | |
|     bl_idname = "wm.theme_install"
 | |
|     bl_label = "Install Theme..."
 | |
| 
 | |
|     overwrite: BoolProperty(
 | |
|         name="Overwrite",
 | |
|         description="Remove existing theme file if exists",
 | |
|         default=True,
 | |
|     )
 | |
|     filepath: StringProperty(
 | |
|         subtype='FILE_PATH',
 | |
|     )
 | |
|     filter_folder: BoolProperty(
 | |
|         name="Filter folders",
 | |
|         default=True,
 | |
|         options={'HIDDEN'},
 | |
|     )
 | |
|     filter_glob: StringProperty(
 | |
|         default="*.xml",
 | |
|         options={'HIDDEN'},
 | |
|     )
 | |
| 
 | |
|     def execute(self, context):
 | |
|         import os
 | |
|         import shutil
 | |
|         import traceback
 | |
| 
 | |
|         xmlfile = self.filepath
 | |
| 
 | |
|         path_themes = bpy.utils.user_resource('SCRIPTS', "presets/interface_theme", create=True)
 | |
| 
 | |
|         if not path_themes:
 | |
|             self.report({'ERROR'}, "Failed to get themes path")
 | |
|             return {'CANCELLED'}
 | |
| 
 | |
|         path_dest = os.path.join(path_themes, os.path.basename(xmlfile))
 | |
| 
 | |
|         if not self.overwrite:
 | |
|             if os.path.exists(path_dest):
 | |
|                 self.report({'WARNING'}, "File already installed to %r\n" % path_dest)
 | |
|                 return {'CANCELLED'}
 | |
| 
 | |
|         try:
 | |
|             shutil.copyfile(xmlfile, path_dest)
 | |
|             bpy.ops.script.execute_preset(
 | |
|                 filepath=path_dest,
 | |
|                 menu_idname="USERPREF_MT_interface_theme_presets",
 | |
|             )
 | |
| 
 | |
|         except:
 | |
|             traceback.print_exc()
 | |
|             return {'CANCELLED'}
 | |
| 
 | |
|         return {'FINISHED'}
 | |
| 
 | |
|     def invoke(self, context, event):
 | |
|         wm = context.window_manager
 | |
|         wm.fileselect_add(self)
 | |
|         return {'RUNNING_MODAL'}
 | |
| 
 | |
| 
 | |
| class WM_OT_addon_refresh(Operator):
 | |
|     """Scan add-on directories for new modules"""
 | |
|     bl_idname = "wm.addon_refresh"
 | |
|     bl_label = "Refresh"
 | |
| 
 | |
|     def execute(self, context):
 | |
|         import addon_utils
 | |
| 
 | |
|         addon_utils.modules_refresh()
 | |
| 
 | |
|         return {'FINISHED'}
 | |
| 
 | |
| 
 | |
| # Note: shares some logic with WM_OT_app_template_install
 | |
| # but not enough to de-duplicate. Fixed here may apply to both.
 | |
| class WM_OT_addon_install(Operator):
 | |
|     """Install an add-on"""
 | |
|     bl_idname = "wm.addon_install"
 | |
|     bl_label = "Install Add-on from File..."
 | |
| 
 | |
|     overwrite: BoolProperty(
 | |
|         name="Overwrite",
 | |
|         description="Remove existing add-ons with the same ID",
 | |
|         default=True,
 | |
|     )
 | |
|     target: EnumProperty(
 | |
|         name="Target Path",
 | |
|         items=(('DEFAULT', "Default", ""),
 | |
|                ('PREFS', "User Prefs", "")),
 | |
|     )
 | |
| 
 | |
|     filepath: StringProperty(
 | |
|         subtype='FILE_PATH',
 | |
|     )
 | |
|     filter_folder: BoolProperty(
 | |
|         name="Filter folders",
 | |
|         default=True,
 | |
|         options={'HIDDEN'},
 | |
|     )
 | |
|     filter_python: BoolProperty(
 | |
|         name="Filter python",
 | |
|         default=True,
 | |
|         options={'HIDDEN'},
 | |
|     )
 | |
|     filter_glob: StringProperty(
 | |
|         default="*.py;*.zip",
 | |
|         options={'HIDDEN'},
 | |
|     )
 | |
| 
 | |
|     def execute(self, context):
 | |
|         import addon_utils
 | |
|         import traceback
 | |
|         import zipfile
 | |
|         import shutil
 | |
|         import os
 | |
| 
 | |
|         pyfile = self.filepath
 | |
| 
 | |
|         if self.target == 'DEFAULT':
 | |
|             # don't use bpy.utils.script_paths("addons") because we may not be able to write to it.
 | |
|             path_addons = bpy.utils.user_resource('SCRIPTS', "addons", create=True)
 | |
|         else:
 | |
|             path_addons = context.user_preferences.filepaths.script_directory
 | |
|             if path_addons:
 | |
|                 path_addons = os.path.join(path_addons, "addons")
 | |
| 
 | |
|         if not path_addons:
 | |
|             self.report({'ERROR'}, "Failed to get add-ons path")
 | |
|             return {'CANCELLED'}
 | |
| 
 | |
|         if not os.path.isdir(path_addons):
 | |
|             try:
 | |
|                 os.makedirs(path_addons, exist_ok=True)
 | |
|             except:
 | |
|                 traceback.print_exc()
 | |
| 
 | |
|         # Check if we are installing from a target path,
 | |
|         # doing so causes 2+ addons of same name or when the same from/to
 | |
|         # location is used, removal of the file!
 | |
|         addon_path = ""
 | |
|         pyfile_dir = os.path.dirname(pyfile)
 | |
|         for addon_path in addon_utils.paths():
 | |
|             if os.path.samefile(pyfile_dir, addon_path):
 | |
|                 self.report({'ERROR'}, "Source file is in the add-on search path: %r" % addon_path)
 | |
|                 return {'CANCELLED'}
 | |
|         del addon_path
 | |
|         del pyfile_dir
 | |
|         # done checking for exceptional case
 | |
| 
 | |
|         addons_old = {mod.__name__ for mod in addon_utils.modules()}
 | |
| 
 | |
|         # check to see if the file is in compressed format (.zip)
 | |
|         if zipfile.is_zipfile(pyfile):
 | |
|             try:
 | |
|                 file_to_extract = zipfile.ZipFile(pyfile, 'r')
 | |
|             except:
 | |
|                 traceback.print_exc()
 | |
|                 return {'CANCELLED'}
 | |
| 
 | |
|             if self.overwrite:
 | |
|                 for f in file_to_extract.namelist():
 | |
|                     module_filesystem_remove(path_addons, f)
 | |
|             else:
 | |
|                 for f in file_to_extract.namelist():
 | |
|                     path_dest = os.path.join(path_addons, os.path.basename(f))
 | |
|                     if os.path.exists(path_dest):
 | |
|                         self.report({'WARNING'}, "File already installed to %r\n" % path_dest)
 | |
|                         return {'CANCELLED'}
 | |
| 
 | |
|             try:  # extract the file to "addons"
 | |
|                 file_to_extract.extractall(path_addons)
 | |
|             except:
 | |
|                 traceback.print_exc()
 | |
|                 return {'CANCELLED'}
 | |
| 
 | |
|         else:
 | |
|             path_dest = os.path.join(path_addons, os.path.basename(pyfile))
 | |
| 
 | |
|             if self.overwrite:
 | |
|                 module_filesystem_remove(path_addons, os.path.basename(pyfile))
 | |
|             elif os.path.exists(path_dest):
 | |
|                 self.report({'WARNING'}, "File already installed to %r\n" % path_dest)
 | |
|                 return {'CANCELLED'}
 | |
| 
 | |
|             # if not compressed file just copy into the addon path
 | |
|             try:
 | |
|                 shutil.copyfile(pyfile, path_dest)
 | |
|             except:
 | |
|                 traceback.print_exc()
 | |
|                 return {'CANCELLED'}
 | |
| 
 | |
|         addons_new = {mod.__name__ for mod in addon_utils.modules()} - addons_old
 | |
|         addons_new.discard("modules")
 | |
| 
 | |
|         # disable any addons we may have enabled previously and removed.
 | |
|         # this is unlikely but do just in case. bug [#23978]
 | |
|         for new_addon in addons_new:
 | |
|             addon_utils.disable(new_addon, default_set=True)
 | |
| 
 | |
|         # possible the zip contains multiple addons, we could disallow this
 | |
|         # but for now just use the first
 | |
|         for mod in addon_utils.modules(refresh=False):
 | |
|             if mod.__name__ in addons_new:
 | |
|                 info = addon_utils.module_bl_info(mod)
 | |
| 
 | |
|                 # show the newly installed addon.
 | |
|                 context.window_manager.addon_filter = 'All'
 | |
|                 context.window_manager.addon_search = info["name"]
 | |
|                 break
 | |
| 
 | |
|         # in case a new module path was created to install this addon.
 | |
|         bpy.utils.refresh_script_paths()
 | |
| 
 | |
|         # print message
 | |
|         msg = (
 | |
|             tip_("Modules Installed (%s) from %r into %r") %
 | |
|             (", ".join(sorted(addons_new)), pyfile, path_addons)
 | |
|         )
 | |
|         print(msg)
 | |
|         self.report({'INFO'}, msg)
 | |
| 
 | |
|         return {'FINISHED'}
 | |
| 
 | |
|     def invoke(self, context, event):
 | |
|         wm = context.window_manager
 | |
|         wm.fileselect_add(self)
 | |
|         return {'RUNNING_MODAL'}
 | |
| 
 | |
| 
 | |
| class WM_OT_addon_remove(Operator):
 | |
|     """Delete the add-on from the file system"""
 | |
|     bl_idname = "wm.addon_remove"
 | |
|     bl_label = "Remove Add-on"
 | |
| 
 | |
|     module: StringProperty(
 | |
|         name="Module",
 | |
|         description="Module name of the add-on to remove",
 | |
|     )
 | |
| 
 | |
|     @staticmethod
 | |
|     def path_from_addon(module):
 | |
|         import os
 | |
|         import addon_utils
 | |
| 
 | |
|         for mod in addon_utils.modules():
 | |
|             if mod.__name__ == module:
 | |
|                 filepath = mod.__file__
 | |
|                 if os.path.exists(filepath):
 | |
|                     if os.path.splitext(os.path.basename(filepath))[0] == "__init__":
 | |
|                         return os.path.dirname(filepath), True
 | |
|                     else:
 | |
|                         return filepath, False
 | |
|         return None, False
 | |
| 
 | |
|     def execute(self, context):
 | |
|         import addon_utils
 | |
|         import os
 | |
| 
 | |
|         path, isdir = WM_OT_addon_remove.path_from_addon(self.module)
 | |
|         if path is None:
 | |
|             self.report({'WARNING'}, "Add-on path %r could not be found" % path)
 | |
|             return {'CANCELLED'}
 | |
| 
 | |
|         # in case its enabled
 | |
|         addon_utils.disable(self.module, default_set=True)
 | |
| 
 | |
|         import shutil
 | |
|         if isdir:
 | |
|             shutil.rmtree(path)
 | |
|         else:
 | |
|             os.remove(path)
 | |
| 
 | |
|         addon_utils.modules_refresh()
 | |
| 
 | |
|         context.area.tag_redraw()
 | |
|         return {'FINISHED'}
 | |
| 
 | |
|     # lame confirmation check
 | |
|     def draw(self, context):
 | |
|         self.layout.label(text="Remove Add-on: %r?" % self.module)
 | |
|         path, _isdir = WM_OT_addon_remove.path_from_addon(self.module)
 | |
|         self.layout.label(text="Path: %r" % path)
 | |
| 
 | |
|     def invoke(self, context, event):
 | |
|         wm = context.window_manager
 | |
|         return wm.invoke_props_dialog(self, width=600)
 | |
| 
 | |
| 
 | |
| class WM_OT_addon_expand(Operator):
 | |
|     """Display information and preferences for this add-on"""
 | |
|     bl_idname = "wm.addon_expand"
 | |
|     bl_label = ""
 | |
|     bl_options = {'INTERNAL'}
 | |
| 
 | |
|     module: StringProperty(
 | |
|         name="Module",
 | |
|         description="Module name of the add-on to expand",
 | |
|     )
 | |
| 
 | |
|     def execute(self, context):
 | |
|         import addon_utils
 | |
| 
 | |
|         module_name = self.module
 | |
| 
 | |
|         mod = addon_utils.addons_fake_modules.get(module_name)
 | |
|         if mod is not None:
 | |
|             info = addon_utils.module_bl_info(mod)
 | |
|             info["show_expanded"] = not info["show_expanded"]
 | |
| 
 | |
|         return {'FINISHED'}
 | |
| 
 | |
| 
 | |
| class WM_OT_addon_userpref_show(Operator):
 | |
|     """Show add-on user preferences"""
 | |
|     bl_idname = "wm.addon_userpref_show"
 | |
|     bl_label = ""
 | |
|     bl_options = {'INTERNAL'}
 | |
| 
 | |
|     module: StringProperty(
 | |
|         name="Module",
 | |
|         description="Module name of the add-on to expand",
 | |
|     )
 | |
| 
 | |
|     def execute(self, context):
 | |
|         import addon_utils
 | |
| 
 | |
|         module_name = self.module
 | |
| 
 | |
|         modules = addon_utils.modules(refresh=False)
 | |
|         mod = addon_utils.addons_fake_modules.get(module_name)
 | |
|         if mod is not None:
 | |
|             info = addon_utils.module_bl_info(mod)
 | |
|             info["show_expanded"] = True
 | |
| 
 | |
|             context.user_preferences.active_section = 'ADDONS'
 | |
|             context.window_manager.addon_filter = 'All'
 | |
|             context.window_manager.addon_search = info["name"]
 | |
|             bpy.ops.screen.userpref_show('INVOKE_DEFAULT')
 | |
| 
 | |
|         return {'FINISHED'}
 | |
| 
 | |
| 
 | |
| # Note: shares some logic with WM_OT_addon_install
 | |
| # but not enough to de-duplicate. Fixes here may apply to both.
 | |
| class WM_OT_app_template_install(Operator):
 | |
|     """Install an application-template"""
 | |
|     bl_idname = "wm.app_template_install"
 | |
|     bl_label = "Install Template from File..."
 | |
| 
 | |
|     overwrite: BoolProperty(
 | |
|         name="Overwrite",
 | |
|         description="Remove existing template with the same ID",
 | |
|         default=True,
 | |
|     )
 | |
| 
 | |
|     filepath: StringProperty(
 | |
|         subtype='FILE_PATH',
 | |
|     )
 | |
|     filter_folder: BoolProperty(
 | |
|         name="Filter folders",
 | |
|         default=True,
 | |
|         options={'HIDDEN'},
 | |
|     )
 | |
|     filter_glob: StringProperty(
 | |
|         default="*.zip",
 | |
|         options={'HIDDEN'},
 | |
|     )
 | |
| 
 | |
|     def execute(self, context):
 | |
|         import traceback
 | |
|         import zipfile
 | |
|         import os
 | |
| 
 | |
|         filepath = self.filepath
 | |
| 
 | |
|         path_app_templates = bpy.utils.user_resource(
 | |
|             'SCRIPTS', os.path.join("startup", "bl_app_templates_user"),
 | |
|             create=True,
 | |
|         )
 | |
| 
 | |
|         if not path_app_templates:
 | |
|             self.report({'ERROR'}, "Failed to get add-ons path")
 | |
|             return {'CANCELLED'}
 | |
| 
 | |
|         if not os.path.isdir(path_app_templates):
 | |
|             try:
 | |
|                 os.makedirs(path_app_templates, exist_ok=True)
 | |
|             except:
 | |
|                 traceback.print_exc()
 | |
| 
 | |
|         app_templates_old = set(os.listdir(path_app_templates))
 | |
| 
 | |
|         # check to see if the file is in compressed format (.zip)
 | |
|         if zipfile.is_zipfile(filepath):
 | |
|             try:
 | |
|                 file_to_extract = zipfile.ZipFile(filepath, 'r')
 | |
|             except:
 | |
|                 traceback.print_exc()
 | |
|                 return {'CANCELLED'}
 | |
| 
 | |
|             if self.overwrite:
 | |
|                 for f in file_to_extract.namelist():
 | |
|                     module_filesystem_remove(path_app_templates, f)
 | |
|             else:
 | |
|                 for f in file_to_extract.namelist():
 | |
|                     path_dest = os.path.join(path_app_templates, os.path.basename(f))
 | |
|                     if os.path.exists(path_dest):
 | |
|                         self.report({'WARNING'}, "File already installed to %r\n" % path_dest)
 | |
|                         return {'CANCELLED'}
 | |
| 
 | |
|             try:  # extract the file to "bl_app_templates_user"
 | |
|                 file_to_extract.extractall(path_app_templates)
 | |
|             except:
 | |
|                 traceback.print_exc()
 | |
|                 return {'CANCELLED'}
 | |
| 
 | |
|         else:
 | |
|             # Only support installing zipfiles
 | |
|             self.report({'WARNING'}, "Expected a zip-file %r\n" % filepath)
 | |
|             return {'CANCELLED'}
 | |
| 
 | |
|         app_templates_new = set(os.listdir(path_app_templates)) - app_templates_old
 | |
| 
 | |
|         # in case a new module path was created to install this addon.
 | |
|         bpy.utils.refresh_script_paths()
 | |
| 
 | |
|         # print message
 | |
|         msg = (
 | |
|             tip_("Template Installed (%s) from %r into %r") %
 | |
|             (", ".join(sorted(app_templates_new)), filepath, path_app_templates)
 | |
|         )
 | |
|         print(msg)
 | |
|         self.report({'INFO'}, msg)
 | |
| 
 | |
|         return {'FINISHED'}
 | |
| 
 | |
|     def invoke(self, context, event):
 | |
|         wm = context.window_manager
 | |
|         wm.fileselect_add(self)
 | |
|         return {'RUNNING_MODAL'}
 | |
| 
 | |
| 
 | |
| class WM_OT_tool_set_by_name(Operator):
 | |
|     """Set the tool by name (for keymaps)"""
 | |
|     bl_idname = "wm.tool_set_by_name"
 | |
|     bl_label = "Set Tool By Name"
 | |
| 
 | |
|     name: StringProperty(
 | |
|         name="Text",
 | |
|         description="Display name of the tool",
 | |
|     )
 | |
|     cycle: BoolProperty(
 | |
|         name="Cycle",
 | |
|         description="Cycle through tools in this group",
 | |
|         default=False,
 | |
|         options={'SKIP_SAVE'},
 | |
|     )
 | |
| 
 | |
|     space_type: rna_space_type_prop
 | |
| 
 | |
|     def execute(self, context):
 | |
|         from bl_ui.space_toolsystem_common import (
 | |
|             activate_by_name,
 | |
|             activate_by_name_or_cycle,
 | |
|         )
 | |
| 
 | |
|         if self.properties.is_property_set("space_type"):
 | |
|             space_type = self.space_type
 | |
|         else:
 | |
|             space_type = context.space_data.type
 | |
| 
 | |
|         fn = activate_by_name_or_cycle if self.cycle else activate_by_name
 | |
|         if fn(context, space_type, self.name):
 | |
|             return {'FINISHED'}
 | |
|         else:
 | |
|             self.report({'WARNING'}, f"Tool {self.name!r:s} not found for space {space_type!r:s}.")
 | |
|             return {'CANCELLED'}
 | |
| 
 | |
| 
 | |
| class WM_OT_toolbar(Operator):
 | |
|     bl_idname = "wm.toolbar"
 | |
|     bl_label = "Toolbar"
 | |
| 
 | |
|     @classmethod
 | |
|     def poll(cls, context):
 | |
|         return context.space_data is not None
 | |
| 
 | |
|     def execute(self, context):
 | |
|         from bl_ui.space_toolsystem_common import (
 | |
|             ToolSelectPanelHelper,
 | |
|             keymap_from_context,
 | |
|         )
 | |
|         space_type = context.space_data.type
 | |
| 
 | |
|         cls = ToolSelectPanelHelper._tool_class_from_space_type(space_type)
 | |
|         if cls is None:
 | |
|             return {'CANCELLED'}
 | |
| 
 | |
|         wm = context.window_manager
 | |
|         keymap = keymap_from_context(context, space_type)
 | |
| 
 | |
|         def draw_menu(popover, context):
 | |
|             layout = popover.layout
 | |
|             layout.operator_context = 'INVOKE_REGION_WIN'
 | |
|             cls.draw_cls(layout, context, detect_layout=False, scale_y=1.0)
 | |
| 
 | |
|         wm.popover(draw_menu, ui_units_x=8, keymap=keymap)
 | |
|         return {'FINISHED'}
 | |
| 
 | |
| 
 | |
| # Studio Light operations
 | |
| class WM_OT_studiolight_install(Operator):
 | |
|     """Install a user defined studio light"""
 | |
|     bl_idname = "wm.studiolight_install"
 | |
|     bl_label = "Install Custom Studio Light"
 | |
| 
 | |
|     files: CollectionProperty(
 | |
|         name="File Path",
 | |
|         type=OperatorFileListElement,
 | |
|     )
 | |
|     directory: StringProperty(
 | |
|         subtype='DIR_PATH',
 | |
|     )
 | |
|     filter_folder: BoolProperty(
 | |
|         name="Filter folders",
 | |
|         default=True,
 | |
|         options={'HIDDEN'},
 | |
|     )
 | |
|     filter_glob: StringProperty(
 | |
|         default="*.png;*.jpg;*.hdr;*.exr",
 | |
|         options={'HIDDEN'},
 | |
|     )
 | |
|     orientation: EnumProperty(
 | |
|         items=(
 | |
|             ('MATCAP', "MatCap", ""),
 | |
|             ('WORLD', "World", ""),
 | |
|             ('CAMERA', "Camera", ""),
 | |
|         )
 | |
|     )
 | |
| 
 | |
|     def execute(self, context):
 | |
|         import traceback
 | |
|         import shutil
 | |
|         import pathlib
 | |
|         userpref = context.user_preferences
 | |
| 
 | |
|         filepaths = [pathlib.Path(self.directory, e.name) for e in self.files]
 | |
|         path_studiolights = bpy.utils.user_resource('DATAFILES')
 | |
| 
 | |
|         if not path_studiolights:
 | |
|             self.report({'ERROR'}, "Failed to get Studio Light path")
 | |
|             return {'CANCELLED'}
 | |
| 
 | |
|         path_studiolights = pathlib.Path(path_studiolights, "studiolights", self.orientation.lower())
 | |
|         if not path_studiolights.exists():
 | |
|             try:
 | |
|                 path_studiolights.mkdir(parents=True, exist_ok=True)
 | |
|             except:
 | |
|                 traceback.print_exc()
 | |
| 
 | |
|         for filepath in filepaths:
 | |
|             shutil.copy(str(filepath), str(path_studiolights))
 | |
|             userpref.studio_lights.new(str(path_studiolights.joinpath(filepath.name)), self.orientation)
 | |
| 
 | |
|         # print message
 | |
|         msg = (
 | |
|             tip_("StudioLight Installed %r into %r") %
 | |
|             (", ".join(str(x.name) for x in self.files), str(path_studiolights))
 | |
|         )
 | |
|         print(msg)
 | |
|         self.report({'INFO'}, msg)
 | |
|         return {'FINISHED'}
 | |
| 
 | |
|     def invoke(self, context, event):
 | |
|         wm = context.window_manager
 | |
|         wm.fileselect_add(self)
 | |
|         return {'RUNNING_MODAL'}
 | |
| 
 | |
| 
 | |
| class WM_OT_studiolight_uninstall(Operator):
 | |
|     bl_idname = 'wm.studiolight_uninstall'
 | |
|     bl_label = "Uninstall Studio Light"
 | |
|     index: bpy.props.IntProperty()
 | |
| 
 | |
|     def _remove_path(self, path):
 | |
|         if path.exists():
 | |
|             path.unlink()
 | |
| 
 | |
|     def execute(self, context):
 | |
|         import pathlib
 | |
|         userpref = context.user_preferences
 | |
|         for studio_light in userpref.studio_lights:
 | |
|             if studio_light.index == self.index:
 | |
|                 if studio_light.path:
 | |
|                     self._remove_path(pathlib.Path(studio_light.path))
 | |
|                 if studio_light.path_irr_cache:
 | |
|                     self._remove_path(pathlib.Path(studio_light.path_irr_cache))
 | |
|                 if studio_light.path_sh_cache:
 | |
|                     self._remove_path(pathlib.Path(studio_light.path_sh_cache))
 | |
|                 userpref.studio_lights.remove(studio_light)
 | |
|                 return {'FINISHED'}
 | |
|         return {'CANCELLED'}
 | |
| 
 | |
| 
 | |
| class WM_OT_studiolight_userpref_show(Operator):
 | |
|     """Show light user preferences"""
 | |
|     bl_idname = "wm.studiolight_userpref_show"
 | |
|     bl_label = ""
 | |
|     bl_options = {'INTERNAL'}
 | |
| 
 | |
|     def execute(self, context):
 | |
|         context.user_preferences.active_section = 'LIGHTS'
 | |
|         bpy.ops.screen.userpref_show('INVOKE_DEFAULT')
 | |
|         return {'FINISHED'}
 | |
| 
 | |
| 
 | |
| class WM_MT_splash(Menu):
 | |
|     bl_label = "Splash"
 | |
| 
 | |
|     def draw_setup(self, context):
 | |
|         layout = self.layout
 | |
|         layout.operator_context = 'EXEC_DEFAULT'
 | |
| 
 | |
|         layout.label(text="Quick Setup")
 | |
| 
 | |
|         split = layout.split(factor=0.25)
 | |
|         split.label()
 | |
|         split = split.split(factor=2.0 / 3.0)
 | |
| 
 | |
|         col = split.column()
 | |
| 
 | |
|         sub = col.column(align=True)
 | |
|         sub.label(text="Input and Shortcuts:")
 | |
|         text = bpy.path.display_name(context.window_manager.keyconfigs.active.name)
 | |
|         if not text:
 | |
|             text = "Blender (default)"
 | |
|         sub.menu("USERPREF_MT_appconfigs", text=text)
 | |
| 
 | |
|         col.separator()
 | |
| 
 | |
|         sub = col.column(align=True)
 | |
|         sub.label(text="Theme:")
 | |
|         label = bpy.types.USERPREF_MT_interface_theme_presets.bl_label
 | |
|         if label == "Presets":
 | |
|             label = "Blender Dark"
 | |
|         sub.menu("USERPREF_MT_interface_theme_presets", text=label)
 | |
| 
 | |
|         # We need to make switching to a language easier first
 | |
|         #sub = col.column(align=False)
 | |
|         # sub.label(text="Language:")
 | |
|         #userpref = context.user_preferences
 | |
|         #sub.prop(userpref.system, "language", text="")
 | |
| 
 | |
|         col.label()
 | |
| 
 | |
|         layout.label()
 | |
| 
 | |
|         row = layout.row()
 | |
| 
 | |
|         sub = row.row()
 | |
|         if bpy.types.WM_OT_copy_prev_settings.poll(context):
 | |
|             old_version = bpy.types.WM_OT_copy_prev_settings.previous_version()
 | |
|             sub.operator("wm.copy_prev_settings", text="Load %d.%d Settings" % old_version)
 | |
|             sub.operator("wm.save_userpref", text="Save New Settings")
 | |
|         else:
 | |
|             sub.label()
 | |
|             sub.label()
 | |
|             sub.operator("wm.save_userpref", text="Next")
 | |
| 
 | |
|         layout.separator()
 | |
| 
 | |
|     def draw(self, context):
 | |
|         # Draw setup screen if no user preferences have been saved yet.
 | |
|         import os
 | |
| 
 | |
|         user_path = bpy.utils.resource_path('USER')
 | |
|         userdef_path = os.path.join(user_path, "config", "userpref.blend")
 | |
| 
 | |
|         if not os.path.isfile(userdef_path):
 | |
|             self.draw_setup(context)
 | |
|             return
 | |
| 
 | |
|         # Pass
 | |
|         layout = self.layout
 | |
|         layout.operator_context = 'EXEC_DEFAULT'
 | |
|         layout.emboss = 'PULLDOWN_MENU'
 | |
| 
 | |
|         split = layout.split()
 | |
| 
 | |
|         # Templates
 | |
|         col1 = split.column()
 | |
|         col1.label(text="New File")
 | |
| 
 | |
|         bpy.types.TOPBAR_MT_file_new.draw_ex(col1, context, use_splash=True)
 | |
| 
 | |
|         # Recent
 | |
|         col2 = split.column()
 | |
|         col2_title = col2.row()
 | |
| 
 | |
|         found_recent = col2.template_recent_files()
 | |
| 
 | |
|         if found_recent:
 | |
|             col2_title.label(text="Recent Files")
 | |
|         else:
 | |
|             # Links if no recent files
 | |
|             col2_title.label(text="Getting Started")
 | |
| 
 | |
|             col2.operator(
 | |
|                 "wm.url_open", text="Manual", icon='URL'
 | |
|             ).url = "https://docs.blender.org/manual/en/dev/"
 | |
|             col2.operator(
 | |
|                 "wm.url_open", text="Release Notes", icon='URL',
 | |
|             ).url = "https://www.blender.org/download/releases/%d-%d/" % bpy.app.version[:2]
 | |
|             col2.operator(
 | |
|                 "wm.url_open", text="Blender Website", icon='URL',
 | |
|             ).url = "https://www.blender.org"
 | |
|             col2.operator(
 | |
|                 "wm.url_open", text="Credits", icon='URL',
 | |
|             ).url = "https://www.blender.org/about/credits/"
 | |
| 
 | |
|         layout.separator()
 | |
| 
 | |
|         split = layout.split()
 | |
| 
 | |
|         col1 = split.column()
 | |
|         sub = col1.row()
 | |
|         sub.operator_context = 'INVOKE_DEFAULT'
 | |
|         sub.operator("wm.open_mainfile", text="Open...", icon='FILE_FOLDER')
 | |
|         col1.operator("wm.recover_last_session", icon='RECOVER_LAST')
 | |
| 
 | |
|         col2 = split.column()
 | |
|         if found_recent:
 | |
|             col2.operator(
 | |
|                 "wm.url_open", text="Release Notes", icon='URL',
 | |
|             ).url = "https://www.blender.org/download/releases/%d-%d/" % bpy.app.version[:2]
 | |
|             col2.operator(
 | |
|                 "wm.url_open", text="Development Fund", icon='URL'
 | |
|             ).url = "https://fund.blender.org"
 | |
|         else:
 | |
|             col2.operator(
 | |
|                 "wm.url_open", text="Development Fund", icon='URL'
 | |
|             ).url = "https://fund.blender.org"
 | |
|             col2.operator(
 | |
|                 "wm.url_open", text="Donate", icon='URL'
 | |
|             ).url = "https://www.blender.org/foundation/donation-payment/"
 | |
| 
 | |
|         layout.separator()
 | |
| 
 | |
| 
 | |
| class WM_OT_drop_blend_file(Operator):
 | |
|     bl_idname = "wm.drop_blend_file"
 | |
|     bl_label = "Handle dropped .blend file"
 | |
|     bl_options = {'INTERNAL'}
 | |
| 
 | |
|     filepath: StringProperty()
 | |
| 
 | |
|     def invoke(self, context, event):
 | |
|         context.window_manager.popup_menu(self.draw_menu, title=bpy.path.basename(self.filepath), icon='QUESTION')
 | |
|         return {"FINISHED"}
 | |
| 
 | |
|     def draw_menu(self, menu, context):
 | |
|         layout = menu.layout
 | |
| 
 | |
|         col = layout.column()
 | |
|         col.operator_context = 'EXEC_DEFAULT'
 | |
|         col.operator("wm.open_mainfile", text="Open", icon='FILE_FOLDER').filepath = self.filepath
 | |
| 
 | |
|         layout.separator()
 | |
|         col = layout.column()
 | |
|         col.operator_context = 'INVOKE_DEFAULT'
 | |
|         col.operator("wm.link", text="Link...", icon='LINK_BLEND').filepath = self.filepath
 | |
|         col.operator("wm.append", text="Append...", icon='APPEND_BLEND').filepath = self.filepath
 | |
| 
 | |
| classes = (
 | |
|     BRUSH_OT_active_index_set,
 | |
|     WM_OT_addon_disable,
 | |
|     WM_OT_addon_enable,
 | |
|     WM_OT_addon_expand,
 | |
|     WM_OT_addon_install,
 | |
|     WM_OT_addon_refresh,
 | |
|     WM_OT_addon_remove,
 | |
|     WM_OT_addon_userpref_show,
 | |
|     WM_OT_app_template_install,
 | |
|     WM_OT_appconfig_activate,
 | |
|     WM_OT_appconfig_default,
 | |
|     WM_OT_context_collection_boolean_set,
 | |
|     WM_OT_context_cycle_array,
 | |
|     WM_OT_context_cycle_enum,
 | |
|     WM_OT_context_cycle_int,
 | |
|     WM_OT_context_menu_enum,
 | |
|     WM_OT_context_modal_mouse,
 | |
|     WM_OT_context_pie_enum,
 | |
|     WM_OT_context_scale_float,
 | |
|     WM_OT_context_scale_int,
 | |
|     WM_OT_context_set_boolean,
 | |
|     WM_OT_context_set_enum,
 | |
|     WM_OT_context_set_float,
 | |
|     WM_OT_context_set_id,
 | |
|     WM_OT_context_set_int,
 | |
|     WM_OT_context_set_string,
 | |
|     WM_OT_context_set_value,
 | |
|     WM_OT_context_toggle,
 | |
|     WM_OT_context_toggle_enum,
 | |
|     WM_OT_copy_prev_settings,
 | |
|     WM_OT_doc_view,
 | |
|     WM_OT_doc_view_manual,
 | |
|     WM_OT_drop_blend_file,
 | |
|     WM_OT_keyconfig_activate,
 | |
|     WM_OT_keyconfig_export,
 | |
|     WM_OT_keyconfig_import,
 | |
|     WM_OT_keyconfig_remove,
 | |
|     WM_OT_keyconfig_test,
 | |
|     WM_OT_keyitem_add,
 | |
|     WM_OT_keyitem_remove,
 | |
|     WM_OT_keyitem_restore,
 | |
|     WM_OT_keymap_restore,
 | |
|     WM_OT_operator_cheat_sheet,
 | |
|     WM_OT_operator_pie_enum,
 | |
|     WM_OT_path_open,
 | |
|     WM_OT_properties_add,
 | |
|     WM_OT_properties_context_change,
 | |
|     WM_OT_properties_edit,
 | |
|     WM_OT_properties_remove,
 | |
|     WM_OT_sysinfo,
 | |
|     WM_OT_theme_install,
 | |
|     WM_OT_owner_disable,
 | |
|     WM_OT_owner_enable,
 | |
|     WM_OT_url_open,
 | |
|     WM_OT_studiolight_install,
 | |
|     WM_OT_studiolight_uninstall,
 | |
|     WM_OT_studiolight_userpref_show,
 | |
|     WM_OT_tool_set_by_name,
 | |
|     WM_OT_toolbar,
 | |
|     WM_MT_splash,
 | |
| )
 |