forked from bartvdbraak/blender
bbe13e7823
* operators now return sets (converted into flags) * can't remove bpy_operator_wrap.c since macro's still use the custom register funcs
214 lines
6.4 KiB
Python
214 lines
6.4 KiB
Python
# ##### BEGIN GPL LICENSE BLOCK #####
|
|
#
|
|
# This program is free software; you can redistribute it and/or
|
|
# modify it under the terms of the GNU General Public License
|
|
# as published by the Free Software Foundation; either version 2
|
|
# of the License, or (at your option) any later version.
|
|
#
|
|
# This program is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU General Public License
|
|
# along with this program; if not, write to the Free Software Foundation,
|
|
# Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
|
|
#
|
|
# ##### END GPL LICENSE BLOCK #####
|
|
|
|
# <pep8 compliant>
|
|
import bpy
|
|
from bpy.props import *
|
|
|
|
|
|
class CONSOLE_HT_header(bpy.types.Header):
|
|
bl_space_type = 'CONSOLE'
|
|
|
|
def draw(self, context):
|
|
sc = context.space_data
|
|
# text = sc.text
|
|
layout = self.layout
|
|
|
|
row = layout.row(align=True)
|
|
row.template_header()
|
|
|
|
if context.area.show_menus:
|
|
sub = row.row(align=True)
|
|
|
|
if sc.console_type == 'REPORT':
|
|
sub.menu("CONSOLE_MT_report")
|
|
else:
|
|
sub.menu("CONSOLE_MT_console")
|
|
|
|
layout.separator()
|
|
layout.prop(sc, "console_type", expand=True)
|
|
|
|
if sc.console_type == 'REPORT':
|
|
row = layout.row(align=True)
|
|
row.prop(sc, "show_report_debug", text="Debug")
|
|
row.prop(sc, "show_report_info", text="Info")
|
|
row.prop(sc, "show_report_operator", text="Operators")
|
|
row.prop(sc, "show_report_warn", text="Warnings")
|
|
row.prop(sc, "show_report_error", text="Errors")
|
|
|
|
row = layout.row()
|
|
row.enabled = sc.show_report_operator
|
|
row.operator("console.report_replay")
|
|
else:
|
|
row = layout.row(align=True)
|
|
row.operator("console.autocomplete", text="Autocomplete")
|
|
|
|
|
|
class CONSOLE_MT_console(bpy.types.Menu):
|
|
bl_label = "Console"
|
|
|
|
def draw(self, context):
|
|
layout = self.layout
|
|
layout.column()
|
|
layout.operator("console.clear")
|
|
layout.operator("console.copy")
|
|
layout.operator("console.paste")
|
|
layout.menu("CONSOLE_MT_language")
|
|
|
|
layout.separator()
|
|
|
|
layout.operator("screen.area_dupli")
|
|
layout.operator("screen.screen_full_area")
|
|
|
|
|
|
class CONSOLE_MT_report(bpy.types.Menu):
|
|
bl_label = "Report"
|
|
|
|
def draw(self, context):
|
|
layout = self.layout
|
|
layout.column()
|
|
layout.operator("console.select_all_toggle")
|
|
layout.operator("console.select_border")
|
|
layout.operator("console.report_delete")
|
|
layout.operator("console.report_copy")
|
|
|
|
|
|
class CONSOLE_MT_language(bpy.types.Menu):
|
|
bl_label = "Languages..."
|
|
|
|
def draw(self, context):
|
|
import sys
|
|
|
|
layout = self.layout
|
|
layout.column()
|
|
|
|
# Collect modules with 'console_*.execute'
|
|
languages = []
|
|
for modname, mod in sys.modules.items():
|
|
if modname.startswith("console_") and hasattr(mod, "execute"):
|
|
languages.append(modname.split('_', 1)[-1])
|
|
|
|
languages.sort()
|
|
|
|
for language in languages:
|
|
layout.operator("console.language", text=language[0].upper() + language[1:]).language = language
|
|
|
|
|
|
def add_scrollback(text, text_type):
|
|
for l in text.split('\n'):
|
|
bpy.ops.console.scrollback_append(text=l.replace('\t', ' '),
|
|
type=text_type)
|
|
|
|
|
|
class ConsoleExec(bpy.types.Operator):
|
|
'''Execute the current console line as a python expression.'''
|
|
bl_idname = "console.execute"
|
|
bl_label = "Console Execute"
|
|
bl_register = False
|
|
|
|
def execute(self, context):
|
|
sc = context.space_data
|
|
|
|
module = __import__("console_" + sc.language)
|
|
execute = getattr(module, "execute", None)
|
|
|
|
if execute:
|
|
return execute(context)
|
|
else:
|
|
print("Error: bpy.ops.console.execute_" + sc.language + " - not found")
|
|
return {'FINISHED'}
|
|
|
|
|
|
class ConsoleAutocomplete(bpy.types.Operator):
|
|
'''Evaluate the namespace up until the cursor and give a list of
|
|
options or complete the name if there is only one.'''
|
|
bl_idname = "console.autocomplete"
|
|
bl_label = "Console Autocomplete"
|
|
bl_register = False
|
|
|
|
def poll(self, context):
|
|
return context.space_data.console_type != 'REPORT'
|
|
|
|
def execute(self, context):
|
|
sc = context.space_data
|
|
module = __import__("console_" + sc.language)
|
|
autocomplete = getattr(module, "autocomplete", None)
|
|
|
|
if autocomplete:
|
|
return autocomplete(context)
|
|
else:
|
|
print("Error: bpy.ops.console.autocomplete_" + sc.language + " - not found")
|
|
return {'FINISHED'}
|
|
|
|
|
|
class ConsoleBanner(bpy.types.Operator):
|
|
'''Print a message whem the terminal initializes'''
|
|
bl_idname = "console.banner"
|
|
bl_label = "Console Banner"
|
|
|
|
def execute(self, context):
|
|
sc = context.space_data
|
|
|
|
# default to python
|
|
if not sc.language:
|
|
sc.language = 'python'
|
|
|
|
module = __import__("console_" + sc.language)
|
|
banner = getattr(module, "banner", None)
|
|
|
|
if banner:
|
|
return banner(context)
|
|
else:
|
|
print("Error: bpy.ops.console.banner_" + sc.language + " - not found")
|
|
return {'FINISHED'}
|
|
|
|
|
|
class ConsoleLanguage(bpy.types.Operator):
|
|
'''Set the current language for this console'''
|
|
bl_idname = "console.language"
|
|
bl_label = "Console Language"
|
|
language = StringProperty(name="Language", maxlen=32, default="")
|
|
|
|
def execute(self, context):
|
|
sc = context.space_data
|
|
|
|
# defailt to python
|
|
sc.language = self.properties.language
|
|
|
|
bpy.ops.console.banner()
|
|
|
|
# insert a new blank line
|
|
bpy.ops.console.history_append(text="", current_character=0,
|
|
remove_duplicates=True)
|
|
|
|
return {'FINISHED'}
|
|
|
|
|
|
bpy.types.register(CONSOLE_HT_header)
|
|
bpy.types.register(CONSOLE_MT_console)
|
|
bpy.types.register(CONSOLE_MT_report)
|
|
bpy.types.register(CONSOLE_MT_language)
|
|
|
|
# Stubs that call the language operators
|
|
bpy.types.register(ConsoleExec)
|
|
bpy.types.register(ConsoleAutocomplete)
|
|
bpy.types.register(ConsoleBanner)
|
|
|
|
# Set the language and call the banner
|
|
bpy.types.register(ConsoleLanguage)
|