default name of keymap to 'keymap.py' + pedantic pep8 edits
[blender.git] / release / scripts / ui / space_console.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., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
16 #
17 # ##### END GPL LICENSE BLOCK #####
18
19 # <pep8 compliant>
20 import bpy
21 from bpy.props import *
22
23
24 class CONSOLE_HT_header(bpy.types.Header):
25     bl_space_type = 'CONSOLE'
26
27     def draw(self, context):
28         sc = context.space_data
29         # text = sc.text
30         layout = self.layout
31
32         row = layout.row(align=True)
33         row.template_header()
34
35         if context.area.show_menus:
36             sub = row.row(align=True)
37
38             if sc.console_type == 'REPORT':
39                 sub.itemM("CONSOLE_MT_report")
40             else:
41                 sub.itemM("CONSOLE_MT_console")
42
43         layout.itemS()
44         layout.itemR(sc, "console_type", expand=True)
45
46         if sc.console_type == 'REPORT':
47             row = layout.row(align=True)
48             row.itemR(sc, "show_report_debug", text="Debug")
49             row.itemR(sc, "show_report_info", text="Info")
50             row.itemR(sc, "show_report_operator", text="Operators")
51             row.itemR(sc, "show_report_warn", text="Warnings")
52             row.itemR(sc, "show_report_error", text="Errors")
53
54             row = layout.row()
55             row.enabled = sc.show_report_operator
56             row.itemO("console.report_replay")
57         else:
58             row = layout.row(align=True)
59             row.itemO("console.autocomplete", text="Autocomplete")
60
61
62 class CONSOLE_MT_console(bpy.types.Menu):
63     bl_label = "Console"
64
65     def draw(self, context):
66         layout = self.layout
67         layout.column()
68         layout.itemO("console.clear")
69         layout.itemO("console.copy")
70         layout.itemO("console.paste")
71         layout.itemM("CONSOLE_MT_language")
72
73
74 class CONSOLE_MT_report(bpy.types.Menu):
75     bl_label = "Report"
76
77     def draw(self, context):
78         layout = self.layout
79         layout.column()
80         layout.itemO("console.select_all_toggle")
81         layout.itemO("console.select_border")
82         layout.itemO("console.report_delete")
83         layout.itemO("console.report_copy")
84
85
86 class CONSOLE_MT_language(bpy.types.Menu):
87     bl_label = "Languages..."
88
89     def draw(self, context):
90         import sys
91         
92         layout = self.layout
93         layout.column()
94
95         # Collect modules with 'console_*.execute'
96         languages = []
97         for modname, mod in sys.modules.items():
98             if modname.startswith("console_") and hasattr(mod, "execute"):
99                 languages.append(modname.split('_', 1)[-1])
100
101         languages.sort()
102
103         for language in languages:
104             layout.item_stringO("console.language", "language", language, text=language[0].upper() + language[1:])
105
106
107 def add_scrollback(text, text_type):
108     for l in text.split('\n'):
109         bpy.ops.console.scrollback_append(text=l.replace('\t', '    '),
110             type=text_type)
111
112
113 class ConsoleExec(bpy.types.Operator):
114     '''Execute the current console line as a python expression.'''
115     bl_idname = "console.execute"
116     bl_label = "Console Execute"
117     bl_register = False
118
119     def execute(self, context):
120         sc = context.space_data
121
122         module = __import__("console_" + sc.language)
123         execute = getattr(module, "execute", None)
124
125         if execute:
126             return execute(context)
127         else:
128             print("Error: bpy.ops.console.execute_" + sc.language + " - not found")
129             return ('FINISHED',)
130
131
132 class ConsoleAutocomplete(bpy.types.Operator):
133     '''Evaluate the namespace up until the cursor and give a list of
134     options or complete the name if there is only one.'''
135     bl_idname = "console.autocomplete"
136     bl_label = "Console Autocomplete"
137     bl_register = False
138
139     def poll(self, context):
140         return context.space_data.console_type != 'REPORT'
141
142     def execute(self, context):
143         sc = context.space_data
144         module =  __import__("console_" + sc.language)
145         autocomplete = getattr(module, "autocomplete", None)
146
147         if autocomplete:
148             return autocomplete(context)
149         else:
150             print("Error: bpy.ops.console.autocomplete_" + sc.language + " - not found")
151             return ('FINISHED',)
152
153
154 class ConsoleBanner(bpy.types.Operator):
155     bl_idname = "console.banner"
156
157     def execute(self, context):
158         sc = context.space_data
159
160         # default to python
161         if not sc.language:
162             sc.language = 'python'
163
164         module =  __import__("console_" + sc.language)
165         banner = getattr(module, "banner", None)
166
167         if banner:
168             return banner(context)
169         else:
170             print("Error: bpy.ops.console.banner_" + sc.language + " - not found")
171             return ('FINISHED',)
172
173
174 class ConsoleLanguage(bpy.types.Operator):
175     '''Set the current language for this console'''
176     bl_idname = "console.language"
177     language = StringProperty(name="Language", maxlen=32, default="")
178
179     def execute(self, context):
180         sc = context.space_data
181
182         # defailt to python
183         sc.language = self.language
184
185         bpy.ops.console.banner()
186
187         # insert a new blank line
188         bpy.ops.console.history_append(text="", current_character=0,
189             remove_duplicates=True)
190
191         return ('FINISHED',)
192
193
194 bpy.types.register(CONSOLE_HT_header)
195 bpy.types.register(CONSOLE_MT_console)
196 bpy.types.register(CONSOLE_MT_report)
197 bpy.types.register(CONSOLE_MT_language)
198
199 # Stubs that call the language operators
200 bpy.ops.add(ConsoleExec)
201 bpy.ops.add(ConsoleAutocomplete)
202 bpy.ops.add(ConsoleBanner)
203
204 # Set the language and call the banner
205 bpy.ops.add(ConsoleLanguage)