Python code coverage for Lib/idlelib/PyShell.py

#	count	content
1	n/a	#! /usr/bin/env python3
2	n/a
3	n/a	import getopt
4	n/a	import os
5	n/a	import os.path
6	n/a	import re
7	n/a	import socket
8	n/a	import subprocess
9	n/a	import sys
10	n/a	import threading
11	n/a	import time
12	n/a	import tokenize
13	n/a	import traceback
14	n/a	import types
15	n/a	import io
16	n/a
17	n/a	import linecache
18	n/a	from code import InteractiveInterpreter
19	n/a	from platform import python_version
20	n/a
21	n/a	try:
22	n/a	from tkinter import *
23	n/a	except ImportError:
24	n/a	print("** IDLE can't import Tkinter. " \
25	n/a	"Your Python may not be configured for Tk. **", file=sys.__stderr__)
26	n/a	sys.exit(1)
27	n/a	import tkinter.messagebox as tkMessageBox
28	n/a
29	n/a	from idlelib.EditorWindow import EditorWindow, fixwordbreaks
30	n/a	from idlelib.FileList import FileList
31	n/a	from idlelib.ColorDelegator import ColorDelegator
32	n/a	from idlelib.UndoDelegator import UndoDelegator
33	n/a	from idlelib.OutputWindow import OutputWindow
34	n/a	from idlelib.configHandler import idleConf
35	n/a	from idlelib import idlever
36	n/a	from idlelib import rpc
37	n/a	from idlelib import Debugger
38	n/a	from idlelib import RemoteDebugger
39	n/a	from idlelib import macosxSupport
40	n/a
41	n/a	HOST = '127.0.0.1' # python execution server on localhost loopback
42	n/a	PORT = 0 # someday pass in host, port for remote debug capability
43	n/a
44	n/a	# Override warnings module to write to warning_stream. Initialize to send IDLE
45	n/a	# internal warnings to the console. ScriptBinding.check_syntax() will
46	n/a	# temporarily redirect the stream to the shell window to display warnings when
47	n/a	# checking user's code.
48	n/a	warning_stream = sys.__stderr__ # None, at least on Windows, if no console.
49	n/a	import warnings
50	n/a
51	n/a	def idle_formatwarning(message, category, filename, lineno, line=None):
52	n/a	"""Format warnings the IDLE way."""
53	n/a
54	n/a	s = "\nWarning (from warnings module):\n"
55	n/a	s += ' File \"%s\", line %s\n' % (filename, lineno)
56	n/a	if line is None:
57	n/a	line = linecache.getline(filename, lineno)
58	n/a	line = line.strip()
59	n/a	if line:
60	n/a	s += " %s\n" % line
61	n/a	s += "%s: %s\n" % (category.__name__, message)
62	n/a	return s
63	n/a
64	n/a	def idle_showwarning(
65	n/a	message, category, filename, lineno, file=None, line=None):
66	n/a	"""Show Idle-format warning (after replacing warnings.showwarning).
67	n/a
68	n/a	The differences are the formatter called, the file=None replacement,
69	n/a	which can be None, the capture of the consequence AttributeError,
70	n/a	and the output of a hard-coded prompt.
71	n/a	"""
72	n/a	if file is None:
73	n/a	file = warning_stream
74	n/a	try:
75	n/a	file.write(idle_formatwarning(
76	n/a	message, category, filename, lineno, line=line))
77	n/a	file.write(">>> ")
78	n/a	except (AttributeError, OSError):
79	n/a	pass # if file (probably __stderr__) is invalid, skip warning.
80	n/a
81	n/a	_warnings_showwarning = None
82	n/a
83	n/a	def capture_warnings(capture):
84	n/a	"Replace warning.showwarning with idle_showwarning, or reverse."
85	n/a
86	n/a	global _warnings_showwarning
87	n/a	if capture:
88	n/a	if _warnings_showwarning is None:
89	n/a	_warnings_showwarning = warnings.showwarning
90	n/a	warnings.showwarning = idle_showwarning
91	n/a	else:
92	n/a	if _warnings_showwarning is not None:
93	n/a	warnings.showwarning = _warnings_showwarning
94	n/a	_warnings_showwarning = None
95	n/a
96	n/a	capture_warnings(True)
97	n/a
98	n/a	def extended_linecache_checkcache(filename=None,
99	n/a	orig_checkcache=linecache.checkcache):
100	n/a	"""Extend linecache.checkcache to preserve the <pyshell#...> entries
101	n/a
102	n/a	Rather than repeating the linecache code, patch it to save the
103	n/a	<pyshell#...> entries, call the original linecache.checkcache()
104	n/a	(skipping them), and then restore the saved entries.
105	n/a
106	n/a	orig_checkcache is bound at definition time to the original
107	n/a	method, allowing it to be patched.
108	n/a	"""
109	n/a	cache = linecache.cache
110	n/a	save = {}
111	n/a	for key in list(cache):
112	n/a	if key[:1] + key[-1:] == '<>':
113	n/a	save[key] = cache.pop(key)
114	n/a	orig_checkcache(filename)
115	n/a	cache.update(save)
116	n/a
117	n/a	# Patch linecache.checkcache():
118	n/a	linecache.checkcache = extended_linecache_checkcache
119	n/a
120	n/a
121	n/a	class PyShellEditorWindow(EditorWindow):
122	n/a	"Regular text edit window in IDLE, supports breakpoints"
123	n/a
124	n/a	def __init__(self, *args):
125	n/a	self.breakpoints = []
126	n/a	EditorWindow.__init__(self, *args)
127	n/a	self.text.bind("<<set-breakpoint-here>>", self.set_breakpoint_here)
128	n/a	self.text.bind("<<clear-breakpoint-here>>", self.clear_breakpoint_here)
129	n/a	self.text.bind("<<open-python-shell>>", self.flist.open_shell)
130	n/a
131	n/a	self.breakpointPath = os.path.join(idleConf.GetUserCfgDir(),
132	n/a	'breakpoints.lst')
133	n/a	# whenever a file is changed, restore breakpoints
134	n/a	def filename_changed_hook(old_hook=self.io.filename_change_hook,
135	n/a	self=self):
136	n/a	self.restore_file_breaks()
137	n/a	old_hook()
138	n/a	self.io.set_filename_change_hook(filename_changed_hook)
139	n/a	if self.io.filename:
140	n/a	self.restore_file_breaks()
141	n/a
142	n/a	rmenu_specs = [
143	n/a	("Cut", "<<cut>>", "rmenu_check_cut"),
144	n/a	("Copy", "<<copy>>", "rmenu_check_copy"),
145	n/a	("Paste", "<<paste>>", "rmenu_check_paste"),
146	n/a	(None, None, None),
147	n/a	("Set Breakpoint", "<<set-breakpoint-here>>", None),
148	n/a	("Clear Breakpoint", "<<clear-breakpoint-here>>", None)
149	n/a	]
150	n/a
151	n/a	def set_breakpoint(self, lineno):
152	n/a	text = self.text
153	n/a	filename = self.io.filename
154	n/a	text.tag_add("BREAK", "%d.0" % lineno, "%d.0" % (lineno+1))
155	n/a	try:
156	n/a	i = self.breakpoints.index(lineno)
157	n/a	except ValueError: # only add if missing, i.e. do once
158	n/a	self.breakpoints.append(lineno)
159	n/a	try: # update the subprocess debugger
160	n/a	debug = self.flist.pyshell.interp.debugger
161	n/a	debug.set_breakpoint_here(filename, lineno)
162	n/a	except: # but debugger may not be active right now....
163	n/a	pass
164	n/a
165	n/a	def set_breakpoint_here(self, event=None):
166	n/a	text = self.text
167	n/a	filename = self.io.filename
168	n/a	if not filename:
169	n/a	text.bell()
170	n/a	return
171	n/a	lineno = int(float(text.index("insert")))
172	n/a	self.set_breakpoint(lineno)
173	n/a
174	n/a	def clear_breakpoint_here(self, event=None):
175	n/a	text = self.text
176	n/a	filename = self.io.filename
177	n/a	if not filename:
178	n/a	text.bell()
179	n/a	return
180	n/a	lineno = int(float(text.index("insert")))
181	n/a	try:
182	n/a	self.breakpoints.remove(lineno)
183	n/a	except:
184	n/a	pass
185	n/a	text.tag_remove("BREAK", "insert linestart",\
186	n/a	"insert lineend +1char")
187	n/a	try:
188	n/a	debug = self.flist.pyshell.interp.debugger
189	n/a	debug.clear_breakpoint_here(filename, lineno)
190	n/a	except:
191	n/a	pass
192	n/a
193	n/a	def clear_file_breaks(self):
194	n/a	if self.breakpoints:
195	n/a	text = self.text
196	n/a	filename = self.io.filename
197	n/a	if not filename:
198	n/a	text.bell()
199	n/a	return
200	n/a	self.breakpoints = []
201	n/a	text.tag_remove("BREAK", "1.0", END)
202	n/a	try:
203	n/a	debug = self.flist.pyshell.interp.debugger
204	n/a	debug.clear_file_breaks(filename)
205	n/a	except:
206	n/a	pass
207	n/a
208	n/a	def store_file_breaks(self):
209	n/a	"Save breakpoints when file is saved"
210	n/a	# XXX 13 Dec 2002 KBK Currently the file must be saved before it can
211	n/a	# be run. The breaks are saved at that time. If we introduce
212	n/a	# a temporary file save feature the save breaks functionality
213	n/a	# needs to be re-verified, since the breaks at the time the
214	n/a	# temp file is created may differ from the breaks at the last
215	n/a	# permanent save of the file. Currently, a break introduced
216	n/a	# after a save will be effective, but not persistent.
217	n/a	# This is necessary to keep the saved breaks synched with the
218	n/a	# saved file.
219	n/a	#
220	n/a	# Breakpoints are set as tagged ranges in the text. Certain
221	n/a	# kinds of edits cause these ranges to be deleted: Inserting
222	n/a	# or deleting a line just before a breakpoint, and certain
223	n/a	# deletions prior to a breakpoint. These issues need to be
224	n/a	# investigated and understood. It's not clear if they are
225	n/a	# Tk issues or IDLE issues, or whether they can actually
226	n/a	# be fixed. Since a modified file has to be saved before it is
227	n/a	# run, and since self.breakpoints (from which the subprocess
228	n/a	# debugger is loaded) is updated during the save, the visible
229	n/a	# breaks stay synched with the subprocess even if one of these
230	n/a	# unexpected breakpoint deletions occurs.
231	n/a	breaks = self.breakpoints
232	n/a	filename = self.io.filename
233	n/a	try:
234	n/a	with open(self.breakpointPath, "r") as fp:
235	n/a	lines = fp.readlines()
236	n/a	except OSError:
237	n/a	lines = []
238	n/a	try:
239	n/a	with open(self.breakpointPath, "w") as new_file:
240	n/a	for line in lines:
241	n/a	if not line.startswith(filename + '='):
242	n/a	new_file.write(line)
243	n/a	self.update_breakpoints()
244	n/a	breaks = self.breakpoints
245	n/a	if breaks:
246	n/a	new_file.write(filename + '=' + str(breaks) + '\n')
247	n/a	except OSError as err:
248	n/a	if not getattr(self.root, "breakpoint_error_displayed", False):
249	n/a	self.root.breakpoint_error_displayed = True
250	n/a	tkMessageBox.showerror(title='IDLE Error',
251	n/a	message='Unable to update breakpoint list:\n%s'
252	n/a	% str(err),
253	n/a	parent=self.text)
254	n/a
255	n/a	def restore_file_breaks(self):
256	n/a	self.text.update() # this enables setting "BREAK" tags to be visible
257	n/a	if self.io is None:
258	n/a	# can happen if IDLE closes due to the .update() call
259	n/a	return
260	n/a	filename = self.io.filename
261	n/a	if filename is None:
262	n/a	return
263	n/a	if os.path.isfile(self.breakpointPath):
264	n/a	with open(self.breakpointPath, "r") as fp:
265	n/a	lines = fp.readlines()
266	n/a	for line in lines:
267	n/a	if line.startswith(filename + '='):
268	n/a	breakpoint_linenumbers = eval(line[len(filename)+1:])
269	n/a	for breakpoint_linenumber in breakpoint_linenumbers:
270	n/a	self.set_breakpoint(breakpoint_linenumber)
271	n/a
272	n/a	def update_breakpoints(self):
273	n/a	"Retrieves all the breakpoints in the current window"
274	n/a	text = self.text
275	n/a	ranges = text.tag_ranges("BREAK")
276	n/a	linenumber_list = self.ranges_to_linenumbers(ranges)
277	n/a	self.breakpoints = linenumber_list
278	n/a
279	n/a	def ranges_to_linenumbers(self, ranges):
280	n/a	lines = []
281	n/a	for index in range(0, len(ranges), 2):
282	n/a	lineno = int(float(ranges[index].string))
283	n/a	end = int(float(ranges[index+1].string))
284	n/a	while lineno < end:
285	n/a	lines.append(lineno)
286	n/a	lineno += 1
287	n/a	return lines
288	n/a
289	n/a	# XXX 13 Dec 2002 KBK Not used currently
290	n/a	# def saved_change_hook(self):
291	n/a	# "Extend base method - clear breaks if module is modified"
292	n/a	# if not self.get_saved():
293	n/a	# self.clear_file_breaks()
294	n/a	# EditorWindow.saved_change_hook(self)
295	n/a
296	n/a	def _close(self):
297	n/a	"Extend base method - clear breaks when module is closed"
298	n/a	self.clear_file_breaks()
299	n/a	EditorWindow._close(self)
300	n/a
301	n/a
302	n/a	class PyShellFileList(FileList):
303	n/a	"Extend base class: IDLE supports a shell and breakpoints"
304	n/a
305	n/a	# override FileList's class variable, instances return PyShellEditorWindow
306	n/a	# instead of EditorWindow when new edit windows are created.
307	n/a	EditorWindow = PyShellEditorWindow
308	n/a
309	n/a	pyshell = None
310	n/a
311	n/a	def open_shell(self, event=None):
312	n/a	if self.pyshell:
313	n/a	self.pyshell.top.wakeup()
314	n/a	else:
315	n/a	self.pyshell = PyShell(self)
316	n/a	if self.pyshell:
317	n/a	if not self.pyshell.begin():
318	n/a	return None
319	n/a	return self.pyshell
320	n/a
321	n/a
322	n/a	class ModifiedColorDelegator(ColorDelegator):
323	n/a	"Extend base class: colorizer for the shell window itself"
324	n/a
325	n/a	def __init__(self):
326	n/a	ColorDelegator.__init__(self)
327	n/a	self.LoadTagDefs()
328	n/a
329	n/a	def recolorize_main(self):
330	n/a	self.tag_remove("TODO", "1.0", "iomark")
331	n/a	self.tag_add("SYNC", "1.0", "iomark")
332	n/a	ColorDelegator.recolorize_main(self)
333	n/a
334	n/a	def LoadTagDefs(self):
335	n/a	ColorDelegator.LoadTagDefs(self)
336	n/a	theme = idleConf.GetOption('main','Theme','name')
337	n/a	self.tagdefs.update({
338	n/a	"stdin": {'background':None,'foreground':None},
339	n/a	"stdout": idleConf.GetHighlight(theme, "stdout"),
340	n/a	"stderr": idleConf.GetHighlight(theme, "stderr"),
341	n/a	"console": idleConf.GetHighlight(theme, "console"),
342	n/a	})
343	n/a
344	n/a	def removecolors(self):
345	n/a	# Don't remove shell color tags before "iomark"
346	n/a	for tag in self.tagdefs:
347	n/a	self.tag_remove(tag, "iomark", "end")
348	n/a
349	n/a	class ModifiedUndoDelegator(UndoDelegator):
350	n/a	"Extend base class: forbid insert/delete before the I/O mark"
351	n/a
352	n/a	def insert(self, index, chars, tags=None):
353	n/a	try:
354	n/a	if self.delegate.compare(index, "<", "iomark"):
355	n/a	self.delegate.bell()
356	n/a	return
357	n/a	except TclError:
358	n/a	pass
359	n/a	UndoDelegator.insert(self, index, chars, tags)
360	n/a
361	n/a	def delete(self, index1, index2=None):
362	n/a	try:
363	n/a	if self.delegate.compare(index1, "<", "iomark"):
364	n/a	self.delegate.bell()
365	n/a	return
366	n/a	except TclError:
367	n/a	pass
368	n/a	UndoDelegator.delete(self, index1, index2)
369	n/a
370	n/a
371	n/a	class MyRPCClient(rpc.RPCClient):
372	n/a
373	n/a	def handle_EOF(self):
374	n/a	"Override the base class - just re-raise EOFError"
375	n/a	raise EOFError
376	n/a
377	n/a
378	n/a	class ModifiedInterpreter(InteractiveInterpreter):
379	n/a
380	n/a	def __init__(self, tkconsole):
381	n/a	self.tkconsole = tkconsole
382	n/a	locals = sys.modules['__main__'].__dict__
383	n/a	InteractiveInterpreter.__init__(self, locals=locals)
384	n/a	self.save_warnings_filters = None
385	n/a	self.restarting = False
386	n/a	self.subprocess_arglist = None
387	n/a	self.port = PORT
388	n/a	self.original_compiler_flags = self.compile.compiler.flags
389	n/a
390	n/a	_afterid = None
391	n/a	rpcclt = None
392	n/a	rpcsubproc = None
393	n/a
394	n/a	def spawn_subprocess(self):
395	n/a	if self.subprocess_arglist is None:
396	n/a	self.subprocess_arglist = self.build_subprocess_arglist()
397	n/a	self.rpcsubproc = subprocess.Popen(self.subprocess_arglist)
398	n/a
399	n/a	def build_subprocess_arglist(self):
400	n/a	assert (self.port!=0), (
401	n/a	"Socket should have been assigned a port number.")
402	n/a	w = ['-W' + s for s in sys.warnoptions]
403	n/a	# Maybe IDLE is installed and is being accessed via sys.path,
404	n/a	# or maybe it's not installed and the idle.py script is being
405	n/a	# run from the IDLE source directory.
406	n/a	del_exitf = idleConf.GetOption('main', 'General', 'delete-exitfunc',
407	n/a	default=False, type='bool')
408	n/a	if __name__ == 'idlelib.PyShell':
409	n/a	command = "__import__('idlelib.run').run.main(%r)" % (del_exitf,)
410	n/a	else:
411	n/a	command = "__import__('run').main(%r)" % (del_exitf,)
412	n/a	return [sys.executable] + w + ["-c", command, str(self.port)]
413	n/a
414	n/a	def start_subprocess(self):
415	n/a	addr = (HOST, self.port)
416	n/a	# GUI makes several attempts to acquire socket, listens for connection
417	n/a	for i in range(3):
418	n/a	time.sleep(i)
419	n/a	try:
420	n/a	self.rpcclt = MyRPCClient(addr)
421	n/a	break
422	n/a	except OSError as err:
423	n/a	pass
424	n/a	else:
425	n/a	self.display_port_binding_error()
426	n/a	return None
427	n/a	# if PORT was 0, system will assign an 'ephemeral' port. Find it out:
428	n/a	self.port = self.rpcclt.listening_sock.getsockname()[1]
429	n/a	# if PORT was not 0, probably working with a remote execution server
430	n/a	if PORT != 0:
431	n/a	# To allow reconnection within the 2MSL wait (cf. Stevens TCP
432	n/a	# V1, 18.6), set SO_REUSEADDR. Note that this can be problematic
433	n/a	# on Windows since the implementation allows two active sockets on
434	n/a	# the same address!
435	n/a	self.rpcclt.listening_sock.setsockopt(socket.SOL_SOCKET,
436	n/a	socket.SO_REUSEADDR, 1)
437	n/a	self.spawn_subprocess()
438	n/a	#time.sleep(20) # test to simulate GUI not accepting connection
439	n/a	# Accept the connection from the Python execution server
440	n/a	self.rpcclt.listening_sock.settimeout(10)
441	n/a	try:
442	n/a	self.rpcclt.accept()
443	n/a	except socket.timeout as err:
444	n/a	self.display_no_subprocess_error()
445	n/a	return None
446	n/a	self.rpcclt.register("console", self.tkconsole)
447	n/a	self.rpcclt.register("stdin", self.tkconsole.stdin)
448	n/a	self.rpcclt.register("stdout", self.tkconsole.stdout)
449	n/a	self.rpcclt.register("stderr", self.tkconsole.stderr)
450	n/a	self.rpcclt.register("flist", self.tkconsole.flist)
451	n/a	self.rpcclt.register("linecache", linecache)
452	n/a	self.rpcclt.register("interp", self)
453	n/a	self.transfer_path(with_cwd=True)
454	n/a	self.poll_subprocess()
455	n/a	return self.rpcclt
456	n/a
457	n/a	def restart_subprocess(self, with_cwd=False):
458	n/a	if self.restarting:
459	n/a	return self.rpcclt
460	n/a	self.restarting = True
461	n/a	# close only the subprocess debugger
462	n/a	debug = self.getdebugger()
463	n/a	if debug:
464	n/a	try:
465	n/a	# Only close subprocess debugger, don't unregister gui_adap!
466	n/a	RemoteDebugger.close_subprocess_debugger(self.rpcclt)
467	n/a	except:
468	n/a	pass
469	n/a	# Kill subprocess, spawn a new one, accept connection.
470	n/a	self.rpcclt.close()
471	n/a	self.terminate_subprocess()
472	n/a	console = self.tkconsole
473	n/a	was_executing = console.executing
474	n/a	console.executing = False
475	n/a	self.spawn_subprocess()
476	n/a	try:
477	n/a	self.rpcclt.accept()
478	n/a	except socket.timeout as err:
479	n/a	self.display_no_subprocess_error()
480	n/a	return None
481	n/a	self.transfer_path(with_cwd=with_cwd)
482	n/a	console.stop_readline()
483	n/a	# annotate restart in shell window and mark it
484	n/a	console.text.delete("iomark", "end-1c")
485	n/a	if was_executing:
486	n/a	console.write('\n')
487	n/a	console.showprompt()
488	n/a	halfbar = ((int(console.width) - 16) // 2) * '='
489	n/a	console.write(halfbar + ' RESTART ' + halfbar)
490	n/a	console.text.mark_set("restart", "end-1c")
491	n/a	console.text.mark_gravity("restart", "left")
492	n/a	console.showprompt()
493	n/a	# restart subprocess debugger
494	n/a	if debug:
495	n/a	# Restarted debugger connects to current instance of debug GUI
496	n/a	gui = RemoteDebugger.restart_subprocess_debugger(self.rpcclt)
497	n/a	# reload remote debugger breakpoints for all PyShellEditWindows
498	n/a	debug.load_breakpoints()
499	n/a	self.compile.compiler.flags = self.original_compiler_flags
500	n/a	self.restarting = False
501	n/a	return self.rpcclt
502	n/a
503	n/a	def __request_interrupt(self):
504	n/a	self.rpcclt.remotecall("exec", "interrupt_the_server", (), {})
505	n/a
506	n/a	def interrupt_subprocess(self):
507	n/a	threading.Thread(target=self.__request_interrupt).start()
508	n/a
509	n/a	def kill_subprocess(self):
510	n/a	if self._afterid is not None:
511	n/a	self.tkconsole.text.after_cancel(self._afterid)
512	n/a	try:
513	n/a	self.rpcclt.listening_sock.close()
514	n/a	except AttributeError: # no socket
515	n/a	pass
516	n/a	try:
517	n/a	self.rpcclt.close()
518	n/a	except AttributeError: # no socket
519	n/a	pass
520	n/a	self.terminate_subprocess()
521	n/a	self.tkconsole.executing = False
522	n/a	self.rpcclt = None
523	n/a
524	n/a	def terminate_subprocess(self):
525	n/a	"Make sure subprocess is terminated"
526	n/a	try:
527	n/a	self.rpcsubproc.kill()
528	n/a	except OSError:
529	n/a	# process already terminated
530	n/a	return
531	n/a	else:
532	n/a	try:
533	n/a	self.rpcsubproc.wait()
534	n/a	except OSError:
535	n/a	return
536	n/a
537	n/a	def transfer_path(self, with_cwd=False):
538	n/a	if with_cwd: # Issue 13506
539	n/a	path = [''] # include Current Working Directory
540	n/a	path.extend(sys.path)
541	n/a	else:
542	n/a	path = sys.path
543	n/a
544	n/a	self.runcommand("""if 1:
545	n/a	import sys as _sys
546	n/a	_sys.path = %r
547	n/a	del _sys
548	n/a	\n""" % (path,))
549	n/a
550	n/a	active_seq = None
551	n/a
552	n/a	def poll_subprocess(self):
553	n/a	clt = self.rpcclt
554	n/a	if clt is None:
555	n/a	return
556	n/a	try:
557	n/a	response = clt.pollresponse(self.active_seq, wait=0.05)
558	n/a	except (EOFError, OSError, KeyboardInterrupt):
559	n/a	# lost connection or subprocess terminated itself, restart
560	n/a	# [the KBI is from rpc.SocketIO.handle_EOF()]
561	n/a	if self.tkconsole.closing:
562	n/a	return
563	n/a	response = None
564	n/a	self.restart_subprocess()
565	n/a	if response:
566	n/a	self.tkconsole.resetoutput()
567	n/a	self.active_seq = None
568	n/a	how, what = response
569	n/a	console = self.tkconsole.console
570	n/a	if how == "OK":
571	n/a	if what is not None:
572	n/a	print(repr(what), file=console)
573	n/a	elif how == "EXCEPTION":
574	n/a	if self.tkconsole.getvar("<<toggle-jit-stack-viewer>>"):
575	n/a	self.remote_stack_viewer()
576	n/a	elif how == "ERROR":
577	n/a	errmsg = "PyShell.ModifiedInterpreter: Subprocess ERROR:\n"
578	n/a	print(errmsg, what, file=sys.__stderr__)
579	n/a	print(errmsg, what, file=console)
580	n/a	# we received a response to the currently active seq number:
581	n/a	try:
582	n/a	self.tkconsole.endexecuting()
583	n/a	except AttributeError: # shell may have closed
584	n/a	pass
585	n/a	# Reschedule myself
586	n/a	if not self.tkconsole.closing:
587	n/a	self._afterid = self.tkconsole.text.after(
588	n/a	self.tkconsole.pollinterval, self.poll_subprocess)
589	n/a
590	n/a	debugger = None
591	n/a
592	n/a	def setdebugger(self, debugger):
593	n/a	self.debugger = debugger
594	n/a
595	n/a	def getdebugger(self):
596	n/a	return self.debugger
597	n/a
598	n/a	def open_remote_stack_viewer(self):
599	n/a	"""Initiate the remote stack viewer from a separate thread.
600	n/a
601	n/a	This method is called from the subprocess, and by returning from this
602	n/a	method we allow the subprocess to unblock. After a bit the shell
603	n/a	requests the subprocess to open the remote stack viewer which returns a
604	n/a	static object looking at the last exception. It is queried through
605	n/a	the RPC mechanism.
606	n/a
607	n/a	"""
608	n/a	self.tkconsole.text.after(300, self.remote_stack_viewer)
609	n/a	return
610	n/a
611	n/a	def remote_stack_viewer(self):
612	n/a	from idlelib import RemoteObjectBrowser
613	n/a	oid = self.rpcclt.remotequeue("exec", "stackviewer", ("flist",), {})
614	n/a	if oid is None:
615	n/a	self.tkconsole.root.bell()
616	n/a	return
617	n/a	item = RemoteObjectBrowser.StubObjectTreeItem(self.rpcclt, oid)
618	n/a	from idlelib.TreeWidget import ScrolledCanvas, TreeNode
619	n/a	top = Toplevel(self.tkconsole.root)
620	n/a	theme = idleConf.GetOption('main','Theme','name')
621	n/a	background = idleConf.GetHighlight(theme, 'normal')['background']
622	n/a	sc = ScrolledCanvas(top, bg=background, highlightthickness=0)
623	n/a	sc.frame.pack(expand=1, fill="both")
624	n/a	node = TreeNode(sc.canvas, None, item)
625	n/a	node.expand()
626	n/a	# XXX Should GC the remote tree when closing the window
627	n/a
628	n/a	gid = 0
629	n/a
630	n/a	def execsource(self, source):
631	n/a	"Like runsource() but assumes complete exec source"
632	n/a	filename = self.stuffsource(source)
633	n/a	self.execfile(filename, source)
634	n/a
635	n/a	def execfile(self, filename, source=None):
636	n/a	"Execute an existing file"
637	n/a	if source is None:
638	n/a	with tokenize.open(filename) as fp:
639	n/a	source = fp.read()
640	n/a	try:
641	n/a	code = compile(source, filename, "exec")
642	n/a	except (OverflowError, SyntaxError):
643	n/a	self.tkconsole.resetoutput()
644	n/a	tkerr = self.tkconsole.stderr
645	n/a	print('*** Error in script or command!\n', file=tkerr)
646	n/a	print('Traceback (most recent call last):', file=tkerr)
647	n/a	InteractiveInterpreter.showsyntaxerror(self, filename)
648	n/a	self.tkconsole.showprompt()
649	n/a	else:
650	n/a	self.runcode(code)
651	n/a
652	n/a	def runsource(self, source):
653	n/a	"Extend base class method: Stuff the source in the line cache first"
654	n/a	filename = self.stuffsource(source)
655	n/a	self.more = 0
656	n/a	self.save_warnings_filters = warnings.filters[:]
657	n/a	warnings.filterwarnings(action="error", category=SyntaxWarning)
658	n/a	# at the moment, InteractiveInterpreter expects str
659	n/a	assert isinstance(source, str)
660	n/a	#if isinstance(source, str):
661	n/a	# from idlelib import IOBinding
662	n/a	# try:
663	n/a	# source = source.encode(IOBinding.encoding)
664	n/a	# except UnicodeError:
665	n/a	# self.tkconsole.resetoutput()
666	n/a	# self.write("Unsupported characters in input\n")
667	n/a	# return
668	n/a	try:
669	n/a	# InteractiveInterpreter.runsource() calls its runcode() method,
670	n/a	# which is overridden (see below)
671	n/a	return InteractiveInterpreter.runsource(self, source, filename)
672	n/a	finally:
673	n/a	if self.save_warnings_filters is not None:
674	n/a	warnings.filters[:] = self.save_warnings_filters
675	n/a	self.save_warnings_filters = None
676	n/a
677	n/a	def stuffsource(self, source):
678	n/a	"Stuff source in the filename cache"
679	n/a	filename = "<pyshell#%d>" % self.gid
680	n/a	self.gid = self.gid + 1
681	n/a	lines = source.split("\n")
682	n/a	linecache.cache[filename] = len(source)+1, 0, lines, filename
683	n/a	return filename
684	n/a
685	n/a	def prepend_syspath(self, filename):
686	n/a	"Prepend sys.path with file's directory if not already included"
687	n/a	self.runcommand("""if 1:
688	n/a	_filename = %r
689	n/a	import sys as _sys
690	n/a	from os.path import dirname as _dirname
691	n/a	_dir = _dirname(_filename)
692	n/a	if not _dir in _sys.path:
693	n/a	_sys.path.insert(0, _dir)
694	n/a	del _filename, _sys, _dirname, _dir
695	n/a	\n""" % (filename,))
696	n/a
697	n/a	def showsyntaxerror(self, filename=None):
698	n/a	"""Override Interactive Interpreter method: Use Colorizing
699	n/a
700	n/a	Color the offending position instead of printing it and pointing at it
701	n/a	with a caret.
702	n/a
703	n/a	"""
704	n/a	tkconsole = self.tkconsole
705	n/a	text = tkconsole.text
706	n/a	text.tag_remove("ERROR", "1.0", "end")
707	n/a	type, value, tb = sys.exc_info()
708	n/a	msg = getattr(value, 'msg', '') or value or "<no detail available>"
709	n/a	lineno = getattr(value, 'lineno', '') or 1
710	n/a	offset = getattr(value, 'offset', '') or 0
711	n/a	if offset == 0:
712	n/a	lineno += 1 #mark end of offending line
713	n/a	if lineno == 1:
714	n/a	pos = "iomark + %d chars" % (offset-1)
715	n/a	else:
716	n/a	pos = "iomark linestart + %d lines + %d chars" % \
717	n/a	(lineno-1, offset-1)
718	n/a	tkconsole.colorize_syntax_error(text, pos)
719	n/a	tkconsole.resetoutput()
720	n/a	self.write("SyntaxError: %s\n" % msg)
721	n/a	tkconsole.showprompt()
722	n/a
723	n/a	def showtraceback(self):
724	n/a	"Extend base class method to reset output properly"
725	n/a	self.tkconsole.resetoutput()
726	n/a	self.checklinecache()
727	n/a	InteractiveInterpreter.showtraceback(self)
728	n/a	if self.tkconsole.getvar("<<toggle-jit-stack-viewer>>"):
729	n/a	self.tkconsole.open_stack_viewer()
730	n/a
731	n/a	def checklinecache(self):
732	n/a	c = linecache.cache
733	n/a	for key in list(c.keys()):
734	n/a	if key[:1] + key[-1:] != "<>":
735	n/a	del c[key]
736	n/a
737	n/a	def runcommand(self, code):
738	n/a	"Run the code without invoking the debugger"
739	n/a	# The code better not raise an exception!
740	n/a	if self.tkconsole.executing:
741	n/a	self.display_executing_dialog()
742	n/a	return 0
743	n/a	if self.rpcclt:
744	n/a	self.rpcclt.remotequeue("exec", "runcode", (code,), {})
745	n/a	else:
746	n/a	exec(code, self.locals)
747	n/a	return 1
748	n/a
749	n/a	def runcode(self, code):
750	n/a	"Override base class method"
751	n/a	if self.tkconsole.executing:
752	n/a	self.interp.restart_subprocess()
753	n/a	self.checklinecache()
754	n/a	if self.save_warnings_filters is not None:
755	n/a	warnings.filters[:] = self.save_warnings_filters
756	n/a	self.save_warnings_filters = None
757	n/a	debugger = self.debugger
758	n/a	try:
759	n/a	self.tkconsole.beginexecuting()
760	n/a	if not debugger and self.rpcclt is not None:
761	n/a	self.active_seq = self.rpcclt.asyncqueue("exec", "runcode",
762	n/a	(code,), {})
763	n/a	elif debugger:
764	n/a	debugger.run(code, self.locals)
765	n/a	else:
766	n/a	exec(code, self.locals)
767	n/a	except SystemExit:
768	n/a	if not self.tkconsole.closing:
769	n/a	if tkMessageBox.askyesno(
770	n/a	"Exit?",
771	n/a	"Do you want to exit altogether?",
772	n/a	default="yes",
773	n/a	master=self.tkconsole.text):
774	n/a	raise
775	n/a	else:
776	n/a	self.showtraceback()
777	n/a	else:
778	n/a	raise
779	n/a	except:
780	n/a	if use_subprocess:
781	n/a	print("IDLE internal error in runcode()",
782	n/a	file=self.tkconsole.stderr)
783	n/a	self.showtraceback()
784	n/a	self.tkconsole.endexecuting()
785	n/a	else:
786	n/a	if self.tkconsole.canceled:
787	n/a	self.tkconsole.canceled = False
788	n/a	print("KeyboardInterrupt", file=self.tkconsole.stderr)
789	n/a	else:
790	n/a	self.showtraceback()
791	n/a	finally:
792	n/a	if not use_subprocess:
793	n/a	try:
794	n/a	self.tkconsole.endexecuting()
795	n/a	except AttributeError: # shell may have closed
796	n/a	pass
797	n/a
798	n/a	def write(self, s):
799	n/a	"Override base class method"
800	n/a	return self.tkconsole.stderr.write(s)
801	n/a
802	n/a	def display_port_binding_error(self):
803	n/a	tkMessageBox.showerror(
804	n/a	"Port Binding Error",
805	n/a	"IDLE can't bind to a TCP/IP port, which is necessary to "
806	n/a	"communicate with its Python execution server. This might be "
807	n/a	"because no networking is installed on this computer. "
808	n/a	"Run IDLE with the -n command line switch to start without a "
809	n/a	"subprocess and refer to Help/IDLE Help 'Running without a "
810	n/a	"subprocess' for further details.",
811	n/a	master=self.tkconsole.text)
812	n/a
813	n/a	def display_no_subprocess_error(self):
814	n/a	tkMessageBox.showerror(
815	n/a	"Subprocess Startup Error",
816	n/a	"IDLE's subprocess didn't make connection. Either IDLE can't "
817	n/a	"start a subprocess or personal firewall software is blocking "
818	n/a	"the connection.",
819	n/a	master=self.tkconsole.text)
820	n/a
821	n/a	def display_executing_dialog(self):
822	n/a	tkMessageBox.showerror(
823	n/a	"Already executing",
824	n/a	"The Python Shell window is already executing a command; "
825	n/a	"please wait until it is finished.",
826	n/a	master=self.tkconsole.text)
827	n/a
828	n/a
829	n/a	class PyShell(OutputWindow):
830	n/a
831	n/a	shell_title = "Python " + python_version() + " Shell"
832	n/a
833	n/a	# Override classes
834	n/a	ColorDelegator = ModifiedColorDelegator
835	n/a	UndoDelegator = ModifiedUndoDelegator
836	n/a
837	n/a	# Override menus
838	n/a	menu_specs = [
839	n/a	("file", "_File"),
840	n/a	("edit", "_Edit"),
841	n/a	("debug", "_Debug"),
842	n/a	("options", "_Options"),
843	n/a	("windows", "_Windows"),
844	n/a	("help", "_Help"),
845	n/a	]
846	n/a
847	n/a	if macosxSupport.runningAsOSXApp():
848	n/a	menu_specs[-2] = ("windows", "_Window")
849	n/a
850	n/a
851	n/a	# New classes
852	n/a	from idlelib.IdleHistory import History
853	n/a
854	n/a	def __init__(self, flist=None):
855	n/a	if use_subprocess:
856	n/a	ms = self.menu_specs
857	n/a	if ms[2][0] != "shell":
858	n/a	ms.insert(2, ("shell", "She_ll"))
859	n/a	self.interp = ModifiedInterpreter(self)
860	n/a	if flist is None:
861	n/a	root = Tk()
862	n/a	fixwordbreaks(root)
863	n/a	root.withdraw()
864	n/a	flist = PyShellFileList(root)
865	n/a	#
866	n/a	OutputWindow.__init__(self, flist, None, None)
867	n/a	#
868	n/a	## self.config(usetabs=1, indentwidth=8, context_use_ps1=1)
869	n/a	self.usetabs = True
870	n/a	# indentwidth must be 8 when using tabs. See note in EditorWindow:
871	n/a	self.indentwidth = 8
872	n/a	self.context_use_ps1 = True
873	n/a	#
874	n/a	text = self.text
875	n/a	text.configure(wrap="char")
876	n/a	text.bind("<<newline-and-indent>>", self.enter_callback)
877	n/a	text.bind("<<plain-newline-and-indent>>", self.linefeed_callback)
878	n/a	text.bind("<<interrupt-execution>>", self.cancel_callback)
879	n/a	text.bind("<<end-of-file>>", self.eof_callback)
880	n/a	text.bind("<<open-stack-viewer>>", self.open_stack_viewer)
881	n/a	text.bind("<<toggle-debugger>>", self.toggle_debugger)
882	n/a	text.bind("<<toggle-jit-stack-viewer>>", self.toggle_jit_stack_viewer)
883	n/a	if use_subprocess:
884	n/a	text.bind("<<view-restart>>", self.view_restart_mark)
885	n/a	text.bind("<<restart-shell>>", self.restart_shell)
886	n/a	#
887	n/a	self.save_stdout = sys.stdout
888	n/a	self.save_stderr = sys.stderr
889	n/a	self.save_stdin = sys.stdin
890	n/a	from idlelib import IOBinding
891	n/a	self.stdin = PseudoInputFile(self, "stdin", IOBinding.encoding)
892	n/a	self.stdout = PseudoOutputFile(self, "stdout", IOBinding.encoding)
893	n/a	self.stderr = PseudoOutputFile(self, "stderr", IOBinding.encoding)
894	n/a	self.console = PseudoOutputFile(self, "console", IOBinding.encoding)
895	n/a	if not use_subprocess:
896	n/a	sys.stdout = self.stdout
897	n/a	sys.stderr = self.stderr
898	n/a	sys.stdin = self.stdin
899	n/a	try:
900	n/a	# page help() text to shell.
901	n/a	import pydoc # import must be done here to capture i/o rebinding.
902	n/a	# XXX KBK 27Dec07 use a textView someday, but must work w/o subproc
903	n/a	pydoc.pager = pydoc.plainpager
904	n/a	except:
905	n/a	sys.stderr = sys.__stderr__
906	n/a	raise
907	n/a	#
908	n/a	self.history = self.History(self.text)
909	n/a	#
910	n/a	self.pollinterval = 50 # millisec
911	n/a
912	n/a	def get_standard_extension_names(self):
913	n/a	return idleConf.GetExtensions(shell_only=True)
914	n/a
915	n/a	reading = False
916	n/a	executing = False
917	n/a	canceled = False
918	n/a	endoffile = False
919	n/a	closing = False
920	n/a	_stop_readline_flag = False
921	n/a
922	n/a	def set_warning_stream(self, stream):
923	n/a	global warning_stream
924	n/a	warning_stream = stream
925	n/a
926	n/a	def get_warning_stream(self):
927	n/a	return warning_stream
928	n/a
929	n/a	def toggle_debugger(self, event=None):
930	n/a	if self.executing:
931	n/a	tkMessageBox.showerror("Don't debug now",
932	n/a	"You can only toggle the debugger when idle",
933	n/a	master=self.text)
934	n/a	self.set_debugger_indicator()
935	n/a	return "break"
936	n/a	else:
937	n/a	db = self.interp.getdebugger()
938	n/a	if db:
939	n/a	self.close_debugger()
940	n/a	else:
941	n/a	self.open_debugger()
942	n/a
943	n/a	def set_debugger_indicator(self):
944	n/a	db = self.interp.getdebugger()
945	n/a	self.setvar("<<toggle-debugger>>", not not db)
946	n/a
947	n/a	def toggle_jit_stack_viewer(self, event=None):
948	n/a	pass # All we need is the variable
949	n/a
950	n/a	def close_debugger(self):
951	n/a	db = self.interp.getdebugger()
952	n/a	if db:
953	n/a	self.interp.setdebugger(None)
954	n/a	db.close()
955	n/a	if self.interp.rpcclt:
956	n/a	RemoteDebugger.close_remote_debugger(self.interp.rpcclt)
957	n/a	self.resetoutput()
958	n/a	self.console.write("[DEBUG OFF]\n")
959	n/a	sys.ps1 = ">>> "
960	n/a	self.showprompt()
961	n/a	self.set_debugger_indicator()
962	n/a
963	n/a	def open_debugger(self):
964	n/a	if self.interp.rpcclt:
965	n/a	dbg_gui = RemoteDebugger.start_remote_debugger(self.interp.rpcclt,
966	n/a	self)
967	n/a	else:
968	n/a	dbg_gui = Debugger.Debugger(self)
969	n/a	self.interp.setdebugger(dbg_gui)
970	n/a	dbg_gui.load_breakpoints()
971	n/a	sys.ps1 = "[DEBUG ON]\n>>> "
972	n/a	self.showprompt()
973	n/a	self.set_debugger_indicator()
974	n/a
975	n/a	def beginexecuting(self):
976	n/a	"Helper for ModifiedInterpreter"
977	n/a	self.resetoutput()
978	n/a	self.executing = 1
979	n/a
980	n/a	def endexecuting(self):
981	n/a	"Helper for ModifiedInterpreter"
982	n/a	self.executing = 0
983	n/a	self.canceled = 0
984	n/a	self.showprompt()
985	n/a
986	n/a	def close(self):
987	n/a	"Extend EditorWindow.close()"
988	n/a	if self.executing:
989	n/a	response = tkMessageBox.askokcancel(
990	n/a	"Kill?",
991	n/a	"The program is still running!\n Do you want to kill it?",
992	n/a	default="ok",
993	n/a	parent=self.text)
994	n/a	if response is False:
995	n/a	return "cancel"
996	n/a	self.stop_readline()
997	n/a	self.canceled = True
998	n/a	self.closing = True
999	n/a	return EditorWindow.close(self)
1000	n/a
1001	n/a	def _close(self):
1002	n/a	"Extend EditorWindow._close(), shut down debugger and execution server"
1003	n/a	self.close_debugger()
1004	n/a	if use_subprocess:
1005	n/a	self.interp.kill_subprocess()
1006	n/a	# Restore std streams
1007	n/a	sys.stdout = self.save_stdout
1008	n/a	sys.stderr = self.save_stderr
1009	n/a	sys.stdin = self.save_stdin
1010	n/a	# Break cycles
1011	n/a	self.interp = None
1012	n/a	self.console = None
1013	n/a	self.flist.pyshell = None
1014	n/a	self.history = None
1015	n/a	EditorWindow._close(self)
1016	n/a
1017	n/a	def ispythonsource(self, filename):
1018	n/a	"Override EditorWindow method: never remove the colorizer"
1019	n/a	return True
1020	n/a
1021	n/a	def short_title(self):
1022	n/a	return self.shell_title
1023	n/a
1024	n/a	COPYRIGHT = \
1025	n/a	'Type "copyright", "credits" or "license()" for more information.'
1026	n/a
1027	n/a	def begin(self):
1028	n/a	self.text.mark_set("iomark", "insert")
1029	n/a	self.resetoutput()
1030	n/a	if use_subprocess:
1031	n/a	nosub = ''
1032	n/a	client = self.interp.start_subprocess()
1033	n/a	if not client:
1034	n/a	self.close()
1035	n/a	return False
1036	n/a	else:
1037	n/a	nosub = ("==== No Subprocess ====\n\n" +
1038	n/a	"WARNING: Running IDLE without a Subprocess is deprecated\n" +
1039	n/a	"and will be removed in a later version. See Help/IDLE Help\n" +
1040	n/a	"for details.\n\n")
1041	n/a	sys.displayhook = rpc.displayhook
1042	n/a
1043	n/a	self.write("Python %s on %s\n%s\n%s" %
1044	n/a	(sys.version, sys.platform, self.COPYRIGHT, nosub))
1045	n/a	self.showprompt()
1046	n/a	import tkinter
1047	n/a	tkinter._default_root = None # 03Jan04 KBK What's this?
1048	n/a	return True
1049	n/a
1050	n/a	def stop_readline(self):
1051	n/a	if not self.reading: # no nested mainloop to exit.
1052	n/a	return
1053	n/a	self._stop_readline_flag = True
1054	n/a	self.top.quit()
1055	n/a
1056	n/a	def readline(self):
1057	n/a	save = self.reading
1058	n/a	try:
1059	n/a	self.reading = 1
1060	n/a	self.top.mainloop() # nested mainloop()
1061	n/a	finally:
1062	n/a	self.reading = save
1063	n/a	if self._stop_readline_flag:
1064	n/a	self._stop_readline_flag = False
1065	n/a	return ""
1066	n/a	line = self.text.get("iomark", "end-1c")
1067	n/a	if len(line) == 0: # may be EOF if we quit our mainloop with Ctrl-C
1068	n/a	line = "\n"
1069	n/a	self.resetoutput()
1070	n/a	if self.canceled:
1071	n/a	self.canceled = 0
1072	n/a	if not use_subprocess:
1073	n/a	raise KeyboardInterrupt
1074	n/a	if self.endoffile:
1075	n/a	self.endoffile = 0
1076	n/a	line = ""
1077	n/a	return line
1078	n/a
1079	n/a	def isatty(self):
1080	n/a	return True
1081	n/a
1082	n/a	def cancel_callback(self, event=None):
1083	n/a	try:
1084	n/a	if self.text.compare("sel.first", "!=", "sel.last"):
1085	n/a	return # Active selection -- always use default binding
1086	n/a	except:
1087	n/a	pass
1088	n/a	if not (self.executing or self.reading):
1089	n/a	self.resetoutput()
1090	n/a	self.interp.write("KeyboardInterrupt\n")
1091	n/a	self.showprompt()
1092	n/a	return "break"
1093	n/a	self.endoffile = 0
1094	n/a	self.canceled = 1
1095	n/a	if (self.executing and self.interp.rpcclt):
1096	n/a	if self.interp.getdebugger():
1097	n/a	self.interp.restart_subprocess()
1098	n/a	else:
1099	n/a	self.interp.interrupt_subprocess()
1100	n/a	if self.reading:
1101	n/a	self.top.quit() # exit the nested mainloop() in readline()
1102	n/a	return "break"
1103	n/a
1104	n/a	def eof_callback(self, event):
1105	n/a	if self.executing and not self.reading:
1106	n/a	return # Let the default binding (delete next char) take over
1107	n/a	if not (self.text.compare("iomark", "==", "insert") and
1108	n/a	self.text.compare("insert", "==", "end-1c")):
1109	n/a	return # Let the default binding (delete next char) take over
1110	n/a	if not self.executing:
1111	n/a	self.resetoutput()
1112	n/a	self.close()
1113	n/a	else:
1114	n/a	self.canceled = 0
1115	n/a	self.endoffile = 1
1116	n/a	self.top.quit()
1117	n/a	return "break"
1118	n/a
1119	n/a	def linefeed_callback(self, event):
1120	n/a	# Insert a linefeed without entering anything (still autoindented)
1121	n/a	if self.reading:
1122	n/a	self.text.insert("insert", "\n")
1123	n/a	self.text.see("insert")
1124	n/a	else:
1125	n/a	self.newline_and_indent_event(event)
1126	n/a	return "break"
1127	n/a
1128	n/a	def enter_callback(self, event):
1129	n/a	if self.executing and not self.reading:
1130	n/a	return # Let the default binding (insert '\n') take over
1131	n/a	# If some text is selected, recall the selection
1132	n/a	# (but only if this before the I/O mark)
1133	n/a	try:
1134	n/a	sel = self.text.get("sel.first", "sel.last")
1135	n/a	if sel:
1136	n/a	if self.text.compare("sel.last", "<=", "iomark"):
1137	n/a	self.recall(sel, event)
1138	n/a	return "break"
1139	n/a	except:
1140	n/a	pass
1141	n/a	# If we're strictly before the line containing iomark, recall
1142	n/a	# the current line, less a leading prompt, less leading or
1143	n/a	# trailing whitespace
1144	n/a	if self.text.compare("insert", "<", "iomark linestart"):
1145	n/a	# Check if there's a relevant stdin range -- if so, use it
1146	n/a	prev = self.text.tag_prevrange("stdin", "insert")
1147	n/a	if prev and self.text.compare("insert", "<", prev[1]):
1148	n/a	self.recall(self.text.get(prev[0], prev[1]), event)
1149	n/a	return "break"
1150	n/a	next = self.text.tag_nextrange("stdin", "insert")
1151	n/a	if next and self.text.compare("insert lineend", ">=", next[0]):
1152	n/a	self.recall(self.text.get(next[0], next[1]), event)
1153	n/a	return "break"
1154	n/a	# No stdin mark -- just get the current line, less any prompt
1155	n/a	indices = self.text.tag_nextrange("console", "insert linestart")
1156	n/a	if indices and \
1157	n/a	self.text.compare(indices[0], "<=", "insert linestart"):
1158	n/a	self.recall(self.text.get(indices[1], "insert lineend"), event)
1159	n/a	else:
1160	n/a	self.recall(self.text.get("insert linestart", "insert lineend"), event)
1161	n/a	return "break"
1162	n/a	# If we're between the beginning of the line and the iomark, i.e.
1163	n/a	# in the prompt area, move to the end of the prompt
1164	n/a	if self.text.compare("insert", "<", "iomark"):
1165	n/a	self.text.mark_set("insert", "iomark")
1166	n/a	# If we're in the current input and there's only whitespace
1167	n/a	# beyond the cursor, erase that whitespace first
1168	n/a	s = self.text.get("insert", "end-1c")
1169	n/a	if s and not s.strip():
1170	n/a	self.text.delete("insert", "end-1c")
1171	n/a	# If we're in the current input before its last line,
1172	n/a	# insert a newline right at the insert point
1173	n/a	if self.text.compare("insert", "<", "end-1c linestart"):
1174	n/a	self.newline_and_indent_event(event)
1175	n/a	return "break"
1176	n/a	# We're in the last line; append a newline and submit it
1177	n/a	self.text.mark_set("insert", "end-1c")
1178	n/a	if self.reading:
1179	n/a	self.text.insert("insert", "\n")
1180	n/a	self.text.see("insert")
1181	n/a	else:
1182	n/a	self.newline_and_indent_event(event)
1183	n/a	self.text.tag_add("stdin", "iomark", "end-1c")
1184	n/a	self.text.update_idletasks()
1185	n/a	if self.reading:
1186	n/a	self.top.quit() # Break out of recursive mainloop()
1187	n/a	else:
1188	n/a	self.runit()
1189	n/a	return "break"
1190	n/a
1191	n/a	def recall(self, s, event):
1192	n/a	# remove leading and trailing empty or whitespace lines
1193	n/a	s = re.sub(r'^\s*\n', '' , s)
1194	n/a	s = re.sub(r'\n\s*$', '', s)
1195	n/a	lines = s.split('\n')
1196	n/a	self.text.undo_block_start()
1197	n/a	try:
1198	n/a	self.text.tag_remove("sel", "1.0", "end")
1199	n/a	self.text.mark_set("insert", "end-1c")
1200	n/a	prefix = self.text.get("insert linestart", "insert")
1201	n/a	if prefix.rstrip().endswith(':'):
1202	n/a	self.newline_and_indent_event(event)
1203	n/a	prefix = self.text.get("insert linestart", "insert")
1204	n/a	self.text.insert("insert", lines[0].strip())
1205	n/a	if len(lines) > 1:
1206	n/a	orig_base_indent = re.search(r'^([ \t]*)', lines[0]).group(0)
1207	n/a	new_base_indent = re.search(r'^([ \t]*)', prefix).group(0)
1208	n/a	for line in lines[1:]:
1209	n/a	if line.startswith(orig_base_indent):
1210	n/a	# replace orig base indentation with new indentation
1211	n/a	line = new_base_indent + line[len(orig_base_indent):]
1212	n/a	self.text.insert('insert', '\n'+line.rstrip())
1213	n/a	finally:
1214	n/a	self.text.see("insert")
1215	n/a	self.text.undo_block_stop()
1216	n/a
1217	n/a	def runit(self):
1218	n/a	line = self.text.get("iomark", "end-1c")
1219	n/a	# Strip off last newline and surrounding whitespace.
1220	n/a	# (To allow you to hit return twice to end a statement.)
1221	n/a	i = len(line)
1222	n/a	while i > 0 and line[i-1] in " \t":
1223	n/a	i = i-1
1224	n/a	if i > 0 and line[i-1] == "\n":
1225	n/a	i = i-1
1226	n/a	while i > 0 and line[i-1] in " \t":
1227	n/a	i = i-1
1228	n/a	line = line[:i]
1229	n/a	more = self.interp.runsource(line)
1230	n/a
1231	n/a	def open_stack_viewer(self, event=None):
1232	n/a	if self.interp.rpcclt:
1233	n/a	return self.interp.remote_stack_viewer()
1234	n/a	try:
1235	n/a	sys.last_traceback
1236	n/a	except:
1237	n/a	tkMessageBox.showerror("No stack trace",
1238	n/a	"There is no stack trace yet.\n"
1239	n/a	"(sys.last_traceback is not defined)",
1240	n/a	master=self.text)
1241	n/a	return
1242	n/a	from idlelib.StackViewer import StackBrowser
1243	n/a	sv = StackBrowser(self.root, self.flist)
1244	n/a
1245	n/a	def view_restart_mark(self, event=None):
1246	n/a	self.text.see("iomark")
1247	n/a	self.text.see("restart")
1248	n/a
1249	n/a	def restart_shell(self, event=None):
1250	n/a	"Callback for Run/Restart Shell Cntl-F6"
1251	n/a	self.interp.restart_subprocess(with_cwd=True)
1252	n/a
1253	n/a	def showprompt(self):
1254	n/a	self.resetoutput()
1255	n/a	try:
1256	n/a	s = str(sys.ps1)
1257	n/a	except:
1258	n/a	s = ""
1259	n/a	self.console.write(s)
1260	n/a	self.text.mark_set("insert", "end-1c")
1261	n/a	self.set_line_and_column()
1262	n/a	self.io.reset_undo()
1263	n/a
1264	n/a	def resetoutput(self):
1265	n/a	source = self.text.get("iomark", "end-1c")
1266	n/a	if self.history:
1267	n/a	self.history.store(source)
1268	n/a	if self.text.get("end-2c") != "\n":
1269	n/a	self.text.insert("end-1c", "\n")
1270	n/a	self.text.mark_set("iomark", "end-1c")
1271	n/a	self.set_line_and_column()
1272	n/a
1273	n/a	def write(self, s, tags=()):
1274	n/a	if isinstance(s, str) and len(s) and max(s) > '\uffff':
1275	n/a	# Tk doesn't support outputting non-BMP characters
1276	n/a	# Let's assume what printed string is not very long,
1277	n/a	# find first non-BMP character and construct informative
1278	n/a	# UnicodeEncodeError exception.
1279	n/a	for start, char in enumerate(s):
1280	n/a	if char > '\uffff':
1281	n/a	break
1282	n/a	raise UnicodeEncodeError("UCS-2", char, start, start+1,
1283	n/a	'Non-BMP character not supported in Tk')
1284	n/a	try:
1285	n/a	self.text.mark_gravity("iomark", "right")
1286	n/a	count = OutputWindow.write(self, s, tags, "iomark")
1287	n/a	self.text.mark_gravity("iomark", "left")
1288	n/a	except:
1289	n/a	raise ###pass # ### 11Aug07 KBK if we are expecting exceptions
1290	n/a	# let's find out what they are and be specific.
1291	n/a	if self.canceled:
1292	n/a	self.canceled = 0
1293	n/a	if not use_subprocess:
1294	n/a	raise KeyboardInterrupt
1295	n/a	return count
1296	n/a
1297	n/a	def rmenu_check_cut(self):
1298	n/a	try:
1299	n/a	if self.text.compare('sel.first', '<', 'iomark'):
1300	n/a	return 'disabled'
1301	n/a	except TclError: # no selection, so the index 'sel.first' doesn't exist
1302	n/a	return 'disabled'
1303	n/a	return super().rmenu_check_cut()
1304	n/a
1305	n/a	def rmenu_check_paste(self):
1306	n/a	if self.text.compare('insert','<','iomark'):
1307	n/a	return 'disabled'
1308	n/a	return super().rmenu_check_paste()
1309	n/a
1310	n/a	class PseudoFile(io.TextIOBase):
1311	n/a
1312	n/a	def __init__(self, shell, tags, encoding=None):
1313	n/a	self.shell = shell
1314	n/a	self.tags = tags
1315	n/a	self._encoding = encoding
1316	n/a
1317	n/a	@property
1318	n/a	def encoding(self):
1319	n/a	return self._encoding
1320	n/a
1321	n/a	@property
1322	n/a	def name(self):
1323	n/a	return '<%s>' % self.tags
1324	n/a
1325	n/a	def isatty(self):
1326	n/a	return True
1327	n/a
1328	n/a
1329	n/a	class PseudoOutputFile(PseudoFile):
1330	n/a
1331	n/a	def writable(self):
1332	n/a	return True
1333	n/a
1334	n/a	def write(self, s):
1335	n/a	if self.closed:
1336	n/a	raise ValueError("write to closed file")
1337	n/a	if not isinstance(s, str):
1338	n/a	raise TypeError('must be str, not ' + type(s).__name__)
1339	n/a	return self.shell.write(s, self.tags)
1340	n/a
1341	n/a
1342	n/a	class PseudoInputFile(PseudoFile):
1343	n/a
1344	n/a	def __init__(self, shell, tags, encoding=None):
1345	n/a	PseudoFile.__init__(self, shell, tags, encoding)
1346	n/a	self._line_buffer = ''
1347	n/a
1348	n/a	def readable(self):
1349	n/a	return True
1350	n/a
1351	n/a	def read(self, size=-1):
1352	n/a	if self.closed:
1353	n/a	raise ValueError("read from closed file")
1354	n/a	if size is None:
1355	n/a	size = -1
1356	n/a	elif not isinstance(size, int):
1357	n/a	raise TypeError('must be int, not ' + type(size).__name__)
1358	n/a	result = self._line_buffer
1359	n/a	self._line_buffer = ''
1360	n/a	if size < 0:
1361	n/a	while True:
1362	n/a	line = self.shell.readline()
1363	n/a	if not line: break
1364	n/a	result += line
1365	n/a	else:
1366	n/a	while len(result) < size:
1367	n/a	line = self.shell.readline()
1368	n/a	if not line: break
1369	n/a	result += line
1370	n/a	self._line_buffer = result[size:]
1371	n/a	result = result[:size]
1372	n/a	return result
1373	n/a
1374	n/a	def readline(self, size=-1):
1375	n/a	if self.closed:
1376	n/a	raise ValueError("read from closed file")
1377	n/a	if size is None:
1378	n/a	size = -1
1379	n/a	elif not isinstance(size, int):
1380	n/a	raise TypeError('must be int, not ' + type(size).__name__)
1381	n/a	line = self._line_buffer or self.shell.readline()
1382	n/a	if size < 0:
1383	n/a	size = len(line)
1384	n/a	self._line_buffer = line[size:]
1385	n/a	return line[:size]
1386	n/a
1387	n/a	def close(self):
1388	n/a	self.shell.close()
1389	n/a
1390	n/a
1391	n/a	usage_msg = """\
1392	n/a
1393	n/a	USAGE: idle [-deins] [-t title] [file]*
1394	n/a	idle [-dns] [-t title] (-c cmd \| -r file) [arg]*
1395	n/a	idle [-dns] [-t title] - [arg]*
1396	n/a
1397	n/a	-h print this help message and exit
1398	n/a	-n run IDLE without a subprocess (DEPRECATED,
1399	n/a	see Help/IDLE Help for details)
1400	n/a
1401	n/a	The following options will override the IDLE 'settings' configuration:
1402	n/a
1403	n/a	-e open an edit window
1404	n/a	-i open a shell window
1405	n/a
1406	n/a	The following options imply -i and will open a shell:
1407	n/a
1408	n/a	-c cmd run the command in a shell, or
1409	n/a	-r file run script from file
1410	n/a
1411	n/a	-d enable the debugger
1412	n/a	-s run $IDLESTARTUP or $PYTHONSTARTUP before anything else
1413	n/a	-t title set title of shell window
1414	n/a
1415	n/a	A default edit window will be bypassed when -c, -r, or - are used.
1416	n/a
1417	n/a	[arg]* are passed to the command (-c) or script (-r) in sys.argv[1:].
1418	n/a
1419	n/a	Examples:
1420	n/a
1421	n/a	idle
1422	n/a	Open an edit window or shell depending on IDLE's configuration.
1423	n/a
1424	n/a	idle foo.py foobar.py
1425	n/a	Edit the files, also open a shell if configured to start with shell.
1426	n/a
1427	n/a	idle -est "Baz" foo.py
1428	n/a	Run $IDLESTARTUP or $PYTHONSTARTUP, edit foo.py, and open a shell
1429	n/a	window with the title "Baz".
1430	n/a
1431	n/a	idle -c "import sys; print(sys.argv)" "foo"
1432	n/a	Open a shell window and run the command, passing "-c" in sys.argv[0]
1433	n/a	and "foo" in sys.argv[1].
1434	n/a
1435	n/a	idle -d -s -r foo.py "Hello World"
1436	n/a	Open a shell window, run a startup script, enable the debugger, and
1437	n/a	run foo.py, passing "foo.py" in sys.argv[0] and "Hello World" in
1438	n/a	sys.argv[1].
1439	n/a
1440	n/a	echo "import sys; print(sys.argv)" \| idle - "foobar"
1441	n/a	Open a shell window, run the script piped in, passing '' in sys.argv[0]
1442	n/a	and "foobar" in sys.argv[1].
1443	n/a	"""
1444	n/a
1445	n/a	def main():
1446	n/a	global flist, root, use_subprocess
1447	n/a
1448	n/a	capture_warnings(True)
1449	n/a	use_subprocess = True
1450	n/a	enable_shell = False
1451	n/a	enable_edit = False
1452	n/a	debug = False
1453	n/a	cmd = None
1454	n/a	script = None
1455	n/a	startup = False
1456	n/a	try:
1457	n/a	opts, args = getopt.getopt(sys.argv[1:], "c:deihnr:st:")
1458	n/a	except getopt.error as msg:
1459	n/a	sys.stderr.write("Error: %s\n" % str(msg))
1460	n/a	sys.stderr.write(usage_msg)
1461	n/a	sys.exit(2)
1462	n/a	for o, a in opts:
1463	n/a	if o == '-c':
1464	n/a	cmd = a
1465	n/a	enable_shell = True
1466	n/a	if o == '-d':
1467	n/a	debug = True
1468	n/a	enable_shell = True
1469	n/a	if o == '-e':
1470	n/a	enable_edit = True
1471	n/a	if o == '-h':
1472	n/a	sys.stdout.write(usage_msg)
1473	n/a	sys.exit()
1474	n/a	if o == '-i':
1475	n/a	enable_shell = True
1476	n/a	if o == '-n':
1477	n/a	print(" Warning: running IDLE without a subprocess is deprecated.",
1478	n/a	file=sys.stderr)
1479	n/a	use_subprocess = False
1480	n/a	if o == '-r':
1481	n/a	script = a
1482	n/a	if os.path.isfile(script):
1483	n/a	pass
1484	n/a	else:
1485	n/a	print("No script file: ", script)
1486	n/a	sys.exit()
1487	n/a	enable_shell = True
1488	n/a	if o == '-s':
1489	n/a	startup = True
1490	n/a	enable_shell = True
1491	n/a	if o == '-t':
1492	n/a	PyShell.shell_title = a
1493	n/a	enable_shell = True
1494	n/a	if args and args[0] == '-':
1495	n/a	cmd = sys.stdin.read()
1496	n/a	enable_shell = True
1497	n/a	# process sys.argv and sys.path:
1498	n/a	for i in range(len(sys.path)):
1499	n/a	sys.path[i] = os.path.abspath(sys.path[i])
1500	n/a	if args and args[0] == '-':
1501	n/a	sys.argv = [''] + args[1:]
1502	n/a	elif cmd:
1503	n/a	sys.argv = ['-c'] + args
1504	n/a	elif script:
1505	n/a	sys.argv = [script] + args
1506	n/a	elif args:
1507	n/a	enable_edit = True
1508	n/a	pathx = []
1509	n/a	for filename in args:
1510	n/a	pathx.append(os.path.dirname(filename))
1511	n/a	for dir in pathx:
1512	n/a	dir = os.path.abspath(dir)
1513	n/a	if not dir in sys.path:
1514	n/a	sys.path.insert(0, dir)
1515	n/a	else:
1516	n/a	dir = os.getcwd()
1517	n/a	if dir not in sys.path:
1518	n/a	sys.path.insert(0, dir)
1519	n/a	# check the IDLE settings configuration (but command line overrides)
1520	n/a	edit_start = idleConf.GetOption('main', 'General',
1521	n/a	'editor-on-startup', type='bool')
1522	n/a	enable_edit = enable_edit or edit_start
1523	n/a	enable_shell = enable_shell or not enable_edit
1524	n/a	# start editor and/or shell windows:
1525	n/a	root = Tk(className="Idle")
1526	n/a
1527	n/a	fixwordbreaks(root)
1528	n/a	root.withdraw()
1529	n/a	flist = PyShellFileList(root)
1530	n/a	macosxSupport.setupApp(root, flist)
1531	n/a
1532	n/a	if enable_edit:
1533	n/a	if not (cmd or script):
1534	n/a	for filename in args[:]:
1535	n/a	if flist.open(filename) is None:
1536	n/a	# filename is a directory actually, disconsider it
1537	n/a	args.remove(filename)
1538	n/a	if not args:
1539	n/a	flist.new()
1540	n/a	if enable_shell:
1541	n/a	shell = flist.open_shell()
1542	n/a	if not shell:
1543	n/a	return # couldn't open shell
1544	n/a
1545	n/a	if macosxSupport.runningAsOSXApp() and flist.dict:
1546	n/a	# On OSX: when the user has double-clicked on a file that causes
1547	n/a	# IDLE to be launched the shell window will open just in front of
1548	n/a	# the file she wants to see. Lower the interpreter window when
1549	n/a	# there are open files.
1550	n/a	shell.top.lower()
1551	n/a
1552	n/a	shell = flist.pyshell
1553	n/a	# handle remaining options:
1554	n/a	if debug:
1555	n/a	shell.open_debugger()
1556	n/a	if startup:
1557	n/a	filename = os.environ.get("IDLESTARTUP") or \
1558	n/a	os.environ.get("PYTHONSTARTUP")
1559	n/a	if filename and os.path.isfile(filename):
1560	n/a	shell.interp.execfile(filename)
1561	n/a	if shell and cmd or script:
1562	n/a	shell.interp.runcommand("""if 1:
1563	n/a	import sys as _sys
1564	n/a	_sys.argv = %r
1565	n/a	del _sys
1566	n/a	\n""" % (sys.argv,))
1567	n/a	if cmd:
1568	n/a	shell.interp.execsource(cmd)
1569	n/a	elif script:
1570	n/a	shell.interp.prepend_syspath(script)
1571	n/a	shell.interp.execfile(script)
1572	n/a
1573	n/a	# Check for problematic OS X Tk versions and print a warning message
1574	n/a	# in the IDLE shell window; this is less intrusive than always opening
1575	n/a	# a separate window.
1576	n/a	tkversionwarning = macosxSupport.tkVersionWarning(root)
1577	n/a	if tkversionwarning:
1578	n/a	shell.interp.runcommand(''.join(("print('", tkversionwarning, "')")))
1579	n/a
1580	n/a	while flist.inversedict: # keep IDLE running while files are open.
1581	n/a	root.mainloop()
1582	n/a	root.destroy()
1583	n/a	capture_warnings(False)
1584	n/a
1585	n/a	if __name__ == "__main__":
1586	n/a	sys.modules['PyShell'] = sys.modules['__main__']
1587	n/a	main()
1588	n/a
1589	n/a	capture_warnings(False) # Make sure turned off; see issue 18081