change python scripts so modules which register with blender have a register() functi...
[blender.git] / release / scripts / op / console_shell.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 os
21 import bpy
22
23 language_id = 'shell'
24
25
26 def add_scrollback(text, text_type):
27     for l in text.split('\n'):
28         bpy.ops.console.scrollback_append(text=l.replace('\t', '    '),
29             type=text_type)
30
31
32 def shell_run(text):
33     import subprocess
34     val, output = subprocess.getstatusoutput(text)
35
36     if not val:
37         style = 'OUTPUT'
38     else:
39         style = 'ERROR'
40
41     add_scrollback(output, style)
42
43 PROMPT = '$ '
44
45
46 def execute(context):
47     sc = context.space_data
48
49     try:
50         line = sc.history[-1].line
51     except:
52         return {'CANCELLED'}
53
54     bpy.ops.console.scrollback_append(text=sc.prompt + line, type='INPUT')
55
56     shell_run(line)
57
58     # insert a new blank line
59     bpy.ops.console.history_append(text="", current_character=0,
60         remove_duplicates=True)
61
62     sc.prompt = os.getcwd() + PROMPT
63     return {'FINISHED'}
64
65
66 def autocomplete(context):
67     # sc = context.space_data
68     # TODO
69     return {'CANCELLED'}
70
71
72 def banner(context):
73     sc = context.space_data
74
75     shell_run("bash --version")
76     sc.prompt = os.getcwd() + PROMPT
77
78     return {'FINISHED'}
79
80
81 def register():
82     pass
83
84 def unregister():
85     pass