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