f966f8a7e8b3eacd6ed7646854d6dcdfd4a12b34
[blender-staging.git] / release / scripts / ui / space_userpref_keymap.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 import bpy
21 import os
22
23 KM_MOD_PREFIX = "keyconfig_"
24
25 KM_HIERARCHY = [
26     ('Window', 'EMPTY', 'WINDOW', []), # file save, window change, exit
27     ('Screen', 'EMPTY', 'WINDOW', [    # full screen, undo, screenshot
28         ('Screen Editing', 'EMPTY', 'WINDOW', []),    # resizing, action corners
29         ]),
30
31     ('View2D', 'EMPTY', 'WINDOW', []),    # view 2d navigation (per region)
32     ('View2D Buttons List', 'EMPTY', 'WINDOW', []), # view 2d with buttons navigation
33     ('Header', 'EMPTY', 'WINDOW', []),    # header stuff (per region)
34     ('Grease Pencil', 'EMPTY', 'WINDOW', []), # grease pencil stuff (per region)
35
36     ('3D View', 'VIEW_3D', 'WINDOW', [ # view 3d navigation and generic stuff (select, transform)
37         ('Object Mode', 'EMPTY', 'WINDOW', []),
38         ('Mesh', 'EMPTY', 'WINDOW', []),
39         ('Curve', 'EMPTY', 'WINDOW', []),
40         ('Armature', 'EMPTY', 'WINDOW', []),
41         ('Metaball', 'EMPTY', 'WINDOW', []),
42         ('Lattice', 'EMPTY', 'WINDOW', []),
43         ('Font', 'EMPTY', 'WINDOW', []),
44
45         ('Pose', 'EMPTY', 'WINDOW', []),
46
47         ('Vertex Paint', 'EMPTY', 'WINDOW', []),
48         ('Weight Paint', 'EMPTY', 'WINDOW', []),
49         ('Face Mask', 'EMPTY', 'WINDOW', []),
50         ('Image Paint', 'EMPTY', 'WINDOW', []), # image and view3d
51         ('Sculpt', 'EMPTY', 'WINDOW', []),
52
53         ('Armature Sketch', 'EMPTY', 'WINDOW', []),
54         ('Particle', 'EMPTY', 'WINDOW', []),
55
56         ('Object Non-modal', 'EMPTY', 'WINDOW', []), # mode change
57
58         ('3D View Generic', 'VIEW_3D', 'WINDOW', [])    # toolbar and properties
59         ]),
60
61     ('Frames', 'EMPTY', 'WINDOW', []),    # frame navigation (per region)
62     ('Markers', 'EMPTY', 'WINDOW', []),    # markers (per region)
63     ('Animation', 'EMPTY', 'WINDOW', []),    # frame change on click, preview range (per region)
64     ('Animation Channels', 'EMPTY', 'WINDOW', []),
65     ('Graph Editor', 'GRAPH_EDITOR', 'WINDOW', [
66         ('Graph Editor Generic', 'GRAPH_EDITOR', 'WINDOW', [])
67         ]),
68     ('Dopesheet', 'DOPESHEET_EDITOR', 'WINDOW', []),
69     ('NLA Editor', 'NLA_EDITOR', 'WINDOW', [
70         ('NLA Channels', 'NLA_EDITOR', 'WINDOW', []),
71         ('NLA Generic', 'NLA_EDITOR', 'WINDOW', [])
72         ]),
73
74     ('Image', 'IMAGE_EDITOR', 'WINDOW', [
75         ('UV Editor', 'EMPTY', 'WINDOW', []), # image (reverse order, UVEdit before Image
76         ('Image Paint', 'EMPTY', 'WINDOW', []), # image and view3d
77         ('Image Generic', 'IMAGE_EDITOR', 'WINDOW', [])
78         ]),
79
80     ('Timeline', 'TIMELINE', 'WINDOW', []),
81     ('Outliner', 'OUTLINER', 'WINDOW', []),
82
83     ('Node Editor', 'NODE_EDITOR', 'WINDOW', [
84         ('Node Generic', 'NODE_EDITOR', 'WINDOW', [])
85         ]),
86     ('Sequencer', 'SEQUENCE_EDITOR', 'WINDOW', []),
87     ('Logic Editor', 'LOGIC_EDITOR', 'WINDOW', []),
88
89     ('File Browser', 'FILE_BROWSER', 'WINDOW', [
90         ('File Browser Main', 'FILE_BROWSER', 'WINDOW', []),
91         ('File Browser Buttons', 'FILE_BROWSER', 'WINDOW', [])
92         ]),
93
94     ('Property Editor', 'PROPERTIES', 'WINDOW', []), # align context menu
95
96     ('Script', 'SCRIPTS_WINDOW', 'WINDOW', []),
97     ('Text', 'TEXT_EDITOR', 'WINDOW', []),
98     ('Console', 'CONSOLE', 'WINDOW', []),
99
100     ('View3D Gesture Circle', 'EMPTY', 'WINDOW', []),
101     ('Gesture Border', 'EMPTY', 'WINDOW', []),
102     ('Standard Modal Map', 'EMPTY', 'WINDOW', []),
103     ('Transform Modal Map', 'EMPTY', 'WINDOW', []),
104     ('View3D Fly Modal', 'EMPTY', 'WINDOW', []),
105     ('View3D Rotate Modal', 'EMPTY', 'WINDOW', []),
106     ('View3D Move Modal', 'EMPTY', 'WINDOW', []),
107     ('View3D Zoom Modal', 'EMPTY', 'WINDOW', []),
108     ]
109
110
111 def _km_exists_in(km, export_keymaps):
112     for km2, kc in export_keymaps:
113         if km2.name == km.name:
114             return True
115     return False
116
117
118 def _merge_keymaps(kc1, kc2):
119     """ note: kc1 takes priority over kc2
120     """
121     merged_keymaps = [(km, kc1) for km in kc1.keymaps]
122     if kc1 != kc2:
123         merged_keymaps.extend([(km, kc2) for km in kc2.keymaps if not _km_exists_in(km, merged_keymaps)])
124
125     return merged_keymaps
126
127
128 class InputKeyMapPanel(bpy.types.Panel):
129     bl_space_type = 'USER_PREFERENCES'
130     bl_label = "Input"
131     bl_region_type = 'WINDOW'
132     bl_show_header = False
133
134     def draw_entry(self, display_keymaps, entry, col, level=0):
135         idname, spaceid, regionid, children = entry
136
137         for km, kc in display_keymaps:
138             if km.name == idname and km.space_type == spaceid and km.region_type == regionid:
139                 self.draw_km(display_keymaps, kc, km, children, col, level)
140
141         '''
142         km = kc.find_keymap(idname, space_type=spaceid, region_type=regionid)
143         if not km:
144             kc = defkc
145             km = kc.find_keymap(idname, space_type=spaceid, region_type=regionid)
146
147         if km:
148             self.draw_km(kc, km, children, col, level)
149         '''
150
151     def indented_layout(self, layout, level):
152         indentpx = 16
153         if level == 0:
154             level = 0.0001   # Tweak so that a percentage of 0 won't split by half
155         indent = level * indentpx / bpy.context.region.width
156
157         split = layout.split(percentage=indent)
158         col = split.column()
159         col = split.column()
160         return col
161
162     def draw_km(self, display_keymaps, kc, km, children, layout, level):
163         km = km.active()
164
165         layout.set_context_pointer("keymap", km)
166
167         col = self.indented_layout(layout, level)
168
169         row = col.row()
170         row.prop(km, "show_expanded_children", text="", emboss=False)
171         row.label(text=km.name)
172
173         row.label()
174         row.label()
175
176         if km.is_modal:
177             row.label(text="", icon='LINKED')
178         if km.is_user_defined:
179             op = row.operator("wm.keymap_restore", text="Restore")
180         else:
181             op = row.operator("wm.keymap_edit", text="Edit")
182
183         if km.show_expanded_children:
184             if children:
185                 # Put the Parent key map's entries in a 'global' sub-category
186                 # equal in hierarchy to the other children categories
187                 subcol = self.indented_layout(col, level + 1)
188                 subrow = subcol.row()
189                 subrow.prop(km, "show_expanded_items", text="", emboss=False)
190                 subrow.label(text="%s (Global)" % km.name)
191             else:
192                 km.show_expanded_items = True
193
194             # Key Map items
195             if km.show_expanded_items:
196                 for kmi in km.items:
197                     self.draw_kmi(display_keymaps, kc, km, kmi, col, level + 1)
198
199                 # "Add New" at end of keymap item list
200                 col = self.indented_layout(col, level + 1)
201                 subcol = col.split(percentage=0.2).column()
202                 subcol.enabled = km.is_user_defined
203                 op = subcol.operator("wm.keyitem_add", text="Add New", icon='ZOOMIN')
204
205             col.separator()
206
207             # Child key maps
208             if children:
209                 subcol = col.column()
210                 row = subcol.row()
211
212                 for entry in children:
213                     self.draw_entry(display_keymaps, entry, col, level + 1)
214
215     def draw_kmi(self, display_keymaps, kc, km, kmi, layout, level):
216         map_type = kmi.map_type
217
218         col = self.indented_layout(layout, level)
219
220         if km.is_user_defined:
221             col = col.column(align=True)
222             box = col.box()
223         else:
224             box = col.column()
225
226         split = box.split(percentage=0.05)
227
228         # header bar
229         row = split.row()
230         row.prop(kmi, "show_expanded", text="", emboss=False)
231
232         row = split.row()
233         row.enabled = km.is_user_defined
234         row.prop(kmi, "active", text="", emboss=False)
235
236         if km.is_modal:
237             row.prop(kmi, "propvalue", text="")
238         else:
239             row.label(text=kmi.name)
240
241         row = split.row()
242         row.enabled = km.is_user_defined
243         row.prop(kmi, "map_type", text="")
244         if map_type == 'KEYBOARD':
245             row.prop(kmi, "type", text="", full_event=True)
246         elif map_type == 'MOUSE':
247             row.prop(kmi, "type", text="", full_event=True)
248         elif map_type == 'TWEAK':
249             subrow = row.row()
250             subrow.prop(kmi, "type", text="")
251             subrow.prop(kmi, "value", text="")
252         elif map_type == 'TIMER':
253             row.prop(kmi, "type", text="")
254         else:
255             row.label()
256
257         if kmi.id:
258             op = row.operator("wm.keyitem_restore", text="", icon='BACK')
259             op.item_id = kmi.id
260         op = row.operator("wm.keyitem_remove", text="", icon='X')
261         op.item_id = kmi.id
262
263         # Expanded, additional event settings
264         if kmi.show_expanded:
265             box = col.box()
266
267             box.enabled = km.is_user_defined
268
269             if map_type not in ('TEXTINPUT', 'TIMER'):
270                 split = box.split(percentage=0.4)
271                 sub = split.row()
272
273                 if km.is_modal:
274                     sub.prop(kmi, "propvalue", text="")
275                 else:
276                     sub.prop(kmi, "idname", text="")
277
278                 sub = split.column()
279                 subrow = sub.row(align=True)
280
281                 if map_type == 'KEYBOARD':
282                     subrow.prop(kmi, "type", text="", event=True)
283                     subrow.prop(kmi, "value", text="")
284                 elif map_type == 'MOUSE':
285                     subrow.prop(kmi, "type", text="")
286                     subrow.prop(kmi, "value", text="")
287
288                 subrow = sub.row()
289                 subrow.scale_x = 0.75
290                 subrow.prop(kmi, "any")
291                 subrow.prop(kmi, "shift")
292                 subrow.prop(kmi, "ctrl")
293                 subrow.prop(kmi, "alt")
294                 subrow.prop(kmi, "oskey", text="Cmd")
295                 subrow.prop(kmi, "key_modifier", text="", event=True)
296
297             def display_properties(properties, title=None):
298                 box.separator()
299                 if title:
300                     box.label(text=title)
301                 flow = box.column_flow(columns=2)
302                 for pname, value in properties.items():
303                     if not properties.is_property_hidden(pname):
304                         if isinstance(value, bpy.types.OperatorProperties):
305                             display_properties(value, title=pname)
306                         else:
307                             flow.prop(properties, pname)
308
309             # Operator properties
310             props = kmi.properties
311             if props is not None:
312                 display_properties(props)
313
314             # Modal key maps attached to this operator
315             if not km.is_modal:
316                 kmm = kc.find_keymap_modal(kmi.idname)
317                 if kmm:
318                     self.draw_km(display_keymaps, kc, kmm, None, layout, level + 1)
319                     layout.set_context_pointer("keymap", km)
320
321     def draw_filtered(self, display_keymaps, filter, layout):
322         for km, kc in display_keymaps:
323             km = km.active()
324             layout.set_context_pointer("keymap", km)
325
326             filtered_items = [kmi for kmi in km.items if filter in kmi.name.lower()]
327
328             if len(filtered_items) != 0:
329                 col = layout.column()
330
331                 row = col.row()
332                 row.label(text=km.name, icon="DOT")
333
334                 row.label()
335                 row.label()
336
337                 if km.is_user_defined:
338                     op = row.operator("wm.keymap_restore", text="Restore")
339                 else:
340                     op = row.operator("wm.keymap_edit", text="Edit")
341
342                 for kmi in filtered_items:
343                     self.draw_kmi(display_keymaps, kc, km, kmi, col, 1)
344
345                 # "Add New" at end of keymap item list
346                 col = self.indented_layout(layout, 1)
347                 subcol = col.split(percentage=0.2).column()
348                 subcol.enabled = km.is_user_defined
349                 op = subcol.operator("wm.keyitem_add", text="Add New", icon='ZOOMIN')
350
351     def draw_hierarchy(self, display_keymaps, layout):
352         for entry in KM_HIERARCHY:
353             self.draw_entry(display_keymaps, entry, layout)
354
355     def draw_keymaps(self, context, layout):
356         wm = context.manager
357         kc = wm.active_keyconfig
358         defkc = wm.default_keyconfig
359
360         col = layout.column()
361         sub = col.column()
362
363         subsplit = sub.split()
364         subcol = subsplit.column()
365
366         row = subcol.row()
367         row.prop_object(wm, "active_keyconfig", wm, "keyconfigs", text="Key Config:")
368         layout.set_context_pointer("keyconfig", wm.active_keyconfig)
369         row.operator("wm.keyconfig_remove", text="", icon='X')
370
371         row.prop(context.space_data, "filter_text", icon="VIEWZOOM")
372
373         col.separator()
374
375         display_keymaps = _merge_keymaps(kc, defkc)
376         if context.space_data.filter_text != "":
377             filter_text = context.space_data.filter_text.lower()
378             self.draw_filtered(display_keymaps, filter, col)
379         else:
380             self.draw_hierarchy(display_keymaps, col)
381
382
383 from bpy.props import *
384
385
386 def export_properties(prefix, properties, lines=None):
387     if lines is None:
388         lines = []
389
390     for value, pname in properties.items():
391         if not properties.is_property_hidden(pname):
392             if isinstance(value, bpy.types.OperatorProperties):
393                 export_properties(prefix + "." + pname, value, lines)
394             elif properties.is_property_set(pname):
395                 value = _string_value(value)
396                 if value != "":
397                     lines.append("%s.%s = %s\n" % (prefix, pname, value))
398     return lines
399
400
401 class WM_OT_keyconfig_test(bpy.types.Operator):
402     "Test keyconfig for conflicts"
403     bl_idname = "wm.keyconfig_test"
404     bl_label = "Test Key Configuration for Conflicts"
405
406     def testEntry(self, kc, entry, src=None, parent=None):
407         result = False
408
409         def kmistr(kmi):
410             if km.is_modal:
411                 s = ["kmi = km.items.add_modal(\'%s\', \'%s\', \'%s\'" % (kmi.propvalue, kmi.type, kmi.value)]
412             else:
413                 s = ["kmi = km.items.add(\'%s\', \'%s\', \'%s\'" % (kmi.idname, kmi.type, kmi.value)]
414
415             if kmi.any:
416                 s.append(", any=True")
417             else:
418                 if kmi.shift:
419                     s.append(", shift=True")
420                 if kmi.ctrl:
421                     s.append(", ctrl=True")
422                 if kmi.alt:
423                     s.append(", alt=True")
424                 if kmi.oskey:
425                     s.append(", oskey=True")
426             if kmi.key_modifier and kmi.key_modifier != 'NONE':
427                 s.append(", key_modifier=\'%s\'" % kmi.key_modifier)
428
429             s.append(")\n")
430
431             props = kmi.properties
432
433             if props is not None:
434                 export_properties("kmi.properties", props, s)
435
436             return "".join(s).strip()
437
438         idname, spaceid, regionid, children = entry
439
440         km = kc.find_keymap(idname, space_type=spaceid, region_type=regionid)
441
442         if km:
443             km = km.active()
444
445             if src:
446                 for item in km.items:
447                     if src.compare(item):
448                         print("===========")
449                         print(parent.name)
450                         print(kmistr(src))
451                         print(km.name)
452                         print(kmistr(item))
453                         result = True
454
455                 for child in children:
456                     if self.testEntry(kc, child, src, parent):
457                         result = True
458             else:
459                 for i in range(len(km.items)):
460                     src = km.items[i]
461
462                     for child in children:
463                         if self.testEntry(kc, child, src, km):
464                             result = True
465
466                     for j in range(len(km.items) - i - 1):
467                         item = km.items[j + i + 1]
468                         if src.compare(item):
469                             print("===========")
470                             print(km.name)
471                             print(kmistr(src))
472                             print(kmistr(item))
473                             result = True
474
475                 for child in children:
476                     if self.testEntry(kc, child):
477                         result = True
478
479         return result
480
481     def testConfig(self, kc):
482         result = False
483         for entry in KM_HIERARCHY:
484             if self.testEntry(kc, entry):
485                 result = True
486         return result
487
488     def execute(self, context):
489         wm = context.manager
490         kc = wm.default_keyconfig
491
492         if self.testConfig(kc):
493             print("CONFLICT")
494
495         return {'FINISHED'}
496
497
498 def _string_value(value):
499     if isinstance(value, str) or isinstance(value, bool) or isinstance(value, float) or isinstance(value, int):
500         result = repr(value)
501     elif getattr(value, '__len__', False):
502         repr(list(value))
503     else:
504         print("Export key configuration: can't write ", value)
505
506     return result
507
508
509 class WM_OT_keyconfig_import(bpy.types.Operator):
510     "Import key configuration from a python script"
511     bl_idname = "wm.keyconfig_import"
512     bl_label = "Import Key Configuration..."
513
514     filepath = StringProperty(name="File Path", description="Filepath to write file to", default="keymap.py")
515     filter_folder = BoolProperty(name="Filter folders", description="", default=True, options={'HIDDEN'})
516     filter_text = BoolProperty(name="Filter text", description="", default=True, options={'HIDDEN'})
517     filter_python = BoolProperty(name="Filter python", description="", default=True, options={'HIDDEN'})
518
519     keep_original = BoolProperty(name="Keep original", description="Keep original file after copying to configuration folder", default=True)
520
521     def execute(self, context):
522         import shutil
523         if not self.properties.is_property_set("filepath"):
524             raise Exception("Filepath not set")
525
526         f = open(self.properties.filepath, "r")
527         if not f:
528             raise Exception("Could not open file")
529
530         config_name = None
531         for line in f:
532             if line.startswith("kc = wm.add_keyconfig("):
533                 config_name = line[23:-3]
534                 break
535
536         if config_name is None:
537             raise Exception("config name not found")
538
539         path = os.path.join(__file__, "..", "..", "cfg") # remove ui/space_userpref.py
540         path = os.path.normpath(path)
541         print(path)
542
543         # create config folder if needed
544         if not os.path.exists(path):
545             os.mkdir(path)
546
547         path = os.path.join(path, config_name + ".py")
548
549         if self.properties.keep_original:
550             shutil.copy(self.properties.filepath, path)
551         else:
552             shutil.move(self.properties.filepath, path)
553
554         # sneaky way to check we're actually running the code.
555         wm = context.manager
556         while config_name in wm.keyconfigs:
557             wm.remove_keyconfig(wm.keyconfigs[config_name])
558
559         wm = context.manager
560         totmap = len(wm.keyconfigs)
561         mod = __import__(config_name)
562         if totmap == len(wm.keyconfigs):
563             reload(mod)
564
565         wm = bpy.context.manager
566         wm.active_keyconfig = wm.keyconfigs[config_name]
567
568         return {'FINISHED'}
569
570     def invoke(self, context, event):
571         wm = context.manager
572         wm.add_fileselect(self)
573         return {'RUNNING_MODAL'}
574
575 # This operator is also used by interaction presets saving - AddPresetBase
576
577
578 class WM_OT_keyconfig_export(bpy.types.Operator):
579     "Export key configuration to a python script"
580     bl_idname = "wm.keyconfig_export"
581     bl_label = "Export Key Configuration..."
582
583     filepath = StringProperty(name="File Path", description="Filepath to write file to", default="keymap.py")
584     filter_folder = BoolProperty(name="Filter folders", description="", default=True, options={'HIDDEN'})
585     filter_text = BoolProperty(name="Filter text", description="", default=True, options={'HIDDEN'})
586     filter_python = BoolProperty(name="Filter python", description="", default=True, options={'HIDDEN'})
587     kc_name = StringProperty(name="KeyConfig Name", description="Name to save the key config as")
588
589     def execute(self, context):
590         if not self.properties.is_property_set("filepath"):
591             raise Exception("Filepath not set")
592
593         f = open(self.properties.filepath, "w")
594         if not f:
595             raise Exception("Could not open file")
596
597         wm = context.manager
598         kc = wm.active_keyconfig
599
600         if self.properties.kc_name != '':
601             name = self.properties.kc_name
602         elif kc.name == 'Blender':
603             name = os.path.splitext(os.path.basename(self.properties.filepath))[0]
604         else:
605             name = kc.name
606
607         f.write("# Configuration %s\n" % name)
608
609         f.write("import bpy\n\n")
610         f.write("wm = bpy.context.manager\n")
611         f.write("kc = wm.add_keyconfig('%s')\n\n" % name)
612
613         # Generate a list of keymaps to export:
614         #
615         # First add all user_defined keymaps (found in inputs.edited_keymaps list),
616         # then add all remaining keymaps from the currently active custom keyconfig.
617         #
618         # This will create a final list of keymaps that can be used as a 'diff' against
619         # the default blender keyconfig, recreating the current setup from a fresh blender
620         # without needing to export keymaps which haven't been edited.
621
622         class FakeKeyConfig():
623             keymaps = []
624         edited_kc = FakeKeyConfig()
625         edited_kc.keymaps.extend(context.user_preferences.inputs.edited_keymaps)
626         # merge edited keymaps with non-default keyconfig, if it exists
627         if kc != wm.default_keyconfig:
628             export_keymaps = _merge_keymaps(edited_kc, kc)
629         else:
630             export_keymaps = _merge_keymaps(edited_kc, edited_kc)
631
632         for km, kc_x in export_keymaps:
633
634             km = km.active()
635
636             f.write("# Map %s\n" % km.name)
637             f.write("km = kc.add_keymap('%s', space_type='%s', region_type='%s', modal=%s)\n\n" % (km.name, km.space_type, km.region_type, km.is_modal))
638             for kmi in km.items:
639                 if km.is_modal:
640                     f.write("kmi = km.items.add_modal('%s', '%s', '%s'" % (kmi.propvalue, kmi.type, kmi.value))
641                 else:
642                     f.write("kmi = km.items.add('%s', '%s', '%s'" % (kmi.idname, kmi.type, kmi.value))
643                 if kmi.any:
644                     f.write(", any=True")
645                 else:
646                     if kmi.shift:
647                         f.write(", shift=True")
648                     if kmi.ctrl:
649                         f.write(", ctrl=True")
650                     if kmi.alt:
651                         f.write(", alt=True")
652                     if kmi.oskey:
653                         f.write(", oskey=True")
654                 if kmi.key_modifier and kmi.key_modifier != 'NONE':
655                     f.write(", key_modifier='%s'" % kmi.key_modifier)
656                 f.write(")\n")
657
658                 props = kmi.properties
659
660                 if props is not None:
661                     f.write("".join(export_properties("kmi.properties", props)))
662
663             f.write("\n")
664
665         f.close()
666
667         return {'FINISHED'}
668
669     def invoke(self, context, event):
670         wm = context.manager
671         wm.add_fileselect(self)
672         return {'RUNNING_MODAL'}
673
674
675 class WM_OT_keymap_edit(bpy.types.Operator):
676     "Edit stored key map"
677     bl_idname = "wm.keymap_edit"
678     bl_label = "Edit Key Map"
679
680     def execute(self, context):
681         wm = context.manager
682         km = context.keymap
683         km.copy_to_user()
684         return {'FINISHED'}
685
686
687 class WM_OT_keymap_restore(bpy.types.Operator):
688     "Restore key map(s)"
689     bl_idname = "wm.keymap_restore"
690     bl_label = "Restore Key Map(s)"
691
692     all = BoolProperty(attr="all", name="All Keymaps", description="Restore all keymaps to default")
693
694     def execute(self, context):
695         wm = context.manager
696
697         if self.properties.all:
698             for km in wm.default_keyconfig.keymaps:
699                 km.restore_to_default()
700         else:
701             km = context.keymap
702             km.restore_to_default()
703
704         return {'FINISHED'}
705
706
707 class WM_OT_keyitem_restore(bpy.types.Operator):
708     "Restore key map item"
709     bl_idname = "wm.keyitem_restore"
710     bl_label = "Restore Key Map Item"
711
712     item_id = IntProperty(attr="item_id", name="Item Identifier", description="Identifier of the item to remove")
713
714     def execute(self, context):
715         wm = context.manager
716         km = context.keymap
717         kmi = km.item_from_id(self.properties.item_id)
718
719         km.restore_item_to_default(kmi)
720
721         return {'FINISHED'}
722
723
724 class WM_OT_keyitem_add(bpy.types.Operator):
725     "Add key map item"
726     bl_idname = "wm.keyitem_add"
727     bl_label = "Add Key Map Item"
728
729     def execute(self, context):
730         wm = context.manager
731         km = context.keymap
732         kc = wm.default_keyconfig
733
734         if km.is_modal:
735             km.items.add_modal("", 'A', 'PRESS') # kmi
736         else:
737             km.items.add("none", 'A', 'PRESS') # kmi
738
739         # clear filter and expand keymap so we can see the newly added item
740         if context.space_data.filter_text != "":
741             context.space_data.filter_text = ""
742             km.show_expanded_items = True
743             km.show_expanded_children = True
744
745         return {'FINISHED'}
746
747
748 class WM_OT_keyitem_remove(bpy.types.Operator):
749     "Remove key map item"
750     bl_idname = "wm.keyitem_remove"
751     bl_label = "Remove Key Map Item"
752
753     item_id = IntProperty(attr="item_id", name="Item Identifier", description="Identifier of the item to remove")
754
755     def execute(self, context):
756         wm = context.manager
757         km = context.keymap
758         kmi = km.item_from_id(self.properties.item_id)
759         km.remove_item(kmi)
760         return {'FINISHED'}
761
762
763 class WM_OT_keyconfig_remove(bpy.types.Operator):
764     "Remove key config"
765     bl_idname = "wm.keyconfig_remove"
766     bl_label = "Remove Key Config"
767
768     @classmethod
769     def poll(cls, context):
770         wm = context.manager
771         return wm.active_keyconfig.is_user_defined
772
773     def execute(self, context):
774         import sys
775         wm = context.manager
776
777         keyconfig = wm.active_keyconfig
778
779         module = sys.modules.get(keyconfig.name)
780
781         if module:
782             path = module.__file__
783             if os.path.exists(path):
784                 os.remove(path)
785
786             path = module.__file__ + "c" # for .pyc
787
788             if os.path.exists(path):
789                 os.remove(path)
790
791         wm.remove_keyconfig(keyconfig)
792         return {'FINISHED'}
793
794 def register():
795     pass
796
797 def unregister():
798     pass
799
800 if __name__ == "__main__":
801     register()