Fix keymap operator polling functions to be a bit safer.

This commit is contained in:
Martin Poirier 2011-03-07 23:53:08 +00:00
parent df3688a05d
commit 988886a73d

@ -710,8 +710,7 @@ class WM_OT_keyitem_restore(bpy.types.Operator):
@classmethod
def poll(cls, context):
km = context.keymap
return km.is_user_defined
return hasattr(context, "keymap") and context.keymap.is_user_defined
def execute(self, context):
km = context.keymap
@ -756,8 +755,7 @@ class WM_OT_keyitem_remove(bpy.types.Operator):
@classmethod
def poll(cls, context):
km = context.keymap
return km.is_user_defined
return hasattr(context, "keymap") and context.keymap.is_user_defined
def execute(self, context):
km = context.keymap