Merging r39312 through r39329 from trunk into soc-2011-tomato
[blender.git] / release / scripts / startup / bl_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., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
16 #
17 # ##### END GPL LICENSE BLOCK #####
18
19 # <pep8 compliant>
20 import bpy
21 from bpy.types import Header, Menu, Operator
22 from bpy.props import StringProperty
23
24
25 class CONSOLE_HT_header(Header):
26     bl_space_type = 'CONSOLE'
27
28     def draw(self, context):
29         layout = self.layout
30
31         row = layout.row(align=True)
32         row.template_header()
33
34         if context.area.show_menus:
35             sub = row.row(align=True)
36             sub.menu("CONSOLE_MT_console")
37
38         row = layout.row(align=True)
39         row.operator("console.autocomplete", text="Autocomplete")
40
41
42 class CONSOLE_MT_console(Menu):
43     bl_label = "Console"
44
45     def draw(self, context):
46         layout = self.layout
47         layout.column()
48         layout.operator("console.clear")
49         layout.operator("console.copy")
50         layout.operator("console.paste")
51         layout.menu("CONSOLE_MT_language")
52
53         layout.separator()
54
55         layout.operator("screen.area_dupli")
56         layout.operator("screen.screen_full_area")
57
58
59 class CONSOLE_MT_language(Menu):
60     bl_label = "Languages..."
61
62     def draw(self, context):
63         import sys
64
65         layout = self.layout
66         layout.column()
67
68         # Collect modules with 'console_*.execute'
69         languages = []
70         for modname, mod in sys.modules.items():
71             if modname.startswith("console_") and hasattr(mod, "execute"):
72                 languages.append(modname.split('_', 1)[-1])
73
74         languages.sort()
75
76         for language in languages:
77             layout.operator("console.language", text=language[0].upper() + language[1:]).language = language
78
79
80 def add_scrollback(text, text_type):
81     for l in text.split('\n'):
82         bpy.ops.console.scrollback_append(text=l.replace('\t', '    '),
83             type=text_type)
84
85
86 class ConsoleExec(Operator):
87     '''Execute the current console line as a python expression'''
88     bl_idname = "console.execute"
89     bl_label = "Console Execute"
90
91     def execute(self, context):
92         sc = context.space_data
93
94         module = __import__("console_" + sc.language)
95         execute = getattr(module, "execute", None)
96
97         if execute:
98             return execute(context)
99         else:
100             print("Error: bpy.ops.console.execute_" + sc.language + " - not found")
101             return {'FINISHED'}
102
103
104 class ConsoleAutocomplete(Operator):
105     '''Evaluate the namespace up until the cursor and give a list of options or complete the name if there is only one'''
106     bl_idname = "console.autocomplete"
107     bl_label = "Console Autocomplete"
108
109     def execute(self, context):
110         sc = context.space_data
111         module = __import__("console_" + sc.language)
112         autocomplete = getattr(module, "autocomplete", None)
113
114         if autocomplete:
115             return autocomplete(context)
116         else:
117             print("Error: bpy.ops.console.autocomplete_" + sc.language + " - not found")
118             return {'FINISHED'}
119
120
121 class ConsoleBanner(Operator):
122     '''Print a message whem the terminal initializes'''
123     bl_idname = "console.banner"
124     bl_label = "Console Banner"
125
126     def execute(self, context):
127         sc = context.space_data
128
129         # default to python
130         if not sc.language:
131             sc.language = 'python'
132
133         module = __import__("console_" + sc.language)
134         banner = getattr(module, "banner", None)
135
136         if banner:
137             return banner(context)
138         else:
139             print("Error: bpy.ops.console.banner_" + sc.language + " - not found")
140             return {'FINISHED'}
141
142
143 class ConsoleLanguage(Operator):
144     '''Set the current language for this console'''
145     bl_idname = "console.language"
146     bl_label = "Console Language"
147     language = StringProperty(name="Language", maxlen=32, default="")
148
149     def execute(self, context):
150         sc = context.space_data
151
152         # defailt to python
153         sc.language = self.language
154
155         bpy.ops.console.banner()
156
157         # insert a new blank line
158         bpy.ops.console.history_append(text="", current_character=0,
159             remove_duplicates=True)
160
161         return {'FINISHED'}
162
163 if __name__ == "__main__":  # only for live edit.
164     bpy.utils.register_module(__name__)