used py error checking tools to fix some bugs & make pep8 corrections
[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         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
106 def add_scrollback(text, text_type):
107     for l in text.split('\n'):
108         bpy.ops.console.scrollback_append(text=l.replace('\t', '    '),
109             type=text_type)
110
111
112 class ConsoleExec(bpy.types.Operator):
113     '''Execute the current console line as a python expression.'''
114     bl_idname = "console.execute"
115     bl_label = "Console Execute"
116     bl_register = False
117
118     def execute(self, context):
119         sc = context.space_data
120
121         execute = getattr(bpy.ops.console, "execute_" + sc.language, None)
122
123         if execute:
124             execute()
125         else:
126             print("Error: bpy.ops.console.execute_" + sc.language + " - not found")
127
128         return ('FINISHED',)
129
130
131 class ConsoleAutocomplete(bpy.types.Operator):
132     '''Evaluate the namespace up until the cursor and give a list of
133     options or complete the name if there is only one.'''
134     bl_idname = "console.autocomplete"
135     bl_label = "Console Autocomplete"
136     bl_register = False
137
138     def poll(self, context):
139         return context.space_data.console_type != 'REPORT'
140
141     def execute(self, context):
142         sc = context.space_data
143
144         autocomplete = getattr(bpy.ops.console, "autocomplete_" + sc.language, None)
145
146         if autocomplete:
147             autocomplete()
148         else:
149             print("Error: bpy.ops.console.autocomplete_" + sc.language + " - not found")
150
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         banner = getattr(bpy.ops.console, "banner_" + sc.language, None)
165
166         if banner:
167             banner()
168         else:
169             print("Error: bpy.ops.console.banner_" + sc.language + " - not found")
170
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)