fix for crash when a python operator or render engine was freed in the C code and...
[blender.git] / release / scripts / startup / bl_operators / wm.py
1 # ##### BEGIN GPL LICENSE BLOCK #####
2 #
3 #  This program is free software; you can redistribute it and/or
4 #  modify it under the terms of the GNU General Public License
5 #  as published by the Free Software Foundation; either version 2
6 #  of the License, or (at your option) any later version.
7 #
8 #  This program is distributed in the hope that it will be useful,
9 #  but WITHOUT ANY WARRANTY; without even the implied warranty of
10 #  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
11 #  GNU General Public License for more details.
12 #
13 #  You should have received a copy of the GNU General Public License
14 #  along with this program; if not, write to the Free Software Foundation,
15 #  Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
16 #
17 # ##### END GPL LICENSE BLOCK #####
18
19 # <pep8 compliant>
20
21 import bpy
22 from bpy.types import Menu, Operator
23 from bpy.props import (StringProperty,
24                        BoolProperty,
25                        IntProperty,
26                        FloatProperty,
27                        EnumProperty,
28                        )
29
30 from rna_prop_ui import rna_idprop_ui_prop_get, rna_idprop_ui_prop_clear
31
32
33 class MESH_OT_delete_edgeloop(Operator):
34     """Delete an edge loop by merging the faces on each side """ \
35     """to a single face loop"""
36     bl_idname = "mesh.delete_edgeloop"
37     bl_label = "Delete Edge Loop"
38
39     @classmethod
40     def poll(cls, context):
41         return bpy.ops.transform.edge_slide.poll()
42
43     def execute(self, context):
44         if 'FINISHED' in bpy.ops.transform.edge_slide(value=1.0):
45             bpy.ops.mesh.select_more()
46             bpy.ops.mesh.remove_doubles()
47             return {'FINISHED'}
48
49         return {'CANCELLED'}
50
51 rna_path_prop = StringProperty(
52         name="Context Attributes",
53         description="RNA context string",
54         maxlen=1024,
55         )
56
57 rna_reverse_prop = BoolProperty(
58         name="Reverse",
59         description="Cycle backwards",
60         default=False,
61         )
62
63 rna_relative_prop = BoolProperty(
64         name="Relative",
65         description="Apply relative to the current value (delta)",
66         default=False,
67         )
68
69
70 def context_path_validate(context, data_path):
71     try:
72         value = eval("context.%s" % data_path) if data_path else Ellipsis
73     except AttributeError as e:
74         if str(e).startswith("'NoneType'"):
75             # One of the items in the rna path is None, just ignore this
76             value = Ellipsis
77         else:
78             # We have a real error in the rna path, don't ignore that
79             raise
80
81     return value
82
83
84 def operator_value_is_undo(value):
85     if value in {None, Ellipsis}:
86         return False
87
88     # typical properties or objects
89     id_data = getattr(value, "id_data", Ellipsis)
90
91     if id_data is None:
92         return False
93     elif id_data is Ellipsis:
94         # handle mathutils types
95         id_data = getattr(getattr(value, "owner", None), "id_data", None)
96
97         if id_data is None:
98             return False
99
100     # return True if its a non window ID type
101     return (isinstance(id_data, bpy.types.ID) and
102             (not isinstance(id_data, (bpy.types.WindowManager,
103                                       bpy.types.Screen,
104                                       bpy.types.Brush,
105                                       ))))
106
107
108 def operator_path_is_undo(context, data_path):
109     # note that if we have data paths that use strings this could fail
110     # luckily we don't do this!
111     #
112     # When we cant find the data owner assume no undo is needed.
113     data_path_head = data_path.rpartition(".")[0]
114
115     if not data_path_head:
116         return False
117
118     value = context_path_validate(context, data_path_head)
119
120     return operator_value_is_undo(value)
121
122
123 def operator_path_undo_return(context, data_path):
124     return {'FINISHED'} if operator_path_is_undo(context, data_path) else {'CANCELLED'}
125
126
127 def operator_value_undo_return(value):
128     return {'FINISHED'} if operator_value_is_undo(value) else {'CANCELLED'}
129
130
131 def execute_context_assign(self, context):
132     data_path = self.data_path
133     if context_path_validate(context, data_path) is Ellipsis:
134         return {'PASS_THROUGH'}
135
136     if getattr(self, "relative", False):
137         exec("context.%s += self.value" % data_path)
138     else:
139         exec("context.%s = self.value" % data_path)
140
141     return operator_path_undo_return(context, data_path)
142
143
144 class BRUSH_OT_active_index_set(Operator):
145     """Set active sculpt/paint brush from it's number"""
146     bl_idname = "brush.active_index_set"
147     bl_label = "Set Brush Number"
148
149     mode = StringProperty(
150             name="Mode",
151             description="Paint mode to set brush for",
152             maxlen=1024,
153             )
154     index = IntProperty(
155             name="Number",
156             description="Brush number",
157             )
158
159     _attr_dict = {"sculpt": "use_paint_sculpt",
160                   "vertex_paint": "use_paint_vertex",
161                   "weight_paint": "use_paint_weight",
162                   "image_paint": "use_paint_image",
163                   }
164
165     def execute(self, context):
166         attr = self._attr_dict.get(self.mode)
167         if attr is None:
168             return {'CANCELLED'}
169
170         toolsettings = context.tool_settings
171         for i, brush in enumerate((cur for cur in bpy.data.brushes if getattr(cur, attr))):
172             if i == self.index:
173                 getattr(toolsettings, self.mode).brush = brush
174                 return {'FINISHED'}
175
176         return {'CANCELLED'}
177
178
179 class WM_OT_context_set_boolean(Operator):
180     """Set a context value"""
181     bl_idname = "wm.context_set_boolean"
182     bl_label = "Context Set Boolean"
183     bl_options = {'UNDO', 'INTERNAL'}
184
185     data_path = rna_path_prop
186     value = BoolProperty(
187             name="Value",
188             description="Assignment value",
189             default=True,
190             )
191
192     execute = execute_context_assign
193
194
195 class WM_OT_context_set_int(Operator):  # same as enum
196     """Set a context value"""
197     bl_idname = "wm.context_set_int"
198     bl_label = "Context Set"
199     bl_options = {'UNDO', 'INTERNAL'}
200
201     data_path = rna_path_prop
202     value = IntProperty(
203             name="Value",
204             description="Assign value",
205             default=0,
206             )
207     relative = rna_relative_prop
208
209     execute = execute_context_assign
210
211
212 class WM_OT_context_scale_int(Operator):
213     """Scale an int context value"""
214     bl_idname = "wm.context_scale_int"
215     bl_label = "Context Set"
216     bl_options = {'UNDO', 'INTERNAL'}
217
218     data_path = rna_path_prop
219     value = FloatProperty(
220             name="Value",
221             description="Assign value",
222             default=1.0,
223             )
224     always_step = BoolProperty(
225             name="Always Step",
226             description="Always adjust the value by a minimum of 1 when 'value' is not 1.0",
227             default=True,
228             )
229
230     def execute(self, context):
231         data_path = self.data_path
232         if context_path_validate(context, data_path) is Ellipsis:
233             return {'PASS_THROUGH'}
234
235         value = self.value
236
237         if value == 1.0:  # nothing to do
238             return {'CANCELLED'}
239
240         if getattr(self, "always_step", False):
241             if value > 1.0:
242                 add = "1"
243                 func = "max"
244             else:
245                 add = "-1"
246                 func = "min"
247             exec("context.%s = %s(round(context.%s * value), context.%s + %s)" %
248                  (data_path, func, data_path, data_path, add))
249         else:
250             exec("context.%s *= value" % data_path)
251
252         return operator_path_undo_return(context, data_path)
253
254
255 class WM_OT_context_set_float(Operator):  # same as enum
256     """Set a context value"""
257     bl_idname = "wm.context_set_float"
258     bl_label = "Context Set Float"
259     bl_options = {'UNDO', 'INTERNAL'}
260
261     data_path = rna_path_prop
262     value = FloatProperty(
263             name="Value",
264             description="Assignment value",
265             default=0.0,
266             )
267     relative = rna_relative_prop
268
269     execute = execute_context_assign
270
271
272 class WM_OT_context_set_string(Operator):  # same as enum
273     """Set a context value"""
274     bl_idname = "wm.context_set_string"
275     bl_label = "Context Set String"
276     bl_options = {'UNDO', 'INTERNAL'}
277
278     data_path = rna_path_prop
279     value = StringProperty(
280             name="Value",
281             description="Assign value",
282             maxlen=1024,
283             )
284
285     execute = execute_context_assign
286
287
288 class WM_OT_context_set_enum(Operator):
289     """Set a context value"""
290     bl_idname = "wm.context_set_enum"
291     bl_label = "Context Set Enum"
292     bl_options = {'UNDO', 'INTERNAL'}
293
294     data_path = rna_path_prop
295     value = StringProperty(
296             name="Value",
297             description="Assignment value (as a string)",
298             maxlen=1024,
299             )
300
301     execute = execute_context_assign
302
303
304 class WM_OT_context_set_value(Operator):
305     """Set a context value"""
306     bl_idname = "wm.context_set_value"
307     bl_label = "Context Set Value"
308     bl_options = {'UNDO', 'INTERNAL'}
309
310     data_path = rna_path_prop
311     value = StringProperty(
312             name="Value",
313             description="Assignment value (as a string)",
314             maxlen=1024,
315             )
316
317     def execute(self, context):
318         data_path = self.data_path
319         if context_path_validate(context, data_path) is Ellipsis:
320             return {'PASS_THROUGH'}
321         exec("context.%s = %s" % (data_path, self.value))
322         return operator_path_undo_return(context, data_path)
323
324
325 class WM_OT_context_toggle(Operator):
326     """Toggle a context value"""
327     bl_idname = "wm.context_toggle"
328     bl_label = "Context Toggle"
329     bl_options = {'UNDO', 'INTERNAL'}
330
331     data_path = rna_path_prop
332
333     def execute(self, context):
334         data_path = self.data_path
335
336         if context_path_validate(context, data_path) is Ellipsis:
337             return {'PASS_THROUGH'}
338
339         exec("context.%s = not (context.%s)" % (data_path, data_path))
340
341         return operator_path_undo_return(context, data_path)
342
343
344 class WM_OT_context_toggle_enum(Operator):
345     """Toggle a context value"""
346     bl_idname = "wm.context_toggle_enum"
347     bl_label = "Context Toggle Values"
348     bl_options = {'UNDO', 'INTERNAL'}
349
350     data_path = rna_path_prop
351     value_1 = StringProperty(
352             name="Value",
353             description="Toggle enum",
354             maxlen=1024,
355             )
356     value_2 = StringProperty(
357             name="Value",
358             description="Toggle enum",
359             maxlen=1024,
360             )
361
362     def execute(self, context):
363         data_path = self.data_path
364
365         if context_path_validate(context, data_path) is Ellipsis:
366             return {'PASS_THROUGH'}
367
368         exec("context.%s = ('%s', '%s')[context.%s != '%s']" %
369              (data_path, self.value_1,
370               self.value_2, data_path,
371               self.value_2,
372               ))
373
374         return operator_path_undo_return(context, data_path)
375
376
377 class WM_OT_context_cycle_int(Operator):
378     """Set a context value (useful for cycling active material, """ \
379     """vertex keys, groups, etc.)"""
380     bl_idname = "wm.context_cycle_int"
381     bl_label = "Context Int Cycle"
382     bl_options = {'UNDO', 'INTERNAL'}
383
384     data_path = rna_path_prop
385     reverse = rna_reverse_prop
386
387     def execute(self, context):
388         data_path = self.data_path
389         value = context_path_validate(context, data_path)
390         if value is Ellipsis:
391             return {'PASS_THROUGH'}
392
393         if self.reverse:
394             value -= 1
395         else:
396             value += 1
397
398         exec("context.%s = value" % data_path)
399
400         if value != eval("context.%s" % data_path):
401             # relies on rna clamping integers out of the range
402             if self.reverse:
403                 value = (1 << 31) - 1
404             else:
405                 value = -1 << 31
406
407             exec("context.%s = value" % data_path)
408
409         return operator_path_undo_return(context, data_path)
410
411
412 class WM_OT_context_cycle_enum(Operator):
413     """Toggle a context value"""
414     bl_idname = "wm.context_cycle_enum"
415     bl_label = "Context Enum Cycle"
416     bl_options = {'UNDO', 'INTERNAL'}
417
418     data_path = rna_path_prop
419     reverse = rna_reverse_prop
420
421     def execute(self, context):
422         data_path = self.data_path
423         value = context_path_validate(context, data_path)
424         if value is Ellipsis:
425             return {'PASS_THROUGH'}
426
427         orig_value = value
428
429         # Have to get rna enum values
430         rna_struct_str, rna_prop_str = data_path.rsplit('.', 1)
431         i = rna_prop_str.find('[')
432
433         # just in case we get "context.foo.bar[0]"
434         if i != -1:
435             rna_prop_str = rna_prop_str[0:i]
436
437         rna_struct = eval("context.%s.rna_type" % rna_struct_str)
438
439         rna_prop = rna_struct.properties[rna_prop_str]
440
441         if type(rna_prop) != bpy.types.EnumProperty:
442             raise Exception("expected an enum property")
443
444         enums = rna_struct.properties[rna_prop_str].enum_items.keys()
445         orig_index = enums.index(orig_value)
446
447         # Have the info we need, advance to the next item
448         if self.reverse:
449             if orig_index == 0:
450                 advance_enum = enums[-1]
451             else:
452                 advance_enum = enums[orig_index - 1]
453         else:
454             if orig_index == len(enums) - 1:
455                 advance_enum = enums[0]
456             else:
457                 advance_enum = enums[orig_index + 1]
458
459         # set the new value
460         exec("context.%s = advance_enum" % data_path)
461         return operator_path_undo_return(context, data_path)
462
463
464 class WM_OT_context_cycle_array(Operator):
465     """Set a context array value """ \
466     """(useful for cycling the active mesh edit mode)"""
467     bl_idname = "wm.context_cycle_array"
468     bl_label = "Context Array Cycle"
469     bl_options = {'UNDO', 'INTERNAL'}
470
471     data_path = rna_path_prop
472     reverse = rna_reverse_prop
473
474     def execute(self, context):
475         data_path = self.data_path
476         value = context_path_validate(context, data_path)
477         if value is Ellipsis:
478             return {'PASS_THROUGH'}
479
480         def cycle(array):
481             if self.reverse:
482                 array.insert(0, array.pop())
483             else:
484                 array.append(array.pop(0))
485             return array
486
487         exec("context.%s = cycle(context.%s[:])" % (data_path, data_path))
488
489         return operator_path_undo_return(context, data_path)
490
491
492 class WM_MT_context_menu_enum(Menu):
493     bl_label = ""
494     data_path = ""  # BAD DESIGN, set from operator below.
495
496     def draw(self, context):
497         data_path = self.data_path
498         value = context_path_validate(bpy.context, data_path)
499         if value is Ellipsis:
500             return {'PASS_THROUGH'}
501         base_path, prop_string = data_path.rsplit(".", 1)
502         value_base = context_path_validate(context, base_path)
503
504         values = [(i.name, i.identifier) for i in value_base.bl_rna.properties[prop_string].enum_items]
505
506         for name, identifier in values:
507             props = self.layout.operator("wm.context_set_enum", text=name)
508             props.data_path = data_path
509             props.value = identifier
510
511
512 class WM_OT_context_menu_enum(Operator):
513     bl_idname = "wm.context_menu_enum"
514     bl_label = "Context Enum Menu"
515     bl_options = {'UNDO', 'INTERNAL'}
516     data_path = rna_path_prop
517
518     def execute(self, context):
519         data_path = self.data_path
520         WM_MT_context_menu_enum.data_path = data_path
521         bpy.ops.wm.call_menu(name="WM_MT_context_menu_enum")
522         return {'PASS_THROUGH'}
523
524
525 class WM_OT_context_set_id(Operator):
526     """Toggle a context value"""
527     bl_idname = "wm.context_set_id"
528     bl_label = "Set Library ID"
529     bl_options = {'UNDO', 'INTERNAL'}
530
531     data_path = rna_path_prop
532     value = StringProperty(
533             name="Value",
534             description="Assign value",
535             maxlen=1024,
536             )
537
538     def execute(self, context):
539         value = self.value
540         data_path = self.data_path
541
542         # match the pointer type from the target property to bpy.data.*
543         # so we lookup the correct list.
544         data_path_base, data_path_prop = data_path.rsplit(".", 1)
545         data_prop_rna = eval("context.%s" % data_path_base).rna_type.properties[data_path_prop]
546         data_prop_rna_type = data_prop_rna.fixed_type
547
548         id_iter = None
549
550         for prop in bpy.data.rna_type.properties:
551             if prop.rna_type.identifier == "CollectionProperty":
552                 if prop.fixed_type == data_prop_rna_type:
553                     id_iter = prop.identifier
554                     break
555
556         if id_iter:
557             value_id = getattr(bpy.data, id_iter).get(value)
558             exec("context.%s = value_id" % data_path)
559
560         return operator_path_undo_return(context, data_path)
561
562
563 doc_id = StringProperty(
564         name="Doc ID",
565         maxlen=1024,
566         options={'HIDDEN'},
567         )
568
569 doc_new = StringProperty(
570         name="Edit Description",
571         maxlen=1024,
572         )
573
574 data_path_iter = StringProperty(
575         description="The data path relative to the context, must point to an iterable")
576
577 data_path_item = StringProperty(
578         description="The data path from each iterable to the value (int or float)")
579
580
581 class WM_OT_context_collection_boolean_set(Operator):
582     """Set boolean values for a collection of items"""
583     bl_idname = "wm.context_collection_boolean_set"
584     bl_label = "Context Collection Boolean Set"
585     bl_options = {'UNDO', 'REGISTER', 'INTERNAL'}
586
587     data_path_iter = data_path_iter
588     data_path_item = data_path_item
589
590     type = EnumProperty(
591             name="Type",
592             items=(('TOGGLE', "Toggle", ""),
593                    ('ENABLE', "Enable", ""),
594                    ('DISABLE', "Disable", ""),
595                    ),
596             )
597
598     def execute(self, context):
599         data_path_iter = self.data_path_iter
600         data_path_item = self.data_path_item
601
602         items = list(getattr(context, data_path_iter))
603         items_ok = []
604         is_set = False
605         for item in items:
606             try:
607                 value_orig = eval("item." + data_path_item)
608             except:
609                 continue
610
611             if value_orig is True:
612                 is_set = True
613             elif value_orig is False:
614                 pass
615             else:
616                 self.report({'WARNING'}, "Non boolean value found: %s[ ].%s" %
617                             (data_path_iter, data_path_item))
618                 return {'CANCELLED'}
619
620             items_ok.append(item)
621
622         # avoid undo push when nothing to do
623         if not items_ok:
624             return {'CANCELLED'}
625
626         if self.type == 'ENABLE':
627             is_set = True
628         elif self.type == 'DISABLE':
629             is_set = False
630         else:
631             is_set = not is_set
632
633         exec_str = "item.%s = %s" % (data_path_item, is_set)
634         for item in items_ok:
635             exec(exec_str)
636
637         return operator_value_undo_return(item)
638
639
640 class WM_OT_context_modal_mouse(Operator):
641     """Adjust arbitrary values with mouse input"""
642     bl_idname = "wm.context_modal_mouse"
643     bl_label = "Context Modal Mouse"
644     bl_options = {'GRAB_POINTER', 'BLOCKING', 'UNDO', 'INTERNAL'}
645
646     data_path_iter = data_path_iter
647     data_path_item = data_path_item
648     header_text = StringProperty(
649             name="Header Text",
650             description="Text to display in header during scale",
651             )
652
653     input_scale = FloatProperty(
654             description="Scale the mouse movement by this value before applying the delta",
655             default=0.01,
656             )
657     invert = BoolProperty(
658             description="Invert the mouse input",
659             default=False,
660             )
661     initial_x = IntProperty(options={'HIDDEN'})
662
663     def _values_store(self, context):
664         data_path_iter = self.data_path_iter
665         data_path_item = self.data_path_item
666
667         self._values = values = {}
668
669         for item in getattr(context, data_path_iter):
670             try:
671                 value_orig = eval("item." + data_path_item)
672             except:
673                 continue
674
675             # check this can be set, maybe this is library data.
676             try:
677                 exec("item.%s = %s" % (data_path_item, value_orig))
678             except:
679                 continue
680
681             values[item] = value_orig
682
683     def _values_delta(self, delta):
684         delta *= self.input_scale
685         if self.invert:
686             delta = - delta
687
688         data_path_item = self.data_path_item
689         for item, value_orig in self._values.items():
690             if type(value_orig) == int:
691                 exec("item.%s = int(%d)" % (data_path_item, round(value_orig + delta)))
692             else:
693                 exec("item.%s = %f" % (data_path_item, value_orig + delta))
694
695     def _values_restore(self):
696         data_path_item = self.data_path_item
697         for item, value_orig in self._values.items():
698             exec("item.%s = %s" % (data_path_item, value_orig))
699
700         self._values.clear()
701
702     def _values_clear(self):
703         self._values.clear()
704
705     def modal(self, context, event):
706         event_type = event.type
707
708         if event_type == 'MOUSEMOVE':
709             delta = event.mouse_x - self.initial_x
710             self._values_delta(delta)
711             header_text = self.header_text
712             if header_text:
713                 if len(self._values) == 1:
714                     (item, ) = self._values.keys()
715                     header_text = header_text % eval("item.%s" % self.data_path_item)
716                 else:
717                     header_text = (self.header_text % delta) + " (delta)"
718                 context.area.header_text_set(header_text)
719
720         elif 'LEFTMOUSE' == event_type:
721             item = next(iter(self._values.keys()))
722             self._values_clear()
723             context.area.header_text_set()
724             return operator_value_undo_return(item)
725
726         elif event_type in {'RIGHTMOUSE', 'ESC'}:
727             self._values_restore()
728             context.area.header_text_set()
729             return {'CANCELLED'}
730
731         return {'RUNNING_MODAL'}
732
733     def invoke(self, context, event):
734         self._values_store(context)
735
736         if not self._values:
737             self.report({'WARNING'}, "Nothing to operate on: %s[ ].%s" %
738                         (self.data_path_iter, self.data_path_item))
739
740             return {'CANCELLED'}
741         else:
742             self.initial_x = event.mouse_x
743
744             context.window_manager.modal_handler_add(self)
745             return {'RUNNING_MODAL'}
746
747
748 class WM_OT_url_open(Operator):
749     "Open a website in the web-browser"
750     bl_idname = "wm.url_open"
751     bl_label = ""
752
753     url = StringProperty(
754             name="URL",
755             description="URL to open",
756             )
757
758     def execute(self, context):
759         import webbrowser
760         webbrowser.open(self.url)
761         return {'FINISHED'}
762
763
764 class WM_OT_path_open(Operator):
765     "Open a path in a file browser"
766     bl_idname = "wm.path_open"
767     bl_label = ""
768
769     filepath = StringProperty(
770             subtype='FILE_PATH',
771             options={'SKIP_SAVE'},
772             )
773
774     def execute(self, context):
775         import sys
776         import os
777         import subprocess
778
779         filepath = self.filepath
780
781         if not filepath:
782             self.report({'ERROR'}, "File path was not set")
783             return {'CANCELLED'}
784
785         filepath = bpy.path.abspath(filepath)
786         filepath = os.path.normpath(filepath)
787
788         if not os.path.exists(filepath):
789             self.report({'ERROR'}, "File '%s' not found" % filepath)
790             return {'CANCELLED'}
791
792         if sys.platform[:3] == "win":
793             os.startfile(filepath)
794         elif sys.platform == "darwin":
795             subprocess.Popen(["open", filepath])
796         else:
797             try:
798                 subprocess.Popen(["xdg-open", filepath])
799             except OSError:
800                 # xdg-open *should* be supported by recent Gnome, KDE, Xfce
801                 pass
802
803         return {'FINISHED'}
804
805
806 def _wm_doc_get_id(doc_id, do_url=True, url_prefix=""):
807     id_split = doc_id.split(".")
808     url = rna = None
809
810     if len(id_split) == 1:  # rna, class
811         if do_url:
812             url = "%s/bpy.types.%s.html" % (url_prefix, id_split[0])
813         else:
814             rna = "bpy.types.%s" % id_split[0]
815
816     elif len(id_split) == 2:  # rna, class.prop
817         class_name, class_prop = id_split
818
819         if hasattr(bpy.types, class_name.upper() + "_OT_" + class_prop):
820             if do_url:
821                 url = ("%s/bpy.ops.%s.html#bpy.ops.%s.%s" % (url_prefix, class_name, class_name, class_prop))
822             else:
823                 rna = "bpy.ops.%s.%s" % (class_name, class_prop)
824         else:
825
826             # detect if this is a inherited member and use that name instead
827             rna_parent = getattr(bpy.types, class_name).bl_rna
828             rna_prop = rna_parent.properties[class_prop]
829             rna_parent = rna_parent.base
830             while rna_parent and rna_prop == rna_parent.properties.get(class_prop):
831                 class_name = rna_parent.identifier
832                 rna_parent = rna_parent.base
833
834             if do_url:
835                 url = ("%s/bpy.types.%s.html#bpy.types.%s.%s" % (url_prefix, class_name, class_name, class_prop))
836             else:
837                 rna = ("bpy.types.%s.%s" % (class_name, class_prop))
838
839     return url if do_url else rna
840
841
842 class WM_OT_doc_view_manual(Operator):
843     """Load online manual"""
844     bl_idname = "wm.doc_view_manual"
845     bl_label = "View Manual"
846
847     doc_id = doc_id
848
849     @staticmethod
850     def _find_reference(rna_id, url_mapping, verbose=True):
851         if verbose:
852             print("online manual check for: '%s'... " % rna_id)
853         from fnmatch import fnmatch
854         for pattern, url_suffix in url_mapping:
855             if fnmatch(rna_id, pattern):
856                 if verbose:
857                     print("            match found: '%s' --> '%s'" % (pattern, url_suffix))
858                 return url_suffix
859         if verbose:
860             print("match not found")
861         return None
862
863     @staticmethod
864     def _lookup_rna_url(rna_id, verbose=True):
865         url = None
866         for prefix, url_manual_mapping in bpy.utils.manual_map():
867             rna_ref = WM_OT_doc_view_manual._find_reference(rna_id, url_manual_mapping, verbose=verbose)
868             if rna_ref is not None:
869                 url = prefix + rna_ref
870                 break
871         return url
872
873     def execute(self, context):
874         rna_id = _wm_doc_get_id(self.doc_id, do_url=False)
875         if rna_id is None:
876             return {'PASS_THROUGH'}
877
878         url = self._lookup_rna_url(rna_id)
879
880         if url is None:
881             self.report({'WARNING'}, "No reference available %r, "
882                                      "Update info in 'rna_wiki_reference.py' "
883                                      " or callback to bpy.utils.manual_map()" %
884                                      self.doc_id)
885             return {'CANCELLED'}
886         else:
887             import webbrowser
888             webbrowser.open(url)
889             return {'FINISHED'}
890
891
892 class WM_OT_doc_view(Operator):
893     """Load online reference docs"""
894     bl_idname = "wm.doc_view"
895     bl_label = "View Documentation"
896
897     doc_id = doc_id
898     if bpy.app.version_cycle == "release":
899         _prefix = ("http://www.blender.org/documentation/blender_python_api_%s%s_release" %
900                    ("_".join(str(v) for v in bpy.app.version[:2]), bpy.app.version_char))
901     else:
902         _prefix = ("http://www.blender.org/documentation/blender_python_api_%s" %
903                    "_".join(str(v) for v in bpy.app.version))
904
905     def execute(self, context):
906         url = _wm_doc_get_id(self.doc_id, do_url=True, url_prefix=self._prefix)
907         if url is None:
908             return {'PASS_THROUGH'}
909
910         import webbrowser
911         webbrowser.open(url)
912
913         return {'FINISHED'}
914
915
916 class WM_OT_doc_edit(Operator):
917     """Load online reference docs"""
918     bl_idname = "wm.doc_edit"
919     bl_label = "Edit Documentation"
920
921     doc_id = doc_id
922     doc_new = doc_new
923
924     _url = "http://www.mindrones.com/blender/svn/xmlrpc.php"
925
926     def _send_xmlrpc(self, data_dict):
927         print("sending data:", data_dict)
928
929         import xmlrpc.client
930         user = "blenderuser"
931         pwd = "blender>user"
932
933         docblog = xmlrpc.client.ServerProxy(self._url)
934         docblog.metaWeblog.newPost(1, user, pwd, data_dict, 1)
935
936     def execute(self, context):
937
938         doc_id = self.doc_id
939         doc_new = self.doc_new
940
941         class_name, class_prop = doc_id.split('.')
942
943         if not doc_new:
944             self.report({'ERROR'}, "No input given for '%s'" % doc_id)
945             return {'CANCELLED'}
946
947         # check if this is an operator
948         op_name = class_name.upper() + '_OT_' + class_prop
949         op_class = getattr(bpy.types, op_name, None)
950
951         # Upload this to the web server
952         upload = {}
953
954         if op_class:
955             rna = op_class.bl_rna
956             doc_orig = rna.description
957             if doc_orig == doc_new:
958                 return {'RUNNING_MODAL'}
959
960             print("op - old:'%s' -> new:'%s'" % (doc_orig, doc_new))
961             upload["title"] = 'OPERATOR %s:%s' % (doc_id, doc_orig)
962         else:
963             rna = getattr(bpy.types, class_name).bl_rna
964             doc_orig = rna.properties[class_prop].description
965             if doc_orig == doc_new:
966                 return {'RUNNING_MODAL'}
967
968             print("rna - old:'%s' -> new:'%s'" % (doc_orig, doc_new))
969             upload["title"] = 'RNA %s:%s' % (doc_id, doc_orig)
970
971         upload["description"] = doc_new
972
973         self._send_xmlrpc(upload)
974
975         return {'FINISHED'}
976
977     def draw(self, context):
978         layout = self.layout
979         layout.label(text="Descriptor ID: '%s'" % self.doc_id)
980         layout.prop(self, "doc_new", text="")
981
982     def invoke(self, context, event):
983         wm = context.window_manager
984         return wm.invoke_props_dialog(self, width=600)
985
986
987 rna_path = StringProperty(
988         name="Property Edit",
989         description="Property data_path edit",
990         maxlen=1024,
991         options={'HIDDEN'},
992         )
993
994 rna_value = StringProperty(
995         name="Property Value",
996         description="Property value edit",
997         maxlen=1024,
998         )
999
1000 rna_property = StringProperty(
1001         name="Property Name",
1002         description="Property name edit",
1003         maxlen=1024,
1004         )
1005
1006 rna_min = FloatProperty(
1007         name="Min",
1008         default=0.0,
1009         precision=3,
1010         )
1011
1012 rna_max = FloatProperty(
1013         name="Max",
1014         default=1.0,
1015         precision=3,
1016         )
1017
1018
1019 class WM_OT_properties_edit(Operator):
1020     """Internal use (edit a property data_path)"""
1021     bl_idname = "wm.properties_edit"
1022     bl_label = "Edit Property"
1023     bl_options = {'REGISTER'}  # only because invoke_props_popup requires.
1024
1025     data_path = rna_path
1026     property = rna_property
1027     value = rna_value
1028     min = rna_min
1029     max = rna_max
1030     description = StringProperty(
1031             name="Tooltip",
1032             )
1033
1034     def execute(self, context):
1035         data_path = self.data_path
1036         value = self.value
1037         prop = self.property
1038
1039         prop_old = getattr(self, "_last_prop", [None])[0]
1040
1041         if prop_old is None:
1042             self.report({'ERROR'}, "Direct execution not supported")
1043             return {'CANCELLED'}
1044
1045         try:
1046             value_eval = eval(value)
1047         except:
1048             value_eval = value
1049
1050         # First remove
1051         item = eval("context.%s" % data_path)
1052
1053         rna_idprop_ui_prop_clear(item, prop_old)
1054         exec_str = "del item['%s']" % prop_old
1055         # print(exec_str)
1056         exec(exec_str)
1057
1058         # Reassign
1059         exec_str = "item['%s'] = %s" % (prop, repr(value_eval))
1060         # print(exec_str)
1061         exec(exec_str)
1062         self._last_prop[:] = [prop]
1063
1064         prop_type = type(item[prop])
1065
1066         prop_ui = rna_idprop_ui_prop_get(item, prop)
1067
1068         if prop_type in {float, int}:
1069             prop_ui["soft_min"] = prop_ui["min"] = prop_type(self.min)
1070             prop_ui["soft_max"] = prop_ui["max"] = prop_type(self.max)
1071
1072         prop_ui["description"] = self.description
1073
1074         # otherwise existing buttons which reference freed
1075         # memory may crash blender [#26510]
1076         # context.area.tag_redraw()
1077         for win in context.window_manager.windows:
1078             for area in win.screen.areas:
1079                 area.tag_redraw()
1080
1081         return {'FINISHED'}
1082
1083     def invoke(self, context, event):
1084         data_path = self.data_path
1085
1086         if not data_path:
1087             self.report({'ERROR'}, "Data path not set")
1088             return {'CANCELLED'}
1089
1090         self._last_prop = [self.property]
1091
1092         item = eval("context.%s" % data_path)
1093
1094         # setup defaults
1095         prop_ui = rna_idprop_ui_prop_get(item, self.property, False)  # don't create
1096         if prop_ui:
1097             self.min = prop_ui.get("min", -1000000000)
1098             self.max = prop_ui.get("max", 1000000000)
1099             self.description = prop_ui.get("description", "")
1100
1101         wm = context.window_manager
1102         return wm.invoke_props_dialog(self)
1103
1104
1105 class WM_OT_properties_add(Operator):
1106     """Internal use (edit a property data_path)"""
1107     bl_idname = "wm.properties_add"
1108     bl_label = "Add Property"
1109     bl_options = {'UNDO'}
1110
1111     data_path = rna_path
1112
1113     def execute(self, context):
1114         data_path = self.data_path
1115         item = eval("context.%s" % data_path)
1116
1117         def unique_name(names):
1118             prop = "prop"
1119             prop_new = prop
1120             i = 1
1121             while prop_new in names:
1122                 prop_new = prop + str(i)
1123                 i += 1
1124
1125             return prop_new
1126
1127         property = unique_name(item.keys())
1128
1129         item[property] = 1.0
1130         return {'FINISHED'}
1131
1132
1133 class WM_OT_properties_context_change(Operator):
1134     "Change the context tab in a Properties Window"
1135     bl_idname = "wm.properties_context_change"
1136     bl_label = ""
1137
1138     context = StringProperty(
1139             name="Context",
1140             maxlen=64,
1141             )
1142
1143     def execute(self, context):
1144         context.space_data.context = self.context
1145         return {'FINISHED'}
1146
1147
1148 class WM_OT_properties_remove(Operator):
1149     """Internal use (edit a property data_path)"""
1150     bl_idname = "wm.properties_remove"
1151     bl_label = "Remove Property"
1152     bl_options = {'UNDO'}
1153
1154     data_path = rna_path
1155     property = rna_property
1156
1157     def execute(self, context):
1158         data_path = self.data_path
1159         item = eval("context.%s" % data_path)
1160         del item[self.property]
1161         return {'FINISHED'}
1162
1163
1164 class WM_OT_keyconfig_activate(Operator):
1165     bl_idname = "wm.keyconfig_activate"
1166     bl_label = "Activate Keyconfig"
1167
1168     filepath = StringProperty(
1169             subtype='FILE_PATH',
1170             )
1171
1172     def execute(self, context):
1173         bpy.utils.keyconfig_set(self.filepath)
1174         return {'FINISHED'}
1175
1176
1177 class WM_OT_appconfig_default(Operator):
1178     bl_idname = "wm.appconfig_default"
1179     bl_label = "Default Application Configuration"
1180
1181     def execute(self, context):
1182         import os
1183
1184         context.window_manager.keyconfigs.active = context.window_manager.keyconfigs.default
1185
1186         filepath = os.path.join(bpy.utils.preset_paths("interaction")[0], "blender.py")
1187
1188         if os.path.exists(filepath):
1189             bpy.ops.script.execute_preset(filepath=filepath, menu_idname="USERPREF_MT_interaction_presets")
1190
1191         return {'FINISHED'}
1192
1193
1194 class WM_OT_appconfig_activate(Operator):
1195     bl_idname = "wm.appconfig_activate"
1196     bl_label = "Activate Application Configuration"
1197
1198     filepath = StringProperty(
1199             subtype='FILE_PATH',
1200             )
1201
1202     def execute(self, context):
1203         import os
1204         bpy.utils.keyconfig_set(self.filepath)
1205
1206         filepath = self.filepath.replace("keyconfig", "interaction")
1207
1208         if os.path.exists(filepath):
1209             bpy.ops.script.execute_preset(filepath=filepath, menu_idname="USERPREF_MT_interaction_presets")
1210
1211         return {'FINISHED'}
1212
1213
1214 class WM_OT_sysinfo(Operator):
1215     """Generate System Info"""
1216     bl_idname = "wm.sysinfo"
1217     bl_label = "System Info"
1218
1219     def execute(self, context):
1220         import sys_info
1221         sys_info.write_sysinfo(self)
1222         return {'FINISHED'}
1223
1224
1225 class WM_OT_copy_prev_settings(Operator):
1226     """Copy settings from previous version"""
1227     bl_idname = "wm.copy_prev_settings"
1228     bl_label = "Copy Previous Settings"
1229
1230     def execute(self, context):
1231         import os
1232         import shutil
1233         ver = bpy.app.version
1234         ver_old = ((ver[0] * 100) + ver[1]) - 1
1235         path_src = bpy.utils.resource_path('USER', ver_old // 100, ver_old % 100)
1236         path_dst = bpy.utils.resource_path('USER')
1237
1238         if os.path.isdir(path_dst):
1239             self.report({'ERROR'}, "Target path %r exists" % path_dst)
1240         elif not os.path.isdir(path_src):
1241             self.report({'ERROR'}, "Source path %r exists" % path_src)
1242         else:
1243             shutil.copytree(path_src, path_dst, symlinks=True)
1244
1245             # in 2.57 and earlier windows installers, system scripts were copied
1246             # into the configuration directory, don't want to copy those
1247             system_script = os.path.join(path_dst, "scripts/modules/bpy_types.py")
1248             if os.path.isfile(system_script):
1249                 shutil.rmtree(os.path.join(path_dst, "scripts"))
1250                 shutil.rmtree(os.path.join(path_dst, "plugins"))
1251
1252             # don't loose users work if they open the splash later.
1253             if bpy.data.is_saved is bpy.data.is_dirty is False:
1254                 bpy.ops.wm.read_homefile()
1255             else:
1256                 self.report({'INFO'}, "Reload Start-Up file to restore settings")
1257             return {'FINISHED'}
1258
1259         return {'CANCELLED'}
1260
1261
1262 class WM_OT_blenderplayer_start(Operator):
1263     """Launch the blender-player with the current blend-file"""
1264     bl_idname = "wm.blenderplayer_start"
1265     bl_label = "Start Game In Player"
1266
1267     def execute(self, context):
1268         import os
1269         import sys
1270         import subprocess
1271
1272         # these remain the same every execution
1273         blender_bin_path = bpy.app.binary_path
1274         blender_bin_dir = os.path.dirname(blender_bin_path)
1275         ext = os.path.splitext(blender_bin_path)[-1]
1276         player_path = os.path.join(blender_bin_dir, "blenderplayer" + ext)
1277         # done static vars
1278
1279         if sys.platform == "darwin":
1280             player_path = os.path.join(blender_bin_dir, "../../../blenderplayer.app/Contents/MacOS/blenderplayer")
1281
1282         if not os.path.exists(player_path):
1283             self.report({'ERROR'}, "Player path: %r not found" % player_path)
1284             return {'CANCELLED'}
1285
1286         filepath = os.path.join(bpy.app.tempdir, "game.blend")
1287         bpy.ops.wm.save_as_mainfile(filepath=filepath, check_existing=False, copy=True)
1288         subprocess.call([player_path, filepath])
1289         return {'FINISHED'}
1290
1291
1292 class WM_OT_keyconfig_test(Operator):
1293     "Test key-config for conflicts"
1294     bl_idname = "wm.keyconfig_test"
1295     bl_label = "Test Key Configuration for Conflicts"
1296
1297     def execute(self, context):
1298         from bpy_extras import keyconfig_utils
1299
1300         wm = context.window_manager
1301         kc = wm.keyconfigs.default
1302
1303         if keyconfig_utils.keyconfig_test(kc):
1304             print("CONFLICT")
1305
1306         return {'FINISHED'}
1307
1308
1309 class WM_OT_keyconfig_import(Operator):
1310     "Import key configuration from a python script"
1311     bl_idname = "wm.keyconfig_import"
1312     bl_label = "Import Key Configuration..."
1313
1314     filepath = StringProperty(
1315             subtype='FILE_PATH',
1316             default="keymap.py",
1317             )
1318     filter_folder = BoolProperty(
1319             name="Filter folders",
1320             default=True,
1321             options={'HIDDEN'},
1322             )
1323     filter_text = BoolProperty(
1324             name="Filter text",
1325             default=True,
1326             options={'HIDDEN'},
1327             )
1328     filter_python = BoolProperty(
1329             name="Filter python",
1330             default=True,
1331             options={'HIDDEN'},
1332             )
1333     keep_original = BoolProperty(
1334             name="Keep original",
1335             description="Keep original file after copying to configuration folder",
1336             default=True,
1337             )
1338
1339     def execute(self, context):
1340         import os
1341         from os.path import basename
1342         import shutil
1343
1344         if not self.filepath:
1345             self.report({'ERROR'}, "Filepath not set")
1346             return {'CANCELLED'}
1347
1348         config_name = basename(self.filepath)
1349
1350         path = bpy.utils.user_resource('SCRIPTS', os.path.join("presets", "keyconfig"), create=True)
1351         path = os.path.join(path, config_name)
1352
1353         try:
1354             if self.keep_original:
1355                 shutil.copy(self.filepath, path)
1356             else:
1357                 shutil.move(self.filepath, path)
1358         except Exception as e:
1359             self.report({'ERROR'}, "Installing keymap failed: %s" % e)
1360             return {'CANCELLED'}
1361
1362         # sneaky way to check we're actually running the code.
1363         bpy.utils.keyconfig_set(path)
1364
1365         return {'FINISHED'}
1366
1367     def invoke(self, context, event):
1368         wm = context.window_manager
1369         wm.fileselect_add(self)
1370         return {'RUNNING_MODAL'}
1371
1372 # This operator is also used by interaction presets saving - AddPresetBase
1373
1374
1375 class WM_OT_keyconfig_export(Operator):
1376     "Export key configuration to a python script"
1377     bl_idname = "wm.keyconfig_export"
1378     bl_label = "Export Key Configuration..."
1379
1380     filepath = StringProperty(
1381             subtype='FILE_PATH',
1382             default="keymap.py",
1383             )
1384     filter_folder = BoolProperty(
1385             name="Filter folders",
1386             default=True,
1387             options={'HIDDEN'},
1388             )
1389     filter_text = BoolProperty(
1390             name="Filter text",
1391             default=True,
1392             options={'HIDDEN'},
1393             )
1394     filter_python = BoolProperty(
1395             name="Filter python",
1396             default=True,
1397             options={'HIDDEN'},
1398             )
1399
1400     def execute(self, context):
1401         from bpy_extras import keyconfig_utils
1402
1403         if not self.filepath:
1404             raise Exception("Filepath not set")
1405
1406         if not self.filepath.endswith(".py"):
1407             self.filepath += ".py"
1408
1409         wm = context.window_manager
1410
1411         keyconfig_utils.keyconfig_export(wm,
1412                                          wm.keyconfigs.active,
1413                                          self.filepath,
1414                                          )
1415
1416         return {'FINISHED'}
1417
1418     def invoke(self, context, event):
1419         wm = context.window_manager
1420         wm.fileselect_add(self)
1421         return {'RUNNING_MODAL'}
1422
1423
1424 class WM_OT_keymap_restore(Operator):
1425     "Restore key map(s)"
1426     bl_idname = "wm.keymap_restore"
1427     bl_label = "Restore Key Map(s)"
1428
1429     all = BoolProperty(
1430             name="All Keymaps",
1431             description="Restore all keymaps to default",
1432             )
1433
1434     def execute(self, context):
1435         wm = context.window_manager
1436
1437         if self.all:
1438             for km in wm.keyconfigs.user.keymaps:
1439                 km.restore_to_default()
1440         else:
1441             km = context.keymap
1442             km.restore_to_default()
1443
1444         return {'FINISHED'}
1445
1446
1447 class WM_OT_keyitem_restore(Operator):
1448     "Restore key map item"
1449     bl_idname = "wm.keyitem_restore"
1450     bl_label = "Restore Key Map Item"
1451
1452     item_id = IntProperty(
1453             name="Item Identifier",
1454             description="Identifier of the item to remove",
1455             )
1456
1457     @classmethod
1458     def poll(cls, context):
1459         keymap = getattr(context, "keymap", None)
1460         return keymap
1461
1462     def execute(self, context):
1463         km = context.keymap
1464         kmi = km.keymap_items.from_id(self.item_id)
1465
1466         if (not kmi.is_user_defined) and kmi.is_user_modified:
1467             km.restore_item_to_default(kmi)
1468
1469         return {'FINISHED'}
1470
1471
1472 class WM_OT_keyitem_add(Operator):
1473     "Add key map item"
1474     bl_idname = "wm.keyitem_add"
1475     bl_label = "Add Key Map Item"
1476
1477     def execute(self, context):
1478         km = context.keymap
1479
1480         if km.is_modal:
1481             km.keymap_items.new_modal("", 'A', 'PRESS')
1482         else:
1483             km.keymap_items.new("none", 'A', 'PRESS')
1484
1485         # clear filter and expand keymap so we can see the newly added item
1486         if context.space_data.filter_text != "":
1487             context.space_data.filter_text = ""
1488             km.show_expanded_items = True
1489             km.show_expanded_children = True
1490
1491         return {'FINISHED'}
1492
1493
1494 class WM_OT_keyitem_remove(Operator):
1495     "Remove key map item"
1496     bl_idname = "wm.keyitem_remove"
1497     bl_label = "Remove Key Map Item"
1498
1499     item_id = IntProperty(
1500             name="Item Identifier",
1501             description="Identifier of the item to remove",
1502             )
1503
1504     @classmethod
1505     def poll(cls, context):
1506         return hasattr(context, "keymap")
1507
1508     def execute(self, context):
1509         km = context.keymap
1510         kmi = km.keymap_items.from_id(self.item_id)
1511         km.keymap_items.remove(kmi)
1512         return {'FINISHED'}
1513
1514
1515 class WM_OT_keyconfig_remove(Operator):
1516     "Remove key config"
1517     bl_idname = "wm.keyconfig_remove"
1518     bl_label = "Remove Key Config"
1519
1520     @classmethod
1521     def poll(cls, context):
1522         wm = context.window_manager
1523         keyconf = wm.keyconfigs.active
1524         return keyconf and keyconf.is_user_defined
1525
1526     def execute(self, context):
1527         wm = context.window_manager
1528         keyconfig = wm.keyconfigs.active
1529         wm.keyconfigs.remove(keyconfig)
1530         return {'FINISHED'}
1531
1532
1533 class WM_OT_operator_cheat_sheet(Operator):
1534     bl_idname = "wm.operator_cheat_sheet"
1535     bl_label = "Operator Cheat Sheet"
1536
1537     def execute(self, context):
1538         op_strings = []
1539         tot = 0
1540         for op_module_name in dir(bpy.ops):
1541             op_module = getattr(bpy.ops, op_module_name)
1542             for op_submodule_name in dir(op_module):
1543                 op = getattr(op_module, op_submodule_name)
1544                 text = repr(op)
1545                 if text.split("\n")[-1].startswith("bpy.ops."):
1546                     op_strings.append(text)
1547                     tot += 1
1548
1549             op_strings.append('')
1550
1551         textblock = bpy.data.texts.new("OperatorList.txt")
1552         textblock.write('# %d Operators\n\n' % tot)
1553         textblock.write('\n'.join(op_strings))
1554         self.report({'INFO'}, "See OperatorList.txt textblock")
1555         return {'FINISHED'}
1556
1557
1558 # -----------------------------------------------------------------------------
1559 # Addon Operators
1560
1561 class WM_OT_addon_enable(Operator):
1562     "Enable an addon"
1563     bl_idname = "wm.addon_enable"
1564     bl_label = "Enable Addon"
1565
1566     module = StringProperty(
1567             name="Module",
1568             description="Module name of the addon to enable",
1569             )
1570
1571     def execute(self, context):
1572         import addon_utils
1573
1574         mod = addon_utils.enable(self.module)
1575
1576         if mod:
1577             info = addon_utils.module_bl_info(mod)
1578
1579             info_ver = info.get("blender", (0, 0, 0))
1580
1581             if info_ver > bpy.app.version:
1582                 self.report({'WARNING'}, ("This script was written Blender "
1583                                           "version %d.%d.%d and might not "
1584                                           "function (correctly), "
1585                                           "though it is enabled") %
1586                                           info_ver)
1587             return {'FINISHED'}
1588         else:
1589             return {'CANCELLED'}
1590
1591
1592 class WM_OT_addon_disable(Operator):
1593     "Disable an addon"
1594     bl_idname = "wm.addon_disable"
1595     bl_label = "Disable Addon"
1596
1597     module = StringProperty(
1598             name="Module",
1599             description="Module name of the addon to disable",
1600             )
1601
1602     def execute(self, context):
1603         import addon_utils
1604
1605         addon_utils.disable(self.module)
1606         return {'FINISHED'}
1607
1608
1609 class WM_OT_theme_install(Operator):
1610     "Install a theme"
1611     bl_idname = "wm.theme_install"
1612     bl_label = "Install Theme..."
1613
1614     overwrite = BoolProperty(
1615             name="Overwrite",
1616             description="Remove existing theme file if exists",
1617             default=True,
1618             )
1619     filepath = StringProperty(
1620             subtype='FILE_PATH',
1621             )
1622     filter_folder = BoolProperty(
1623             name="Filter folders",
1624             default=True,
1625             options={'HIDDEN'},
1626             )
1627     filter_glob = StringProperty(
1628             default="*.xml",
1629             options={'HIDDEN'},
1630             )
1631
1632     def execute(self, context):
1633         import os
1634         import shutil
1635         import traceback
1636
1637         xmlfile = self.filepath
1638
1639         path_themes = bpy.utils.user_resource('SCRIPTS', "presets/interface_theme", create=True)
1640
1641         if not path_themes:
1642             self.report({'ERROR'}, "Failed to get themes path")
1643             return {'CANCELLED'}
1644
1645         path_dest = os.path.join(path_themes, os.path.basename(xmlfile))
1646
1647         if not self.overwrite:
1648             if os.path.exists(path_dest):
1649                 self.report({'WARNING'}, "File already installed to %r\n" % path_dest)
1650                 return {'CANCELLED'}
1651
1652         try:
1653             shutil.copyfile(xmlfile, path_dest)
1654             bpy.ops.script.execute_preset(filepath=path_dest, menu_idname="USERPREF_MT_interface_theme_presets")
1655
1656         except:
1657             traceback.print_exc()
1658             return {'CANCELLED'}
1659
1660         return {'FINISHED'}
1661
1662     def invoke(self, context, event):
1663         wm = context.window_manager
1664         wm.fileselect_add(self)
1665         return {'RUNNING_MODAL'}
1666
1667
1668 class WM_OT_addon_install(Operator):
1669     "Install an addon"
1670     bl_idname = "wm.addon_install"
1671     bl_label = "Install from File..."
1672
1673     overwrite = BoolProperty(
1674             name="Overwrite",
1675             description="Remove existing addons with the same ID",
1676             default=True,
1677             )
1678     target = EnumProperty(
1679             name="Target Path",
1680             items=(('DEFAULT', "Default", ""),
1681                    ('PREFS', "User Prefs", "")),
1682             )
1683
1684     filepath = StringProperty(
1685             subtype='FILE_PATH',
1686             )
1687     filter_folder = BoolProperty(
1688             name="Filter folders",
1689             default=True,
1690             options={'HIDDEN'},
1691             )
1692     filter_python = BoolProperty(
1693             name="Filter python",
1694             default=True,
1695             options={'HIDDEN'},
1696             )
1697     filter_glob = StringProperty(
1698             default="*.py;*.zip",
1699             options={'HIDDEN'},
1700             )
1701
1702     @staticmethod
1703     def _module_remove(path_addons, module):
1704         import os
1705         module = os.path.splitext(module)[0]
1706         for f in os.listdir(path_addons):
1707             f_base = os.path.splitext(f)[0]
1708             if f_base == module:
1709                 f_full = os.path.join(path_addons, f)
1710
1711                 if os.path.isdir(f_full):
1712                     os.rmdir(f_full)
1713                 else:
1714                     os.remove(f_full)
1715
1716     def execute(self, context):
1717         import addon_utils
1718         import traceback
1719         import zipfile
1720         import shutil
1721         import os
1722
1723         pyfile = self.filepath
1724
1725         if self.target == 'DEFAULT':
1726             # don't use bpy.utils.script_paths("addons") because we may not be able to write to it.
1727             path_addons = bpy.utils.user_resource('SCRIPTS', "addons", create=True)
1728         else:
1729             path_addons = bpy.context.user_preferences.filepaths.script_directory
1730             if path_addons:
1731                 path_addons = os.path.join(path_addons, "addons")
1732
1733         if not path_addons:
1734             self.report({'ERROR'}, "Failed to get addons path")
1735             return {'CANCELLED'}
1736
1737         # create dir is if missing.
1738         if not os.path.exists(path_addons):
1739             os.makedirs(path_addons)
1740
1741         # Check if we are installing from a target path,
1742         # doing so causes 2+ addons of same name or when the same from/to
1743         # location is used, removal of the file!
1744         addon_path = ""
1745         pyfile_dir = os.path.dirname(pyfile)
1746         for addon_path in addon_utils.paths():
1747             if os.path.samefile(pyfile_dir, addon_path):
1748                 self.report({'ERROR'}, "Source file is in the addon search path: %r" % addon_path)
1749                 return {'CANCELLED'}
1750         del addon_path
1751         del pyfile_dir
1752         # done checking for exceptional case
1753
1754         addons_old = {mod.__name__ for mod in addon_utils.modules(addon_utils.addons_fake_modules)}
1755
1756         #check to see if the file is in compressed format (.zip)
1757         if zipfile.is_zipfile(pyfile):
1758             try:
1759                 file_to_extract = zipfile.ZipFile(pyfile, 'r')
1760             except:
1761                 traceback.print_exc()
1762                 return {'CANCELLED'}
1763
1764             if self.overwrite:
1765                 for f in file_to_extract.namelist():
1766                     WM_OT_addon_install._module_remove(path_addons, f)
1767             else:
1768                 for f in file_to_extract.namelist():
1769                     path_dest = os.path.join(path_addons, os.path.basename(f))
1770                     if os.path.exists(path_dest):
1771                         self.report({'WARNING'}, "File already installed to %r\n" % path_dest)
1772                         return {'CANCELLED'}
1773
1774             try:  # extract the file to "addons"
1775                 file_to_extract.extractall(path_addons)
1776             except:
1777                 traceback.print_exc()
1778                 return {'CANCELLED'}
1779
1780         else:
1781             path_dest = os.path.join(path_addons, os.path.basename(pyfile))
1782
1783             if self.overwrite:
1784                 WM_OT_addon_install._module_remove(path_addons, os.path.basename(pyfile))
1785             elif os.path.exists(path_dest):
1786                 self.report({'WARNING'}, "File already installed to %r\n" % path_dest)
1787                 return {'CANCELLED'}
1788
1789             #if not compressed file just copy into the addon path
1790             try:
1791                 shutil.copyfile(pyfile, path_dest)
1792
1793             except:
1794                 traceback.print_exc()
1795                 return {'CANCELLED'}
1796
1797         addons_new = {mod.__name__ for mod in addon_utils.modules(addon_utils.addons_fake_modules)} - addons_old
1798         addons_new.discard("modules")
1799
1800         # disable any addons we may have enabled previously and removed.
1801         # this is unlikely but do just in case. bug [#23978]
1802         for new_addon in addons_new:
1803             addon_utils.disable(new_addon)
1804
1805         # possible the zip contains multiple addons, we could disallow this
1806         # but for now just use the first
1807         for mod in addon_utils.modules(addon_utils.addons_fake_modules):
1808             if mod.__name__ in addons_new:
1809                 info = addon_utils.module_bl_info(mod)
1810
1811                 # show the newly installed addon.
1812                 context.window_manager.addon_filter = 'All'
1813                 context.window_manager.addon_search = info["name"]
1814                 break
1815
1816         # in case a new module path was created to install this addon.
1817         bpy.utils.refresh_script_paths()
1818
1819         # print message
1820         msg = "Modules Installed from %r into %r (%s)" % (pyfile, path_addons, ", ".join(sorted(addons_new)))
1821         print(msg)
1822         self.report({'INFO'}, msg)
1823
1824         return {'FINISHED'}
1825
1826     def invoke(self, context, event):
1827         wm = context.window_manager
1828         wm.fileselect_add(self)
1829         return {'RUNNING_MODAL'}
1830
1831
1832 class WM_OT_addon_remove(Operator):
1833     "Disable an addon"
1834     bl_idname = "wm.addon_remove"
1835     bl_label = "Remove Addon"
1836
1837     module = StringProperty(
1838             name="Module",
1839             description="Module name of the addon to remove",
1840             )
1841
1842     @staticmethod
1843     def path_from_addon(module):
1844         import os
1845         import addon_utils
1846
1847         for mod in addon_utils.modules(addon_utils.addons_fake_modules):
1848             if mod.__name__ == module:
1849                 filepath = mod.__file__
1850                 if os.path.exists(filepath):
1851                     if os.path.splitext(os.path.basename(filepath))[0] == "__init__":
1852                         return os.path.dirname(filepath), True
1853                     else:
1854                         return filepath, False
1855         return None, False
1856
1857     def execute(self, context):
1858         import addon_utils
1859         import os
1860
1861         path, isdir = WM_OT_addon_remove.path_from_addon(self.module)
1862         if path is None:
1863             self.report({'WARNING'}, "Addon path %r could not be found" % path)
1864             return {'CANCELLED'}
1865
1866         # in case its enabled
1867         addon_utils.disable(self.module)
1868
1869         import shutil
1870         if isdir:
1871             shutil.rmtree(path)
1872         else:
1873             os.remove(path)
1874
1875         context.area.tag_redraw()
1876         return {'FINISHED'}
1877
1878     # lame confirmation check
1879     def draw(self, context):
1880         self.layout.label(text="Remove Addon: %r?" % self.module)
1881         path, isdir = WM_OT_addon_remove.path_from_addon(self.module)
1882         self.layout.label(text="Path: %r" % path)
1883
1884     def invoke(self, context, event):
1885         wm = context.window_manager
1886         return wm.invoke_props_dialog(self, width=600)
1887
1888
1889 class WM_OT_addon_expand(Operator):
1890     "Display more information on this addon"
1891     bl_idname = "wm.addon_expand"
1892     bl_label = ""
1893
1894     module = StringProperty(
1895             name="Module",
1896             description="Module name of the addon to expand",
1897             )
1898
1899     def execute(self, context):
1900         import addon_utils
1901
1902         module_name = self.module
1903
1904         # unlikely to fail, module should have already been imported
1905         try:
1906             # mod = __import__(module_name)
1907             mod = addon_utils.addons_fake_modules.get(module_name)
1908         except:
1909             import traceback
1910             traceback.print_exc()
1911             return {'CANCELLED'}
1912
1913         info = addon_utils.module_bl_info(mod)
1914         info["show_expanded"] = not info["show_expanded"]
1915         return {'FINISHED'}