pep8 whitespace commit
[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 sys
21 import bpy
22 from bpy.props import *
23
24
25 class CONSOLE_HT_header(bpy.types.Header):
26     bl_space_type = 'CONSOLE'
27
28     def draw(self, context):
29         sc = context.space_data
30         # text = sc.text
31         layout = self.layout
32
33         row = layout.row(align=True)
34         row.template_header()
35
36         if context.area.show_menus:
37             sub = row.row(align=True)
38
39             if sc.console_type == 'REPORT':
40                 sub.itemM("CONSOLE_MT_report")
41             else:
42                 sub.itemM("CONSOLE_MT_console")
43
44         layout.itemS()
45         layout.itemR(sc, "console_type", expand=True)
46
47         if sc.console_type == 'REPORT':
48             row = layout.row(align=True)
49             row.itemR(sc, "show_report_debug", text="Debug")
50             row.itemR(sc, "show_report_info", text="Info")
51             row.itemR(sc, "show_report_operator", text="Operators")
52             row.itemR(sc, "show_report_warn", text="Warnings")
53             row.itemR(sc, "show_report_error", text="Errors")
54
55             row = layout.row()
56             row.enabled = sc.show_report_operator
57             row.itemO("console.report_replay")
58         else:
59             row = layout.row(align=True)
60             row.itemO("console.autocomplete", text="Autocomplete")
61
62
63 class CONSOLE_MT_console(bpy.types.Menu):
64     bl_label = "Console"
65
66     def draw(self, context):
67         layout = self.layout
68         layout.column()
69         layout.itemO("console.clear")
70         layout.itemO("console.copy")
71         layout.itemO("console.paste")
72         layout.itemM("CONSOLE_MT_language")
73
74
75 class CONSOLE_MT_report(bpy.types.Menu):
76     bl_label = "Report"
77
78     def draw(self, context):
79         layout = self.layout
80         layout.column()
81         layout.itemO("console.select_all_toggle")
82         layout.itemO("console.select_border")
83         layout.itemO("console.report_delete")
84         layout.itemO("console.report_copy")
85
86 class CONSOLE_MT_language(bpy.types.Menu):
87     bl_label = "Languages..."
88
89     def draw(self, context):
90         layout = self.layout
91         layout.column()
92
93         mod = bpy.ops.console
94         languages = []
95         for opname in dir(mod):
96             # execute_python, execute_shell etc.
97             if opname.startswith("execute_"):
98                 languages.append(opname.split('_', 1)[-1])
99
100         languages.sort()
101
102         for language in languages:
103             layout.item_stringO("console.language", "language", language, text=language[0].upper() + language[1:])
104
105 def add_scrollback(text, text_type):
106     for l in text.split('\n'):
107         bpy.ops.console.scrollback_append(text=l.replace('\t', '    '),
108             type=text_type)
109
110
111 class ConsoleExec(bpy.types.Operator):
112     '''Execute the current console line as a python expression.'''
113     bl_idname = "console.execute"
114     bl_label = "Console Execute"
115     bl_register = False
116
117     def execute(self, context):
118         sc = context.space_data
119
120         execute = getattr(bpy.ops.console, "execute_" + sc.language, None)
121
122         if execute:
123             execute()
124         else:
125             print("Error: bpy.ops.console.execute_" + sc.language + " - not found")
126
127         return ('FINISHED',)
128
129
130 class ConsoleAutocomplete(bpy.types.Operator):
131     '''Evaluate the namespace up until the cursor and give a list of
132     options or complete the name if there is only one.'''
133     bl_idname = "console.autocomplete"
134     bl_label = "Console Autocomplete"
135     bl_register = False
136
137     def poll(self, context):
138         return context.space_data.console_type != 'REPORT'
139
140     def execute(self, context):
141         sc = context.space_data
142
143         autocomplete = getattr(bpy.ops.console, "autocomplete_" + sc.language, None)
144
145         if autocomplete:
146             autocomplete()
147         else:
148             print("Error: bpy.ops.console.autocomplete_" + sc.language + " - not found")
149
150         return ('FINISHED',)
151
152
153 class ConsoleBanner(bpy.types.Operator):
154     bl_idname = "console.banner"
155
156     def execute(self, context):
157         sc = context.space_data
158
159         # default to python
160         if not sc.language:
161             sc.language = 'python'
162
163         banner = getattr(bpy.ops.console, "banner_" + sc.language, None)
164
165         if banner:
166             banner()
167         else:
168             print("Error: bpy.ops.console.banner_" + sc.language + " - not found")
169
170         return ('FINISHED',)
171
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)