blob: 4e7440038ac997cd91569e97c1bbdc5f6e6da546 [file] [log] [blame]
Benjamin Peterson90f5ba52010-03-11 22:53:45 +00001#! /usr/bin/env python3
David Scherer7aced172000-08-15 01:13:23 +00002
Jelle Zijlstra0579e812017-03-26 19:17:39 -07003import sys
Terry Jan Reedy6a258c82019-03-26 19:58:19 -04004if __name__ == "__main__":
5 sys.modules['idlelib.pyshell'] = sys.modules['__main__']
Jelle Zijlstra0579e812017-03-26 19:17:39 -07006
Terry Jan Reedy1080d132016-06-09 21:09:15 -04007try:
8 from tkinter import *
9except ImportError:
10 print("** IDLE can't import Tkinter.\n"
11 "Your Python may not be configured for Tk. **", file=sys.__stderr__)
Terry Jan Reedye16265d2017-01-04 23:17:47 -050012 raise SystemExit(1)
Terry Jan Reedy800415e2018-06-11 14:14:32 -040013
Terry Jan Reedyfd88f312018-06-11 15:45:27 -040014# Valid arguments for the ...Awareness call below are defined in the following.
15# https://msdn.microsoft.com/en-us/library/windows/desktop/dn280512(v=vs.85).aspx
Terry Jan Reedy800415e2018-06-11 14:14:32 -040016if sys.platform == 'win32':
Terry Jan Reedy800415e2018-06-11 14:14:32 -040017 try:
Terry Jan Reedyd274afb2018-10-13 20:00:31 -040018 import ctypes
Terry Jan Reedy34a49aa2020-03-28 12:51:21 -040019 PROCESS_SYSTEM_DPI_AWARE = 1 # Int required.
Terry Jan Reedyfd88f312018-06-11 15:45:27 -040020 ctypes.OleDLL('shcore').SetProcessDpiAwareness(PROCESS_SYSTEM_DPI_AWARE)
Terry Jan Reedyd274afb2018-10-13 20:00:31 -040021 except (ImportError, AttributeError, OSError):
Terry Jan Reedy800415e2018-06-11 14:14:32 -040022 pass
23
Terry Jan Reedy879986d2021-01-25 06:33:18 -050024from tkinter import messagebox
Terry Jan Reedy1080d132016-06-09 21:09:15 -040025if TkVersion < 8.5:
26 root = Tk() # otherwise create root in main
27 root.withdraw()
Serhiy Storchakaa96c96f2017-09-21 11:20:06 +030028 from idlelib.run import fix_scaling
29 fix_scaling(root)
Terry Jan Reedy879986d2021-01-25 06:33:18 -050030 messagebox.showerror("Idle Cannot Start",
Terry Jan Reedye16265d2017-01-04 23:17:47 -050031 "Idle requires tcl/tk 8.5+, not %s." % TkVersion,
Terry Jan Reedy1080d132016-06-09 21:09:15 -040032 parent=root)
Terry Jan Reedye16265d2017-01-04 23:17:47 -050033 raise SystemExit(1)
Terry Jan Reedy1080d132016-06-09 21:09:15 -040034
Terry Jan Reedybfbaa6b2016-08-31 00:50:55 -040035from code import InteractiveInterpreter
Tal Einatb43cc312021-05-03 05:27:38 +030036import itertools
Terry Jan Reedybfbaa6b2016-08-31 00:50:55 -040037import linecache
David Scherer7aced172000-08-15 01:13:23 +000038import os
Kurt B. Kaiser6e44cc22002-11-30 06:18:00 +000039import os.path
Serhiy Storchaka7e4db2f2017-05-04 08:17:47 +030040from platform import python_version
David Scherer7aced172000-08-15 01:13:23 +000041import re
Chui Tey5d2af632002-05-26 13:36:41 +000042import socket
Victor Stinner979482a2011-09-02 01:00:40 +020043import subprocess
Tal Einat9ebe8792018-12-07 08:32:21 +020044from textwrap import TextWrapper
Kurt B. Kaiser003091c2003-02-17 18:57:16 +000045import threading
Victor Stinner979482a2011-09-02 01:00:40 +020046import time
47import tokenize
Terry Jan Reedybfbaa6b2016-08-31 00:50:55 -040048import warnings
David Scherer7aced172000-08-15 01:13:23 +000049
Terry Jan Reedy6fa5bdc2016-05-28 13:22:31 -040050from idlelib.colorizer import ColorDelegator
Terry Jan Reedy6fa5bdc2016-05-28 13:22:31 -040051from idlelib.config import idleConf
Tal Einat15d38612021-04-29 01:27:55 +030052from idlelib.delegator import Delegator
Terry Jan Reedy6fa5bdc2016-05-28 13:22:31 -040053from idlelib import debugger
54from idlelib import debugger_r
Terry Jan Reedybfbaa6b2016-08-31 00:50:55 -040055from idlelib.editor import EditorWindow, fixwordbreaks
56from idlelib.filelist import FileList
Terry Jan Reedybfbaa6b2016-08-31 00:50:55 -040057from idlelib.outwin import OutputWindow
Tal Einat15d38612021-04-29 01:27:55 +030058from idlelib import replace
Terry Jan Reedybfbaa6b2016-08-31 00:50:55 -040059from idlelib import rpc
Serhiy Storchakab690a272019-10-08 14:32:25 +030060from idlelib.run import idle_formatwarning, StdInputFile, StdOutputFile
Terry Jan Reedybfbaa6b2016-08-31 00:50:55 -040061from idlelib.undo import UndoDelegator
Chui Tey5d2af632002-05-26 13:36:41 +000062
Tal Einat15d38612021-04-29 01:27:55 +030063# Default for testing; defaults to True in main() for running.
64use_subprocess = False
65
Kurt B. Kaisere866c812009-04-04 21:07:39 +000066HOST = '127.0.0.1' # python execution server on localhost loopback
67PORT = 0 # someday pass in host, port for remote debug capability
Kurt B. Kaiserb9764192002-09-23 04:10:37 +000068
Kurt B. Kaiser49a5fe12004-07-04 01:25:56 +000069# Override warnings module to write to warning_stream. Initialize to send IDLE
70# internal warnings to the console. ScriptBinding.check_syntax() will
71# temporarily redirect the stream to the shell window to display warnings when
72# checking user's code.
Terry Jan Reedy95a3f112013-06-28 23:50:12 -040073warning_stream = sys.__stderr__ # None, at least on Windows, if no console.
Terry Jan Reedy95a3f112013-06-28 23:50:12 -040074
Terry Jan Reedy95a3f112013-06-28 23:50:12 -040075def idle_showwarning(
76 message, category, filename, lineno, file=None, line=None):
77 """Show Idle-format warning (after replacing warnings.showwarning).
78
79 The differences are the formatter called, the file=None replacement,
80 which can be None, the capture of the consequence AttributeError,
81 and the output of a hard-coded prompt.
82 """
83 if file is None:
84 file = warning_stream
85 try:
86 file.write(idle_formatwarning(
87 message, category, filename, lineno, line=line))
88 file.write(">>> ")
89 except (AttributeError, OSError):
90 pass # if file (probably __stderr__) is invalid, skip warning.
91
92_warnings_showwarning = None
93
94def capture_warnings(capture):
95 "Replace warning.showwarning with idle_showwarning, or reverse."
96
97 global _warnings_showwarning
98 if capture:
99 if _warnings_showwarning is None:
100 _warnings_showwarning = warnings.showwarning
101 warnings.showwarning = idle_showwarning
102 else:
103 if _warnings_showwarning is not None:
104 warnings.showwarning = _warnings_showwarning
105 _warnings_showwarning = None
106
107capture_warnings(True)
Chui Tey5d2af632002-05-26 13:36:41 +0000108
Kurt B. Kaiserf7a88992004-11-13 21:05:58 +0000109def extended_linecache_checkcache(filename=None,
110 orig_checkcache=linecache.checkcache):
Kurt B. Kaiser45186c42002-10-23 04:48:08 +0000111 """Extend linecache.checkcache to preserve the <pyshell#...> entries
112
Kurt B. Kaiserf7a88992004-11-13 21:05:58 +0000113 Rather than repeating the linecache code, patch it to save the
114 <pyshell#...> entries, call the original linecache.checkcache()
Guilherme Polo1fff0082009-08-14 15:05:30 +0000115 (skipping them), and then restore the saved entries.
Kurt B. Kaiserf7a88992004-11-13 21:05:58 +0000116
117 orig_checkcache is bound at definition time to the original
118 method, allowing it to be patched.
Kurt B. Kaiser45186c42002-10-23 04:48:08 +0000119 """
David Scherer7aced172000-08-15 01:13:23 +0000120 cache = linecache.cache
121 save = {}
Guilherme Polo1fff0082009-08-14 15:05:30 +0000122 for key in list(cache):
123 if key[:1] + key[-1:] == '<>':
124 save[key] = cache.pop(key)
125 orig_checkcache(filename)
David Scherer7aced172000-08-15 01:13:23 +0000126 cache.update(save)
Kurt B. Kaiser6655e4b2002-12-31 16:03:23 +0000127
Kurt B. Kaiser81885592002-11-29 22:10:53 +0000128# Patch linecache.checkcache():
129linecache.checkcache = extended_linecache_checkcache
David Scherer7aced172000-08-15 01:13:23 +0000130
Kurt B. Kaiser45186c42002-10-23 04:48:08 +0000131
David Scherer7aced172000-08-15 01:13:23 +0000132class PyShellEditorWindow(EditorWindow):
Kurt B. Kaiser183403a2004-08-22 05:14:32 +0000133 "Regular text edit window in IDLE, supports breakpoints"
Kurt B. Kaiser45186c42002-10-23 04:48:08 +0000134
David Scherer7aced172000-08-15 01:13:23 +0000135 def __init__(self, *args):
Kurt B. Kaiser45186c42002-10-23 04:48:08 +0000136 self.breakpoints = []
Raymond Hettinger931237e2003-07-09 18:48:24 +0000137 EditorWindow.__init__(self, *args)
David Scherer7aced172000-08-15 01:13:23 +0000138 self.text.bind("<<set-breakpoint-here>>", self.set_breakpoint_here)
Kurt B. Kaiser45186c42002-10-23 04:48:08 +0000139 self.text.bind("<<clear-breakpoint-here>>", self.clear_breakpoint_here)
David Scherer7aced172000-08-15 01:13:23 +0000140 self.text.bind("<<open-python-shell>>", self.flist.open_shell)
141
Terry Jan Reedy0048afc2019-09-16 19:04:21 -0400142 #TODO: don't read/write this from/to .idlerc when testing
terryjreedy223c7e72017-07-07 22:28:06 -0400143 self.breakpointPath = os.path.join(
144 idleConf.userdir, 'breakpoints.lst')
Chui Teya2adb0f2002-11-04 22:14:54 +0000145 # whenever a file is changed, restore breakpoints
Kurt B. Kaiserbfed3462002-12-14 04:38:51 +0000146 def filename_changed_hook(old_hook=self.io.filename_change_hook,
147 self=self):
Chui Teya2adb0f2002-11-04 22:14:54 +0000148 self.restore_file_breaks()
149 old_hook()
150 self.io.set_filename_change_hook(filename_changed_hook)
Roger Serwyd7c9d9c2013-04-02 22:37:12 -0500151 if self.io.filename:
152 self.restore_file_breaks()
Terry Jan Reedyda26cca2014-08-08 23:33:16 -0400153 self.color_breakpoint_text()
Chui Teya2adb0f2002-11-04 22:14:54 +0000154
Andrew Svetlovd1837672012-11-01 22:41:19 +0200155 rmenu_specs = [
156 ("Cut", "<<cut>>", "rmenu_check_cut"),
157 ("Copy", "<<copy>>", "rmenu_check_copy"),
158 ("Paste", "<<paste>>", "rmenu_check_paste"),
159 (None, None, None),
160 ("Set Breakpoint", "<<set-breakpoint-here>>", None),
161 ("Clear Breakpoint", "<<clear-breakpoint-here>>", None)
162 ]
David Scherer7aced172000-08-15 01:13:23 +0000163
Terry Jan Reedyda26cca2014-08-08 23:33:16 -0400164 def color_breakpoint_text(self, color=True):
165 "Turn colorizing of breakpoint text on or off"
Terry Jan Reedyefc72582014-10-12 22:58:47 -0400166 if self.io is None:
167 # possible due to update in restore_file_breaks
168 return
Terry Jan Reedyda26cca2014-08-08 23:33:16 -0400169 if color:
Terry Jan Reedyd0c0f002015-11-12 15:02:57 -0500170 theme = idleConf.CurrentTheme()
Terry Jan Reedyda26cca2014-08-08 23:33:16 -0400171 cfg = idleConf.GetHighlight(theme, "break")
172 else:
173 cfg = {'foreground': '', 'background': ''}
174 self.text.tag_config('BREAK', cfg)
175
Chui Teya2adb0f2002-11-04 22:14:54 +0000176 def set_breakpoint(self, lineno):
177 text = self.text
178 filename = self.io.filename
179 text.tag_add("BREAK", "%d.0" % lineno, "%d.0" % (lineno+1))
180 try:
Terry Jan Reedy038c16b2015-05-15 23:03:17 -0400181 self.breakpoints.index(lineno)
Chui Teya2adb0f2002-11-04 22:14:54 +0000182 except ValueError: # only add if missing, i.e. do once
183 self.breakpoints.append(lineno)
184 try: # update the subprocess debugger
185 debug = self.flist.pyshell.interp.debugger
186 debug.set_breakpoint_here(filename, lineno)
187 except: # but debugger may not be active right now....
188 pass
189
David Scherer7aced172000-08-15 01:13:23 +0000190 def set_breakpoint_here(self, event=None):
Kurt B. Kaiser45186c42002-10-23 04:48:08 +0000191 text = self.text
192 filename = self.io.filename
193 if not filename:
194 text.bell()
David Scherer7aced172000-08-15 01:13:23 +0000195 return
Kurt B. Kaiser45186c42002-10-23 04:48:08 +0000196 lineno = int(float(text.index("insert")))
Chui Teya2adb0f2002-11-04 22:14:54 +0000197 self.set_breakpoint(lineno)
David Scherer7aced172000-08-15 01:13:23 +0000198
Kurt B. Kaiser669f4c32002-06-20 04:01:47 +0000199 def clear_breakpoint_here(self, event=None):
Kurt B. Kaiser45186c42002-10-23 04:48:08 +0000200 text = self.text
201 filename = self.io.filename
202 if not filename:
203 text.bell()
Kurt B. Kaiser669f4c32002-06-20 04:01:47 +0000204 return
Kurt B. Kaiser45186c42002-10-23 04:48:08 +0000205 lineno = int(float(text.index("insert")))
206 try:
207 self.breakpoints.remove(lineno)
208 except:
209 pass
210 text.tag_remove("BREAK", "insert linestart",\
211 "insert lineend +1char")
212 try:
213 debug = self.flist.pyshell.interp.debugger
214 debug.clear_breakpoint_here(filename, lineno)
215 except:
216 pass
217
218 def clear_file_breaks(self):
219 if self.breakpoints:
220 text = self.text
221 filename = self.io.filename
222 if not filename:
223 text.bell()
224 return
225 self.breakpoints = []
226 text.tag_remove("BREAK", "1.0", END)
227 try:
228 debug = self.flist.pyshell.interp.debugger
229 debug.clear_file_breaks(filename)
230 except:
231 pass
232
Chui Teya2adb0f2002-11-04 22:14:54 +0000233 def store_file_breaks(self):
Kurt B. Kaiserbfed3462002-12-14 04:38:51 +0000234 "Save breakpoints when file is saved"
235 # XXX 13 Dec 2002 KBK Currently the file must be saved before it can
236 # be run. The breaks are saved at that time. If we introduce
237 # a temporary file save feature the save breaks functionality
238 # needs to be re-verified, since the breaks at the time the
239 # temp file is created may differ from the breaks at the last
Kurt B. Kaiser7f38ec02003-05-15 03:19:42 +0000240 # permanent save of the file. Currently, a break introduced
241 # after a save will be effective, but not persistent.
242 # This is necessary to keep the saved breaks synched with the
243 # saved file.
Kurt B. Kaiserbfed3462002-12-14 04:38:51 +0000244 #
Terry Jan Reedy6b04dc92014-10-12 01:11:05 -0400245 # Breakpoints are set as tagged ranges in the text.
246 # Since a modified file has to be saved before it is
Kurt B. Kaiserbfed3462002-12-14 04:38:51 +0000247 # run, and since self.breakpoints (from which the subprocess
248 # debugger is loaded) is updated during the save, the visible
249 # breaks stay synched with the subprocess even if one of these
250 # unexpected breakpoint deletions occurs.
251 breaks = self.breakpoints
252 filename = self.io.filename
Chui Teya2adb0f2002-11-04 22:14:54 +0000253 try:
Victor Stinner85c67722011-09-02 00:57:04 +0200254 with open(self.breakpointPath, "r") as fp:
255 lines = fp.readlines()
Andrew Svetlovf7a17b42012-12-25 16:47:37 +0200256 except OSError:
Kurt B. Kaiserbfed3462002-12-14 04:38:51 +0000257 lines = []
Ned Deilyf505b742011-12-14 14:58:24 -0800258 try:
259 with open(self.breakpointPath, "w") as new_file:
260 for line in lines:
261 if not line.startswith(filename + '='):
262 new_file.write(line)
263 self.update_breakpoints()
264 breaks = self.breakpoints
265 if breaks:
266 new_file.write(filename + '=' + str(breaks) + '\n')
Andrew Svetlovf7a17b42012-12-25 16:47:37 +0200267 except OSError as err:
Ned Deilyf505b742011-12-14 14:58:24 -0800268 if not getattr(self.root, "breakpoint_error_displayed", False):
269 self.root.breakpoint_error_displayed = True
Terry Jan Reedy879986d2021-01-25 06:33:18 -0500270 messagebox.showerror(title='IDLE Error',
Ned Deilyf505b742011-12-14 14:58:24 -0800271 message='Unable to update breakpoint list:\n%s'
272 % str(err),
273 parent=self.text)
Chui Teya2adb0f2002-11-04 22:14:54 +0000274
275 def restore_file_breaks(self):
276 self.text.update() # this enables setting "BREAK" tags to be visible
Roger Serwyd7c9d9c2013-04-02 22:37:12 -0500277 if self.io is None:
278 # can happen if IDLE closes due to the .update() call
279 return
Kurt B. Kaiserbfed3462002-12-14 04:38:51 +0000280 filename = self.io.filename
281 if filename is None:
282 return
Chui Tey69371d62002-11-04 23:39:45 +0000283 if os.path.isfile(self.breakpointPath):
Victor Stinner85c67722011-09-02 00:57:04 +0200284 with open(self.breakpointPath, "r") as fp:
285 lines = fp.readlines()
Chui Tey69371d62002-11-04 23:39:45 +0000286 for line in lines:
Kurt B. Kaiserbfed3462002-12-14 04:38:51 +0000287 if line.startswith(filename + '='):
Kurt B. Kaiser6655e4b2002-12-31 16:03:23 +0000288 breakpoint_linenumbers = eval(line[len(filename)+1:])
Chui Tey69371d62002-11-04 23:39:45 +0000289 for breakpoint_linenumber in breakpoint_linenumbers:
290 self.set_breakpoint(breakpoint_linenumber)
Chui Teya2adb0f2002-11-04 22:14:54 +0000291
Kurt B. Kaiserbfed3462002-12-14 04:38:51 +0000292 def update_breakpoints(self):
293 "Retrieves all the breakpoints in the current window"
Chui Teya2adb0f2002-11-04 22:14:54 +0000294 text = self.text
Kurt B. Kaiserbfed3462002-12-14 04:38:51 +0000295 ranges = text.tag_ranges("BREAK")
296 linenumber_list = self.ranges_to_linenumbers(ranges)
297 self.breakpoints = linenumber_list
298
299 def ranges_to_linenumbers(self, ranges):
300 lines = []
301 for index in range(0, len(ranges), 2):
Andrew Svetlov06c5c6d2012-07-31 19:48:00 +0300302 lineno = int(float(ranges[index].string))
303 end = int(float(ranges[index+1].string))
Kurt B. Kaiserbfed3462002-12-14 04:38:51 +0000304 while lineno < end:
305 lines.append(lineno)
306 lineno += 1
307 return lines
308
Kurt B. Kaiser11220fa2002-12-24 00:57:22 +0000309# XXX 13 Dec 2002 KBK Not used currently
Kurt B. Kaiserbfed3462002-12-14 04:38:51 +0000310# def saved_change_hook(self):
311# "Extend base method - clear breaks if module is modified"
312# if not self.get_saved():
313# self.clear_file_breaks()
314# EditorWindow.saved_change_hook(self)
Kurt B. Kaiser45186c42002-10-23 04:48:08 +0000315
316 def _close(self):
317 "Extend base method - clear breaks when module is closed"
318 self.clear_file_breaks()
319 EditorWindow._close(self)
Kurt B. Kaiser6655e4b2002-12-31 16:03:23 +0000320
David Scherer7aced172000-08-15 01:13:23 +0000321
322class PyShellFileList(FileList):
Kurt B. Kaiser183403a2004-08-22 05:14:32 +0000323 "Extend base class: IDLE supports a shell and breakpoints"
David Scherer7aced172000-08-15 01:13:23 +0000324
Kurt B. Kaiser183403a2004-08-22 05:14:32 +0000325 # override FileList's class variable, instances return PyShellEditorWindow
326 # instead of EditorWindow when new edit windows are created.
David Scherer7aced172000-08-15 01:13:23 +0000327 EditorWindow = PyShellEditorWindow
328
329 pyshell = None
330
331 def open_shell(self, event=None):
332 if self.pyshell:
Kurt B. Kaiser183403a2004-08-22 05:14:32 +0000333 self.pyshell.top.wakeup()
David Scherer7aced172000-08-15 01:13:23 +0000334 else:
335 self.pyshell = PyShell(self)
Kurt B. Kaiseraf3eb872004-01-21 18:54:30 +0000336 if self.pyshell:
337 if not self.pyshell.begin():
338 return None
David Scherer7aced172000-08-15 01:13:23 +0000339 return self.pyshell
340
341
342class ModifiedColorDelegator(ColorDelegator):
Kurt B. Kaiser83118c62002-06-24 17:03:37 +0000343 "Extend base class: colorizer for the shell window itself"
David Scherer7aced172000-08-15 01:13:23 +0000344 def recolorize_main(self):
345 self.tag_remove("TODO", "1.0", "iomark")
346 self.tag_add("SYNC", "1.0", "iomark")
347 ColorDelegator.recolorize_main(self)
Kurt B. Kaiser6655e4b2002-12-31 16:03:23 +0000348
Ned Deily8b2a56b2012-05-31 09:17:29 -0700349 def removecolors(self):
350 # Don't remove shell color tags before "iomark"
351 for tag in self.tagdefs:
352 self.tag_remove(tag, "iomark", "end")
353
Tal Einat15d38612021-04-29 01:27:55 +0300354
David Scherer7aced172000-08-15 01:13:23 +0000355class ModifiedUndoDelegator(UndoDelegator):
Kurt B. Kaiser83118c62002-06-24 17:03:37 +0000356 "Extend base class: forbid insert/delete before the I/O mark"
David Scherer7aced172000-08-15 01:13:23 +0000357 def insert(self, index, chars, tags=None):
358 try:
359 if self.delegate.compare(index, "<", "iomark"):
360 self.delegate.bell()
361 return
362 except TclError:
363 pass
364 UndoDelegator.insert(self, index, chars, tags)
365
366 def delete(self, index1, index2=None):
367 try:
368 if self.delegate.compare(index1, "<", "iomark"):
369 self.delegate.bell()
370 return
371 except TclError:
372 pass
373 UndoDelegator.delete(self, index1, index2)
374
Tal Einat15d38612021-04-29 01:27:55 +0300375 def undo_event(self, event):
376 # Temporarily monkey-patch the delegate's .insert() method to
377 # always use the "stdin" tag. This is needed for undo-ing
378 # deletions to preserve the "stdin" tag, because UndoDelegator
379 # doesn't preserve tags for deleted text.
380 orig_insert = self.delegate.insert
381 self.delegate.insert = \
382 lambda index, chars: orig_insert(index, chars, "stdin")
383 try:
384 super().undo_event(event)
385 finally:
386 self.delegate.insert = orig_insert
387
388
389class UserInputTaggingDelegator(Delegator):
390 """Delegator used to tag user input with "stdin"."""
391 def insert(self, index, chars, tags=None):
392 if tags is None:
393 tags = "stdin"
394 self.delegate.insert(index, chars, tags)
395
Kurt B. Kaiser67fd0ea2003-05-24 20:59:15 +0000396
397class MyRPCClient(rpc.RPCClient):
398
399 def handle_EOF(self):
400 "Override the base class - just re-raise EOFError"
401 raise EOFError
402
Terry Jan Reedy38da8052019-09-06 13:54:45 -0400403def restart_line(width, filename): # See bpo-38141.
404 """Return width long restart line formatted with filename.
405
406 Fill line with balanced '='s, with any extras and at least one at
407 the beginning. Do not end with a trailing space.
408 """
409 tag = f"= RESTART: {filename or 'Shell'} ="
410 if width >= len(tag):
411 div, mod = divmod((width -len(tag)), 2)
412 return f"{(div+mod)*'='}{tag}{div*'='}"
413 else:
414 return tag[:-2] # Remove ' ='.
415
Kurt B. Kaiser8d1f11b2003-05-26 22:20:34 +0000416
David Scherer7aced172000-08-15 01:13:23 +0000417class ModifiedInterpreter(InteractiveInterpreter):
418
419 def __init__(self, tkconsole):
420 self.tkconsole = tkconsole
421 locals = sys.modules['__main__'].__dict__
422 InteractiveInterpreter.__init__(self, locals=locals)
Kurt B. Kaiser6f805942003-05-24 21:12:46 +0000423 self.restarting = False
Kurt B. Kaisere866c812009-04-04 21:07:39 +0000424 self.subprocess_arglist = None
425 self.port = PORT
Ned Deily7aff4512011-10-30 20:01:35 -0700426 self.original_compiler_flags = self.compile.compiler.flags
David Scherer7aced172000-08-15 01:13:23 +0000427
Roger Serwy036e8492013-06-11 22:13:17 -0500428 _afterid = None
Chui Tey5d2af632002-05-26 13:36:41 +0000429 rpcclt = None
Ned Deilye5cad232011-08-02 18:47:13 -0700430 rpcsubproc = None
Chui Tey5d2af632002-05-26 13:36:41 +0000431
Kurt B. Kaiser6655e4b2002-12-31 16:03:23 +0000432 def spawn_subprocess(self):
Florent Xiclunafd1b0932010-03-28 00:25:02 +0000433 if self.subprocess_arglist is None:
Kurt B. Kaisere866c812009-04-04 21:07:39 +0000434 self.subprocess_arglist = self.build_subprocess_arglist()
Ned Deilye5cad232011-08-02 18:47:13 -0700435 self.rpcsubproc = subprocess.Popen(self.subprocess_arglist)
Kurt B. Kaiser63857a42002-09-05 02:31:20 +0000436
Tony Lowndsf53dec22002-12-20 04:24:43 +0000437 def build_subprocess_arglist(self):
Kurt B. Kaisere866c812009-04-04 21:07:39 +0000438 assert (self.port!=0), (
439 "Socket should have been assigned a port number.")
Tony Lownds2398d572003-05-13 15:28:21 +0000440 w = ['-W' + s for s in sys.warnoptions]
441 # Maybe IDLE is installed and is being accessed via sys.path,
442 # or maybe it's not installed and the idle.py script is being
443 # run from the IDLE source directory.
Kurt B. Kaiser62df0442003-05-28 01:47:46 +0000444 del_exitf = idleConf.GetOption('main', 'General', 'delete-exitfunc',
445 default=False, type='bool')
Terry Jan Reedy6a258c82019-03-26 19:58:19 -0400446 command = "__import__('idlelib.run').run.main(%r)" % (del_exitf,)
Ned Deilye5cad232011-08-02 18:47:13 -0700447 return [sys.executable] + w + ["-c", command, str(self.port)]
Tony Lowndsf2324b92002-09-29 00:34:10 +0000448
Kurt B. Kaiser63857a42002-09-05 02:31:20 +0000449 def start_subprocess(self):
Kurt B. Kaisere866c812009-04-04 21:07:39 +0000450 addr = (HOST, self.port)
451 # GUI makes several attempts to acquire socket, listens for connection
Kurt B. Kaiser5db48432003-05-15 03:40:51 +0000452 for i in range(3):
Chui Tey5d2af632002-05-26 13:36:41 +0000453 time.sleep(i)
454 try:
Kurt B. Kaiser67fd0ea2003-05-24 20:59:15 +0000455 self.rpcclt = MyRPCClient(addr)
Chui Tey5d2af632002-05-26 13:36:41 +0000456 break
Terry Jan Reedy038c16b2015-05-15 23:03:17 -0400457 except OSError:
Kurt B. Kaiseraf3eb872004-01-21 18:54:30 +0000458 pass
Chui Tey5d2af632002-05-26 13:36:41 +0000459 else:
Kurt B. Kaiseraf3eb872004-01-21 18:54:30 +0000460 self.display_port_binding_error()
461 return None
Kurt B. Kaisere866c812009-04-04 21:07:39 +0000462 # if PORT was 0, system will assign an 'ephemeral' port. Find it out:
463 self.port = self.rpcclt.listening_sock.getsockname()[1]
464 # if PORT was not 0, probably working with a remote execution server
465 if PORT != 0:
466 # To allow reconnection within the 2MSL wait (cf. Stevens TCP
467 # V1, 18.6), set SO_REUSEADDR. Note that this can be problematic
468 # on Windows since the implementation allows two active sockets on
469 # the same address!
470 self.rpcclt.listening_sock.setsockopt(socket.SOL_SOCKET,
471 socket.SO_REUSEADDR, 1)
472 self.spawn_subprocess()
473 #time.sleep(20) # test to simulate GUI not accepting connection
Kurt B. Kaiserb4179362002-07-26 00:06:42 +0000474 # Accept the connection from the Python execution server
Kurt B. Kaiseraf3eb872004-01-21 18:54:30 +0000475 self.rpcclt.listening_sock.settimeout(10)
476 try:
477 self.rpcclt.accept()
Christian Heimes03c8ddd2020-11-20 09:26:07 +0100478 except TimeoutError:
Kurt B. Kaiseraf3eb872004-01-21 18:54:30 +0000479 self.display_no_subprocess_error()
480 return None
Serhiy Storchaka39e70a42013-01-25 15:30:58 +0200481 self.rpcclt.register("console", self.tkconsole)
482 self.rpcclt.register("stdin", self.tkconsole.stdin)
Kurt B. Kaiser969de452002-06-12 03:28:57 +0000483 self.rpcclt.register("stdout", self.tkconsole.stdout)
484 self.rpcclt.register("stderr", self.tkconsole.stderr)
Chui Tey5d2af632002-05-26 13:36:41 +0000485 self.rpcclt.register("flist", self.tkconsole.flist)
Kurt B. Kaiser8cd0def2003-01-31 05:06:43 +0000486 self.rpcclt.register("linecache", linecache)
Kurt B. Kaiser9f366092003-06-02 01:50:19 +0000487 self.rpcclt.register("interp", self)
Terry Jan Reedyda4c4672012-01-31 02:26:32 -0500488 self.transfer_path(with_cwd=True)
Chui Tey5d2af632002-05-26 13:36:41 +0000489 self.poll_subprocess()
Kurt B. Kaiseraf3eb872004-01-21 18:54:30 +0000490 return self.rpcclt
Chui Tey5d2af632002-05-26 13:36:41 +0000491
Terry Jan Reedy231007f2015-07-31 22:34:37 -0400492 def restart_subprocess(self, with_cwd=False, filename=''):
Kurt B. Kaiser6f805942003-05-24 21:12:46 +0000493 if self.restarting:
Kurt B. Kaiseraf3eb872004-01-21 18:54:30 +0000494 return self.rpcclt
Kurt B. Kaiser6f805942003-05-24 21:12:46 +0000495 self.restarting = True
Kurt B. Kaiser63857a42002-09-05 02:31:20 +0000496 # close only the subprocess debugger
Kurt B. Kaiser45186c42002-10-23 04:48:08 +0000497 debug = self.getdebugger()
498 if debug:
Kurt B. Kaiser003091c2003-02-17 18:57:16 +0000499 try:
Kurt B. Kaiser9ec454e2003-05-12 02:33:47 +0000500 # Only close subprocess debugger, don't unregister gui_adap!
Terry Jan Reedy6fa5bdc2016-05-28 13:22:31 -0400501 debugger_r.close_subprocess_debugger(self.rpcclt)
Kurt B. Kaiser003091c2003-02-17 18:57:16 +0000502 except:
503 pass
504 # Kill subprocess, spawn a new one, accept connection.
Kurt B. Kaisera00050f2003-05-08 20:26:55 +0000505 self.rpcclt.close()
Ned Deilye5cad232011-08-02 18:47:13 -0700506 self.terminate_subprocess()
Kurt B. Kaiser7f38ec02003-05-15 03:19:42 +0000507 console = self.tkconsole
Kurt B. Kaiser6f805942003-05-24 21:12:46 +0000508 was_executing = console.executing
Kurt B. Kaiser7f38ec02003-05-15 03:19:42 +0000509 console.executing = False
Kurt B. Kaiser63857a42002-09-05 02:31:20 +0000510 self.spawn_subprocess()
Kurt B. Kaiseraf3eb872004-01-21 18:54:30 +0000511 try:
512 self.rpcclt.accept()
Christian Heimes03c8ddd2020-11-20 09:26:07 +0100513 except TimeoutError:
Kurt B. Kaiseraf3eb872004-01-21 18:54:30 +0000514 self.display_no_subprocess_error()
515 return None
Terry Jan Reedyda4c4672012-01-31 02:26:32 -0500516 self.transfer_path(with_cwd=with_cwd)
Roger Serwy6b7a5ae2013-04-03 00:42:24 -0500517 console.stop_readline()
Kurt B. Kaiser1061e722003-01-04 01:43:53 +0000518 # annotate restart in shell window and mark it
Kurt B. Kaiser4cc5ef52003-01-22 00:23:23 +0000519 console.text.delete("iomark", "end-1c")
Terry Jan Reedy38da8052019-09-06 13:54:45 -0400520 console.write('\n')
521 console.write(restart_line(console.width, filename))
Kurt B. Kaiser1061e722003-01-04 01:43:53 +0000522 console.text.mark_set("restart", "end-1c")
523 console.text.mark_gravity("restart", "left")
Terry Jan Reedy231007f2015-07-31 22:34:37 -0400524 if not filename:
525 console.showprompt()
Kurt B. Kaiser003091c2003-02-17 18:57:16 +0000526 # restart subprocess debugger
Kurt B. Kaiser45186c42002-10-23 04:48:08 +0000527 if debug:
Kurt B. Kaiser9ec454e2003-05-12 02:33:47 +0000528 # Restarted debugger connects to current instance of debug GUI
Terry Jan Reedy6fa5bdc2016-05-28 13:22:31 -0400529 debugger_r.restart_subprocess_debugger(self.rpcclt)
Kurt B. Kaiser45186c42002-10-23 04:48:08 +0000530 # reload remote debugger breakpoints for all PyShellEditWindows
531 debug.load_breakpoints()
Ned Deily7aff4512011-10-30 20:01:35 -0700532 self.compile.compiler.flags = self.original_compiler_flags
Kurt B. Kaiser6f805942003-05-24 21:12:46 +0000533 self.restarting = False
Kurt B. Kaiseraf3eb872004-01-21 18:54:30 +0000534 return self.rpcclt
Kurt B. Kaiser45186c42002-10-23 04:48:08 +0000535
Kurt B. Kaiser003091c2003-02-17 18:57:16 +0000536 def __request_interrupt(self):
Kurt B. Kaisera00050f2003-05-08 20:26:55 +0000537 self.rpcclt.remotecall("exec", "interrupt_the_server", (), {})
Kurt B. Kaiser003091c2003-02-17 18:57:16 +0000538
539 def interrupt_subprocess(self):
Kurt B. Kaisera00050f2003-05-08 20:26:55 +0000540 threading.Thread(target=self.__request_interrupt).start()
Kurt B. Kaiser003091c2003-02-17 18:57:16 +0000541
Kurt B. Kaisera00050f2003-05-08 20:26:55 +0000542 def kill_subprocess(self):
Roger Serwy036e8492013-06-11 22:13:17 -0500543 if self._afterid is not None:
544 self.tkconsole.text.after_cancel(self._afterid)
Kurt B. Kaiseraf3eb872004-01-21 18:54:30 +0000545 try:
Ned Deily55f87572011-11-05 22:36:44 -0700546 self.rpcclt.listening_sock.close()
547 except AttributeError: # no socket
548 pass
549 try:
Kurt B. Kaiseraf3eb872004-01-21 18:54:30 +0000550 self.rpcclt.close()
551 except AttributeError: # no socket
552 pass
Ned Deilye5cad232011-08-02 18:47:13 -0700553 self.terminate_subprocess()
Kurt B. Kaisera00050f2003-05-08 20:26:55 +0000554 self.tkconsole.executing = False
555 self.rpcclt = None
Kurt B. Kaiser11c53e22003-03-22 19:40:19 +0000556
Ned Deilye5cad232011-08-02 18:47:13 -0700557 def terminate_subprocess(self):
558 "Make sure subprocess is terminated"
559 try:
560 self.rpcsubproc.kill()
561 except OSError:
562 # process already terminated
563 return
564 else:
Kurt B. Kaisera00050f2003-05-08 20:26:55 +0000565 try:
Ned Deilye5cad232011-08-02 18:47:13 -0700566 self.rpcsubproc.wait()
Kurt B. Kaisera00050f2003-05-08 20:26:55 +0000567 except OSError:
Kurt B. Kaisera00050f2003-05-08 20:26:55 +0000568 return
Kurt B. Kaiser11c53e22003-03-22 19:40:19 +0000569
Terry Jan Reedyda4c4672012-01-31 02:26:32 -0500570 def transfer_path(self, with_cwd=False):
571 if with_cwd: # Issue 13506
572 path = [''] # include Current Working Directory
573 path.extend(sys.path)
574 else:
575 path = sys.path
Terry Jan Reedy4d82ade2012-01-31 02:57:29 -0500576
Kurt B. Kaiserf06eed02002-12-11 04:42:04 +0000577 self.runcommand("""if 1:
578 import sys as _sys
Walter Dörwald70a6b492004-02-12 17:35:32 +0000579 _sys.path = %r
Kurt B. Kaiserf06eed02002-12-11 04:42:04 +0000580 del _sys
Terry Jan Reedyda4c4672012-01-31 02:26:32 -0500581 \n""" % (path,))
Kurt B. Kaiserf06eed02002-12-11 04:42:04 +0000582
Chui Tey5d2af632002-05-26 13:36:41 +0000583 active_seq = None
584
585 def poll_subprocess(self):
586 clt = self.rpcclt
587 if clt is None:
588 return
Kurt B. Kaiser003091c2003-02-17 18:57:16 +0000589 try:
Kurt B. Kaisera00050f2003-05-08 20:26:55 +0000590 response = clt.pollresponse(self.active_seq, wait=0.05)
Andrew Svetlovf7a17b42012-12-25 16:47:37 +0200591 except (EOFError, OSError, KeyboardInterrupt):
Kurt B. Kaisera00050f2003-05-08 20:26:55 +0000592 # lost connection or subprocess terminated itself, restart
593 # [the KBI is from rpc.SocketIO.handle_EOF()]
Kurt B. Kaiser67fd0ea2003-05-24 20:59:15 +0000594 if self.tkconsole.closing:
595 return
Kurt B. Kaiser003091c2003-02-17 18:57:16 +0000596 response = None
Kurt B. Kaiser003091c2003-02-17 18:57:16 +0000597 self.restart_subprocess()
Chui Tey5d2af632002-05-26 13:36:41 +0000598 if response:
599 self.tkconsole.resetoutput()
600 self.active_seq = None
601 how, what = response
Kurt B. Kaiserbc286132003-01-25 21:33:40 +0000602 console = self.tkconsole.console
Chui Tey5d2af632002-05-26 13:36:41 +0000603 if how == "OK":
604 if what is not None:
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000605 print(repr(what), file=console)
Chui Tey5d2af632002-05-26 13:36:41 +0000606 elif how == "EXCEPTION":
Chui Tey5d2af632002-05-26 13:36:41 +0000607 if self.tkconsole.getvar("<<toggle-jit-stack-viewer>>"):
608 self.remote_stack_viewer()
609 elif how == "ERROR":
Terry Jan Reedy6fa5bdc2016-05-28 13:22:31 -0400610 errmsg = "pyshell.ModifiedInterpreter: Subprocess ERROR:\n"
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000611 print(errmsg, what, file=sys.__stderr__)
612 print(errmsg, what, file=console)
Kurt B. Kaiserbc286132003-01-25 21:33:40 +0000613 # we received a response to the currently active seq number:
Thomas Wouters00ee7ba2006-08-21 19:07:27 +0000614 try:
615 self.tkconsole.endexecuting()
616 except AttributeError: # shell may have closed
617 pass
Kurt B. Kaiser88957d82003-05-19 23:11:51 +0000618 # Reschedule myself
619 if not self.tkconsole.closing:
Roger Serwy036e8492013-06-11 22:13:17 -0500620 self._afterid = self.tkconsole.text.after(
621 self.tkconsole.pollinterval, self.poll_subprocess)
Chui Tey5d2af632002-05-26 13:36:41 +0000622
Kurt B. Kaiser45186c42002-10-23 04:48:08 +0000623 debugger = None
624
625 def setdebugger(self, debugger):
626 self.debugger = debugger
627
628 def getdebugger(self):
629 return self.debugger
630
Kurt B. Kaiser9f366092003-06-02 01:50:19 +0000631 def open_remote_stack_viewer(self):
632 """Initiate the remote stack viewer from a separate thread.
633
634 This method is called from the subprocess, and by returning from this
635 method we allow the subprocess to unblock. After a bit the shell
636 requests the subprocess to open the remote stack viewer which returns a
Ezio Melotti60861182010-07-23 16:48:22 +0000637 static object looking at the last exception. It is queried through
Kurt B. Kaiser9f366092003-06-02 01:50:19 +0000638 the RPC mechanism.
639
640 """
641 self.tkconsole.text.after(300, self.remote_stack_viewer)
642 return
643
Chui Tey5d2af632002-05-26 13:36:41 +0000644 def remote_stack_viewer(self):
Terry Jan Reedy6fa5bdc2016-05-28 13:22:31 -0400645 from idlelib import debugobj_r
Kurt B. Kaisera00050f2003-05-08 20:26:55 +0000646 oid = self.rpcclt.remotequeue("exec", "stackviewer", ("flist",), {})
Chui Tey5d2af632002-05-26 13:36:41 +0000647 if oid is None:
648 self.tkconsole.root.bell()
649 return
Terry Jan Reedy6fa5bdc2016-05-28 13:22:31 -0400650 item = debugobj_r.StubObjectTreeItem(self.rpcclt, oid)
651 from idlelib.tree import ScrolledCanvas, TreeNode
Chui Tey5d2af632002-05-26 13:36:41 +0000652 top = Toplevel(self.tkconsole.root)
Terry Jan Reedyd0c0f002015-11-12 15:02:57 -0500653 theme = idleConf.CurrentTheme()
Kurt B. Kaiser73360a32004-03-08 18:15:31 +0000654 background = idleConf.GetHighlight(theme, 'normal')['background']
655 sc = ScrolledCanvas(top, bg=background, highlightthickness=0)
Chui Tey5d2af632002-05-26 13:36:41 +0000656 sc.frame.pack(expand=1, fill="both")
657 node = TreeNode(sc.canvas, None, item)
658 node.expand()
659 # XXX Should GC the remote tree when closing the window
660
David Scherer7aced172000-08-15 01:13:23 +0000661 gid = 0
662
663 def execsource(self, source):
Kurt B. Kaiser83118c62002-06-24 17:03:37 +0000664 "Like runsource() but assumes complete exec source"
David Scherer7aced172000-08-15 01:13:23 +0000665 filename = self.stuffsource(source)
666 self.execfile(filename, source)
667
668 def execfile(self, filename, source=None):
Kurt B. Kaiser83118c62002-06-24 17:03:37 +0000669 "Execute an existing file"
David Scherer7aced172000-08-15 01:13:23 +0000670 if source is None:
Victor Stinner979482a2011-09-02 01:00:40 +0200671 with tokenize.open(filename) as fp:
Victor Stinner85c67722011-09-02 00:57:04 +0200672 source = fp.read()
Terry Jan Reedy22c82be2018-03-05 11:02:46 -0500673 if use_subprocess:
674 source = (f"__file__ = r'''{os.path.abspath(filename)}'''\n"
675 + source + "\ndel __file__")
David Scherer7aced172000-08-15 01:13:23 +0000676 try:
677 code = compile(source, filename, "exec")
678 except (OverflowError, SyntaxError):
679 self.tkconsole.resetoutput()
Terry Jan Reedy81b062f2014-09-19 22:38:41 -0400680 print('*** Error in script or command!\n'
681 'Traceback (most recent call last):',
682 file=self.tkconsole.stderr)
David Scherer7aced172000-08-15 01:13:23 +0000683 InteractiveInterpreter.showsyntaxerror(self, filename)
Kurt B. Kaiser6e44cc22002-11-30 06:18:00 +0000684 self.tkconsole.showprompt()
David Scherer7aced172000-08-15 01:13:23 +0000685 else:
686 self.runcode(code)
687
688 def runsource(self, source):
Kurt B. Kaiser83118c62002-06-24 17:03:37 +0000689 "Extend base class method: Stuff the source in the line cache first"
David Scherer7aced172000-08-15 01:13:23 +0000690 filename = self.stuffsource(source)
Martin v. Löwis98ff8982007-08-13 06:03:15 +0000691 # at the moment, InteractiveInterpreter expects str
692 assert isinstance(source, str)
Terry Jan Reedy1039f392019-08-26 02:27:56 -0400693 # InteractiveInterpreter.runsource() calls its runcode() method,
694 # which is overridden (see below)
695 return InteractiveInterpreter.runsource(self, source, filename)
David Scherer7aced172000-08-15 01:13:23 +0000696
697 def stuffsource(self, source):
Kurt B. Kaiser83118c62002-06-24 17:03:37 +0000698 "Stuff source in the filename cache"
David Scherer7aced172000-08-15 01:13:23 +0000699 filename = "<pyshell#%d>" % self.gid
700 self.gid = self.gid + 1
Kurt B. Kaiser837d15c2002-09-18 02:29:59 +0000701 lines = source.split("\n")
David Scherer7aced172000-08-15 01:13:23 +0000702 linecache.cache[filename] = len(source)+1, 0, lines, filename
703 return filename
Kurt B. Kaiser6655e4b2002-12-31 16:03:23 +0000704
Kurt B. Kaiser11659ad2003-05-15 23:23:21 +0000705 def prepend_syspath(self, filename):
706 "Prepend sys.path with file's directory if not already included"
707 self.runcommand("""if 1:
Walter Dörwald70a6b492004-02-12 17:35:32 +0000708 _filename = %r
Kurt B. Kaiser11659ad2003-05-15 23:23:21 +0000709 import sys as _sys
710 from os.path import dirname as _dirname
711 _dir = _dirname(_filename)
712 if not _dir in _sys.path:
713 _sys.path.insert(0, _dir)
714 del _filename, _sys, _dirname, _dir
Walter Dörwald70a6b492004-02-12 17:35:32 +0000715 \n""" % (filename,))
Kurt B. Kaiser11659ad2003-05-15 23:23:21 +0000716
David Scherer7aced172000-08-15 01:13:23 +0000717 def showsyntaxerror(self, filename=None):
Guido van Rossum33d26892007-08-05 15:29:28 +0000718 """Override Interactive Interpreter method: Use Colorizing
Kurt B. Kaiser83118c62002-06-24 17:03:37 +0000719
720 Color the offending position instead of printing it and pointing at it
721 with a caret.
722
723 """
Guido van Rossum33d26892007-08-05 15:29:28 +0000724 tkconsole = self.tkconsole
725 text = tkconsole.text
726 text.tag_remove("ERROR", "1.0", "end")
David Scherer7aced172000-08-15 01:13:23 +0000727 type, value, tb = sys.exc_info()
Ned Deily79746422011-09-14 14:49:14 -0700728 msg = getattr(value, 'msg', '') or value or "<no detail available>"
729 lineno = getattr(value, 'lineno', '') or 1
730 offset = getattr(value, 'offset', '') or 0
Guido van Rossum33d26892007-08-05 15:29:28 +0000731 if offset == 0:
732 lineno += 1 #mark end of offending line
733 if lineno == 1:
734 pos = "iomark + %d chars" % (offset-1)
David Scherer7aced172000-08-15 01:13:23 +0000735 else:
Guido van Rossum33d26892007-08-05 15:29:28 +0000736 pos = "iomark linestart + %d lines + %d chars" % \
737 (lineno-1, offset-1)
738 tkconsole.colorize_syntax_error(text, pos)
739 tkconsole.resetoutput()
740 self.write("SyntaxError: %s\n" % msg)
741 tkconsole.showprompt()
David Scherer7aced172000-08-15 01:13:23 +0000742
743 def showtraceback(self):
Kurt B. Kaiser83118c62002-06-24 17:03:37 +0000744 "Extend base class method to reset output properly"
David Scherer7aced172000-08-15 01:13:23 +0000745 self.tkconsole.resetoutput()
746 self.checklinecache()
747 InteractiveInterpreter.showtraceback(self)
Chui Tey5d2af632002-05-26 13:36:41 +0000748 if self.tkconsole.getvar("<<toggle-jit-stack-viewer>>"):
749 self.tkconsole.open_stack_viewer()
David Scherer7aced172000-08-15 01:13:23 +0000750
751 def checklinecache(self):
752 c = linecache.cache
Guido van Rossum36e0a922007-07-20 04:05:57 +0000753 for key in list(c.keys()):
David Scherer7aced172000-08-15 01:13:23 +0000754 if key[:1] + key[-1:] != "<>":
755 del c[key]
756
Chui Tey5d2af632002-05-26 13:36:41 +0000757 def runcommand(self, code):
Kurt B. Kaiser83118c62002-06-24 17:03:37 +0000758 "Run the code without invoking the debugger"
Chui Tey5d2af632002-05-26 13:36:41 +0000759 # The code better not raise an exception!
760 if self.tkconsole.executing:
Neal Norwitzf4c4f112002-11-30 18:49:10 +0000761 self.display_executing_dialog()
Chui Tey5d2af632002-05-26 13:36:41 +0000762 return 0
763 if self.rpcclt:
Kurt B. Kaisera00050f2003-05-08 20:26:55 +0000764 self.rpcclt.remotequeue("exec", "runcode", (code,), {})
Chui Tey5d2af632002-05-26 13:36:41 +0000765 else:
Georg Brandl7cae87c2006-09-06 06:51:57 +0000766 exec(code, self.locals)
Chui Tey5d2af632002-05-26 13:36:41 +0000767 return 1
768
David Scherer7aced172000-08-15 01:13:23 +0000769 def runcode(self, code):
Kurt B. Kaiser83118c62002-06-24 17:03:37 +0000770 "Override base class method"
Chui Tey5d2af632002-05-26 13:36:41 +0000771 if self.tkconsole.executing:
Terry Jan Reedye41bfd12020-11-30 12:09:43 -0500772 self.restart_subprocess()
Chui Tey5d2af632002-05-26 13:36:41 +0000773 self.checklinecache()
David Scherer7aced172000-08-15 01:13:23 +0000774 debugger = self.debugger
775 try:
Kurt B. Kaiser7f38ec02003-05-15 03:19:42 +0000776 self.tkconsole.beginexecuting()
Thomas Wouterscf297e42007-02-23 15:07:44 +0000777 if not debugger and self.rpcclt is not None:
778 self.active_seq = self.rpcclt.asyncqueue("exec", "runcode",
779 (code,), {})
780 elif debugger:
781 debugger.run(code, self.locals)
782 else:
783 exec(code, self.locals)
784 except SystemExit:
785 if not self.tkconsole.closing:
Terry Jan Reedy879986d2021-01-25 06:33:18 -0500786 if messagebox.askyesno(
Thomas Wouterscf297e42007-02-23 15:07:44 +0000787 "Exit?",
788 "Do you want to exit altogether?",
789 default="yes",
Terry Jan Reedy3be2e542015-09-25 22:22:55 -0400790 parent=self.tkconsole.text):
Thomas Wouters00ee7ba2006-08-21 19:07:27 +0000791 raise
Thomas Wouterscf297e42007-02-23 15:07:44 +0000792 else:
Guido van Rossum5dc0d972007-02-25 22:37:36 +0000793 self.showtraceback()
Thomas Wouterscf297e42007-02-23 15:07:44 +0000794 else:
795 raise
796 except:
797 if use_subprocess:
798 print("IDLE internal error in runcode()",
799 file=self.tkconsole.stderr)
David Scherer7aced172000-08-15 01:13:23 +0000800 self.showtraceback()
Thomas Wouterscf297e42007-02-23 15:07:44 +0000801 self.tkconsole.endexecuting()
802 else:
803 if self.tkconsole.canceled:
804 self.tkconsole.canceled = False
805 print("KeyboardInterrupt", file=self.tkconsole.stderr)
806 else:
807 self.showtraceback()
Kurt B. Kaiser7f38ec02003-05-15 03:19:42 +0000808 finally:
809 if not use_subprocess:
Thomas Wouters00ee7ba2006-08-21 19:07:27 +0000810 try:
811 self.tkconsole.endexecuting()
812 except AttributeError: # shell may have closed
813 pass
David Scherer7aced172000-08-15 01:13:23 +0000814
815 def write(self, s):
Kurt B. Kaiser83118c62002-06-24 17:03:37 +0000816 "Override base class method"
Martin v. Löwis30d5e6c2012-07-25 11:32:26 +0200817 return self.tkconsole.stderr.write(s)
David Scherer7aced172000-08-15 01:13:23 +0000818
Kurt B. Kaiseraf3eb872004-01-21 18:54:30 +0000819 def display_port_binding_error(self):
Terry Jan Reedy879986d2021-01-25 06:33:18 -0500820 messagebox.showerror(
Kurt B. Kaiseraf3eb872004-01-21 18:54:30 +0000821 "Port Binding Error",
Kurt B. Kaisere866c812009-04-04 21:07:39 +0000822 "IDLE can't bind to a TCP/IP port, which is necessary to "
823 "communicate with its Python execution server. This might be "
824 "because no networking is installed on this computer. "
825 "Run IDLE with the -n command line switch to start without a "
826 "subprocess and refer to Help/IDLE Help 'Running without a "
827 "subprocess' for further details.",
Terry Jan Reedy3be2e542015-09-25 22:22:55 -0400828 parent=self.tkconsole.text)
Kurt B. Kaiseraf3eb872004-01-21 18:54:30 +0000829
830 def display_no_subprocess_error(self):
Terry Jan Reedy879986d2021-01-25 06:33:18 -0500831 messagebox.showerror(
Terry Jan Reedy8fac1222019-06-17 17:23:28 -0400832 "Subprocess Connection Error",
833 "IDLE's subprocess didn't make connection.\n"
834 "See the 'Startup failure' section of the IDLE doc, online at\n"
835 "https://docs.python.org/3/library/idle.html#startup-failure",
Terry Jan Reedy3be2e542015-09-25 22:22:55 -0400836 parent=self.tkconsole.text)
Kurt B. Kaiseraf3eb872004-01-21 18:54:30 +0000837
838 def display_executing_dialog(self):
Terry Jan Reedy879986d2021-01-25 06:33:18 -0500839 messagebox.showerror(
Kurt B. Kaiseraf3eb872004-01-21 18:54:30 +0000840 "Already executing",
841 "The Python Shell window is already executing a command; "
842 "please wait until it is finished.",
Terry Jan Reedy3be2e542015-09-25 22:22:55 -0400843 parent=self.tkconsole.text)
Kurt B. Kaiseraf3eb872004-01-21 18:54:30 +0000844
845
David Scherer7aced172000-08-15 01:13:23 +0000846class PyShell(OutputWindow):
Tal Einat15d38612021-04-29 01:27:55 +0300847 from idlelib.squeezer import Squeezer
David Scherer7aced172000-08-15 01:13:23 +0000848
Terry Jan Reedy05cc8812020-09-24 15:30:09 -0400849 shell_title = "IDLE Shell " + python_version()
David Scherer7aced172000-08-15 01:13:23 +0000850
851 # Override classes
852 ColorDelegator = ModifiedColorDelegator
853 UndoDelegator = ModifiedUndoDelegator
854
Kurt B. Kaiserf06eed02002-12-11 04:42:04 +0000855 # Override menus
Kurt B. Kaiserdc1e7092002-07-11 04:33:41 +0000856 menu_specs = [
857 ("file", "_File"),
858 ("edit", "_Edit"),
Kurt B. Kaiser4cc5ef52003-01-22 00:23:23 +0000859 ("debug", "_Debug"),
Kurt B. Kaiser1061e722003-01-04 01:43:53 +0000860 ("options", "_Options"),
Terry Jan Reedy33c74202018-06-20 22:49:55 -0400861 ("window", "_Window"),
Kurt B. Kaiserdc1e7092002-07-11 04:33:41 +0000862 ("help", "_Help"),
863 ]
David Scherer7aced172000-08-15 01:13:23 +0000864
Tal Einat604e7b92018-09-25 15:10:14 +0300865 # Extend right-click context menu
866 rmenu_specs = OutputWindow.rmenu_specs + [
867 ("Squeeze", "<<squeeze-current-text>>"),
868 ]
Tal Einatb43cc312021-05-03 05:27:38 +0300869 _idx = 1 + len(list(itertools.takewhile(
870 lambda rmenu_item: rmenu_item[0] != "Copy", rmenu_specs)
871 ))
872 rmenu_specs.insert(_idx, ("Copy with prompts",
873 "<<copy-with-prompts>>",
874 "rmenu_check_copy"))
875 del _idx
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000876
Tal Einat7123ea02019-07-23 15:22:11 +0300877 allow_line_numbers = False
Tal Einat15d38612021-04-29 01:27:55 +0300878 user_input_insert_tags = "stdin"
Tal Einat7123ea02019-07-23 15:22:11 +0300879
David Scherer7aced172000-08-15 01:13:23 +0000880 # New classes
Terry Jan Reedy6fa5bdc2016-05-28 13:22:31 -0400881 from idlelib.history import History
Tal Einat15d38612021-04-29 01:27:55 +0300882 from idlelib.sidebar import ShellSidebar
David Scherer7aced172000-08-15 01:13:23 +0000883
884 def __init__(self, flist=None):
Kurt B. Kaiser8f570a72003-05-15 18:52:51 +0000885 if use_subprocess:
Kurt B. Kaiser67fd0ea2003-05-24 20:59:15 +0000886 ms = self.menu_specs
887 if ms[2][0] != "shell":
Thomas Wouters00ee7ba2006-08-21 19:07:27 +0000888 ms.insert(2, ("shell", "She_ll"))
David Scherer7aced172000-08-15 01:13:23 +0000889 self.interp = ModifiedInterpreter(self)
890 if flist is None:
891 root = Tk()
892 fixwordbreaks(root)
893 root.withdraw()
894 flist = PyShellFileList(root)
Terry Jan Reedye86172d2017-10-27 20:26:12 -0400895
Tal Einat15d38612021-04-29 01:27:55 +0300896 self.shell_sidebar = None # initialized below
897
David Scherer7aced172000-08-15 01:13:23 +0000898 OutputWindow.__init__(self, flist, None, None)
Terry Jan Reedye86172d2017-10-27 20:26:12 -0400899
Terry Jan Reedy8ec2f0d2021-04-29 06:48:18 -0400900 self.usetabs = False
Kurt B. Kaiser6af44982005-01-19 00:22:59 +0000901 # indentwidth must be 8 when using tabs. See note in EditorWindow:
Terry Jan Reedy8ec2f0d2021-04-29 06:48:18 -0400902 self.indentwidth = 4
Cheryl Sabella6bdc4de2019-06-02 14:56:47 -0400903
Terry Jan Reedy8ec2f0d2021-04-29 06:48:18 -0400904 self.sys_ps1 = sys.ps1 if hasattr(sys, 'ps1') else '>>>\n'
Terry Jan Reedye86172d2017-10-27 20:26:12 -0400905 self.prompt_last_line = self.sys_ps1.split('\n')[-1]
906 self.prompt = self.sys_ps1 # Changes when debug active
907
David Scherer7aced172000-08-15 01:13:23 +0000908 text = self.text
909 text.configure(wrap="char")
910 text.bind("<<newline-and-indent>>", self.enter_callback)
911 text.bind("<<plain-newline-and-indent>>", self.linefeed_callback)
912 text.bind("<<interrupt-execution>>", self.cancel_callback)
David Scherer7aced172000-08-15 01:13:23 +0000913 text.bind("<<end-of-file>>", self.eof_callback)
914 text.bind("<<open-stack-viewer>>", self.open_stack_viewer)
Kurt B. Kaiser57bfe5d2003-05-10 00:09:52 +0000915 text.bind("<<toggle-debugger>>", self.toggle_debugger)
David Scherer7aced172000-08-15 01:13:23 +0000916 text.bind("<<toggle-jit-stack-viewer>>", self.toggle_jit_stack_viewer)
Tal Einatb43cc312021-05-03 05:27:38 +0300917 text.bind("<<copy-with-prompts>>", self.copy_with_prompts_callback)
Kurt B. Kaiser8f570a72003-05-15 18:52:51 +0000918 if use_subprocess:
919 text.bind("<<view-restart>>", self.view_restart_mark)
920 text.bind("<<restart-shell>>", self.restart_shell)
Tal Einat15d38612021-04-29 01:27:55 +0300921 self.squeezer = self.Squeezer(self)
Tal Einat39a33e92019-01-13 17:01:50 +0200922 text.bind("<<squeeze-current-text>>",
Tal Einat15d38612021-04-29 01:27:55 +0300923 self.squeeze_current_text_event)
Terry Jan Reedye86172d2017-10-27 20:26:12 -0400924
David Scherer7aced172000-08-15 01:13:23 +0000925 self.save_stdout = sys.stdout
926 self.save_stderr = sys.stderr
927 self.save_stdin = sys.stdin
Terry Jan Reedy6fa5bdc2016-05-28 13:22:31 -0400928 from idlelib import iomenu
Serhiy Storchakab690a272019-10-08 14:32:25 +0300929 self.stdin = StdInputFile(self, "stdin",
930 iomenu.encoding, iomenu.errors)
931 self.stdout = StdOutputFile(self, "stdout",
932 iomenu.encoding, iomenu.errors)
933 self.stderr = StdOutputFile(self, "stderr",
934 iomenu.encoding, "backslashreplace")
935 self.console = StdOutputFile(self, "console",
936 iomenu.encoding, iomenu.errors)
Chui Tey5d2af632002-05-26 13:36:41 +0000937 if not use_subprocess:
938 sys.stdout = self.stdout
Kurt B. Kaiser7cec2522007-08-31 04:15:13 +0000939 sys.stderr = self.stderr
Martin v. Löwisc882b7c2012-07-25 10:47:20 +0200940 sys.stdin = self.stdin
Kurt B. Kaiserf609a342007-12-28 03:57:56 +0000941 try:
942 # page help() text to shell.
943 import pydoc # import must be done here to capture i/o rebinding.
csabella42bc8be2017-06-29 18:42:17 -0400944 # XXX KBK 27Dec07 use text viewer someday, but must work w/o subproc
Kurt B. Kaiserf609a342007-12-28 03:57:56 +0000945 pydoc.pager = pydoc.plainpager
946 except:
947 sys.stderr = sys.__stderr__
948 raise
Kurt B. Kaiser5afa1df2002-10-10 08:25:24 +0000949 #
David Scherer7aced172000-08-15 01:13:23 +0000950 self.history = self.History(self.text)
Kurt B. Kaiser5afa1df2002-10-10 08:25:24 +0000951 #
Kurt B. Kaiser88957d82003-05-19 23:11:51 +0000952 self.pollinterval = 50 # millisec
Chui Tey5d2af632002-05-26 13:36:41 +0000953
Tal Einat15d38612021-04-29 01:27:55 +0300954 self.shell_sidebar = self.ShellSidebar(self)
955
956 # Insert UserInputTaggingDelegator at the top of the percolator,
957 # but make calls to text.insert() skip it. This causes only insert
958 # events generated in Tcl/Tk to go through this delegator.
959 self.text.insert = self.per.top.insert
960 self.per.insertfilter(UserInputTaggingDelegator())
961
962 def ResetFont(self):
963 super().ResetFont()
964
965 if self.shell_sidebar is not None:
966 self.shell_sidebar.update_font()
967
968 def ResetColorizer(self):
969 super().ResetColorizer()
970
971 theme = idleConf.CurrentTheme()
972 tag_colors = {
973 "stdin": {'background': None, 'foreground': None},
974 "stdout": idleConf.GetHighlight(theme, "stdout"),
975 "stderr": idleConf.GetHighlight(theme, "stderr"),
976 "console": idleConf.GetHighlight(theme, "normal"),
977 }
978 for tag, tag_colors_config in tag_colors.items():
979 self.text.tag_configure(tag, **tag_colors_config)
980
981 if self.shell_sidebar is not None:
982 self.shell_sidebar.update_colors()
983
984 def replace_event(self, event):
985 replace.replace(self.text, insert_tags="stdin")
986 return "break"
987
Kurt B. Kaiser4d5bc602004-06-06 01:29:22 +0000988 def get_standard_extension_names(self):
989 return idleConf.GetExtensions(shell_only=True)
990
Tal Einatb43cc312021-05-03 05:27:38 +0300991 def copy_with_prompts_callback(self, event=None):
992 """Copy selected lines to the clipboard, with prompts.
993
994 This makes the copied text useful for doc-tests and interactive
995 shell code examples.
996
997 This always copies entire lines, even if only part of the first
998 and/or last lines is selected.
999 """
1000 text = self.text
1001
1002 selection_indexes = (
1003 self.text.index("sel.first linestart"),
1004 self.text.index("sel.last +1line linestart"),
1005 )
1006 if selection_indexes[0] is None:
1007 # There is no selection, so do nothing.
1008 return
1009
1010 selected_text = self.text.get(*selection_indexes)
1011 selection_lineno_range = range(
1012 int(float(selection_indexes[0])),
1013 int(float(selection_indexes[1]))
1014 )
1015 prompts = [
1016 self.shell_sidebar.line_prompts.get(lineno)
1017 for lineno in selection_lineno_range
1018 ]
1019 selected_text_with_prompts = "\n".join(
1020 line if prompt is None else f"{prompt} {line}"
1021 for prompt, line in zip(prompts, selected_text.splitlines())
1022 ) + "\n"
1023
1024 text.clipboard_clear()
1025 text.clipboard_append(selected_text_with_prompts)
1026
Kurt B. Kaiser003091c2003-02-17 18:57:16 +00001027 reading = False
1028 executing = False
1029 canceled = False
1030 endoffile = False
1031 closing = False
Roger Serwy6b7a5ae2013-04-03 00:42:24 -05001032 _stop_readline_flag = False
David Scherer7aced172000-08-15 01:13:23 +00001033
Kurt B. Kaiser49a5fe12004-07-04 01:25:56 +00001034 def set_warning_stream(self, stream):
Skip Montanarod9161422004-07-06 21:53:27 +00001035 global warning_stream
1036 warning_stream = stream
Kurt B. Kaiser49a5fe12004-07-04 01:25:56 +00001037
1038 def get_warning_stream(self):
1039 return warning_stream
1040
David Scherer7aced172000-08-15 01:13:23 +00001041 def toggle_debugger(self, event=None):
1042 if self.executing:
Terry Jan Reedy879986d2021-01-25 06:33:18 -05001043 messagebox.showerror("Don't debug now",
David Scherer7aced172000-08-15 01:13:23 +00001044 "You can only toggle the debugger when idle",
Terry Jan Reedy3be2e542015-09-25 22:22:55 -04001045 parent=self.text)
David Scherer7aced172000-08-15 01:13:23 +00001046 self.set_debugger_indicator()
1047 return "break"
1048 else:
1049 db = self.interp.getdebugger()
1050 if db:
1051 self.close_debugger()
1052 else:
1053 self.open_debugger()
1054
1055 def set_debugger_indicator(self):
1056 db = self.interp.getdebugger()
1057 self.setvar("<<toggle-debugger>>", not not db)
1058
Kurt B. Kaiser1061e722003-01-04 01:43:53 +00001059 def toggle_jit_stack_viewer(self, event=None):
David Scherer7aced172000-08-15 01:13:23 +00001060 pass # All we need is the variable
1061
1062 def close_debugger(self):
1063 db = self.interp.getdebugger()
1064 if db:
1065 self.interp.setdebugger(None)
1066 db.close()
Kurt B. Kaiserffd3a422002-06-26 02:32:09 +00001067 if self.interp.rpcclt:
Terry Jan Reedy6fa5bdc2016-05-28 13:22:31 -04001068 debugger_r.close_remote_debugger(self.interp.rpcclt)
David Scherer7aced172000-08-15 01:13:23 +00001069 self.resetoutput()
1070 self.console.write("[DEBUG OFF]\n")
Terry Jan Reedye86172d2017-10-27 20:26:12 -04001071 self.prompt = self.sys_ps1
David Scherer7aced172000-08-15 01:13:23 +00001072 self.showprompt()
1073 self.set_debugger_indicator()
1074
1075 def open_debugger(self):
Chui Tey5d2af632002-05-26 13:36:41 +00001076 if self.interp.rpcclt:
Terry Jan Reedy6fa5bdc2016-05-28 13:22:31 -04001077 dbg_gui = debugger_r.start_remote_debugger(self.interp.rpcclt,
Kurt B. Kaiser7f38ec02003-05-15 03:19:42 +00001078 self)
1079 else:
Terry Jan Reedy6fa5bdc2016-05-28 13:22:31 -04001080 dbg_gui = debugger.Debugger(self)
Kurt B. Kaiser7f38ec02003-05-15 03:19:42 +00001081 self.interp.setdebugger(dbg_gui)
1082 dbg_gui.load_breakpoints()
Terry Jan Reedye86172d2017-10-27 20:26:12 -04001083 self.prompt = "[DEBUG ON]\n" + self.sys_ps1
Chui Tey5d2af632002-05-26 13:36:41 +00001084 self.showprompt()
1085 self.set_debugger_indicator()
1086
Zackery Spytz23a567c2021-01-28 16:13:22 -07001087 def debug_menu_postcommand(self):
1088 state = 'disabled' if self.executing else 'normal'
1089 self.update_menu_state('debug', '*tack*iewer', state)
1090
David Scherer7aced172000-08-15 01:13:23 +00001091 def beginexecuting(self):
Kurt B. Kaiserffd3a422002-06-26 02:32:09 +00001092 "Helper for ModifiedInterpreter"
David Scherer7aced172000-08-15 01:13:23 +00001093 self.resetoutput()
Terry Jan Reedy34a49aa2020-03-28 12:51:21 -04001094 self.executing = True
David Scherer7aced172000-08-15 01:13:23 +00001095
1096 def endexecuting(self):
Kurt B. Kaiser83118c62002-06-24 17:03:37 +00001097 "Helper for ModifiedInterpreter"
Terry Jan Reedy34a49aa2020-03-28 12:51:21 -04001098 self.executing = False
1099 self.canceled = False
Chui Tey5d2af632002-05-26 13:36:41 +00001100 self.showprompt()
David Scherer7aced172000-08-15 01:13:23 +00001101
1102 def close(self):
Kurt B. Kaiser83118c62002-06-24 17:03:37 +00001103 "Extend EditorWindow.close()"
David Scherer7aced172000-08-15 01:13:23 +00001104 if self.executing:
Terry Jan Reedy879986d2021-01-25 06:33:18 -05001105 response = messagebox.askokcancel(
David Scherer7aced172000-08-15 01:13:23 +00001106 "Kill?",
Terry Jan Reedy4379d152015-11-20 12:21:48 -05001107 "Your program is still running!\n Do you want to kill it?",
David Scherer7aced172000-08-15 01:13:23 +00001108 default="ok",
Kurt B. Kaiser7f38ec02003-05-15 03:19:42 +00001109 parent=self.text)
Benjamin Peterson2a691a82008-03-31 01:51:45 +00001110 if response is False:
David Scherer7aced172000-08-15 01:13:23 +00001111 return "cancel"
Roger Serwy6b7a5ae2013-04-03 00:42:24 -05001112 self.stop_readline()
Kurt B. Kaiser5c3df352004-12-23 04:20:59 +00001113 self.canceled = True
Kurt B. Kaiser67fd0ea2003-05-24 20:59:15 +00001114 self.closing = True
Kurt B. Kaiser88957d82003-05-19 23:11:51 +00001115 return EditorWindow.close(self)
David Scherer7aced172000-08-15 01:13:23 +00001116
1117 def _close(self):
Kurt B. Kaiser83118c62002-06-24 17:03:37 +00001118 "Extend EditorWindow._close(), shut down debugger and execution server"
David Scherer7aced172000-08-15 01:13:23 +00001119 self.close_debugger()
Kurt B. Kaiser7f38ec02003-05-15 03:19:42 +00001120 if use_subprocess:
1121 self.interp.kill_subprocess()
David Scherer7aced172000-08-15 01:13:23 +00001122 # Restore std streams
1123 sys.stdout = self.save_stdout
1124 sys.stderr = self.save_stderr
1125 sys.stdin = self.save_stdin
1126 # Break cycles
1127 self.interp = None
1128 self.console = None
David Scherer7aced172000-08-15 01:13:23 +00001129 self.flist.pyshell = None
1130 self.history = None
Kurt B. Kaiser83118c62002-06-24 17:03:37 +00001131 EditorWindow._close(self)
David Scherer7aced172000-08-15 01:13:23 +00001132
1133 def ispythonsource(self, filename):
Kurt B. Kaiser83118c62002-06-24 17:03:37 +00001134 "Override EditorWindow method: never remove the colorizer"
Kurt B. Kaiser837d15c2002-09-18 02:29:59 +00001135 return True
David Scherer7aced172000-08-15 01:13:23 +00001136
1137 def short_title(self):
1138 return self.shell_title
1139
Kurt B. Kaiser94bd7742001-07-14 00:13:28 +00001140 COPYRIGHT = \
Stéphane Wirtel9d49f852018-06-16 23:20:56 +02001141 'Type "help", "copyright", "credits" or "license()" for more information.'
Kurt B. Kaiser94bd7742001-07-14 00:13:28 +00001142
David Scherer7aced172000-08-15 01:13:23 +00001143 def begin(self):
Kurt B. Kaiser7378df72007-08-11 04:20:51 +00001144 self.text.mark_set("iomark", "insert")
David Scherer7aced172000-08-15 01:13:23 +00001145 self.resetoutput()
Kurt B. Kaiser7f38ec02003-05-15 03:19:42 +00001146 if use_subprocess:
1147 nosub = ''
Kurt B. Kaiseraf3eb872004-01-21 18:54:30 +00001148 client = self.interp.start_subprocess()
1149 if not client:
1150 self.close()
Kurt B. Kaiser76637292004-01-21 22:10:01 +00001151 return False
Kurt B. Kaiser7f38ec02003-05-15 03:19:42 +00001152 else:
Andrew Svetlov9b1e59a2012-10-05 22:12:44 +03001153 nosub = ("==== No Subprocess ====\n\n" +
Andrew Svetlov64478ac2012-10-05 22:16:55 +03001154 "WARNING: Running IDLE without a Subprocess is deprecated\n" +
Andrew Svetlov9b1e59a2012-10-05 22:12:44 +03001155 "and will be removed in a later version. See Help/IDLE Help\n" +
1156 "for details.\n\n")
Andrew Svetlovcd49d532012-03-25 11:43:02 +03001157 sys.displayhook = rpc.displayhook
1158
Raymond Hettingera2a8e8b2009-01-27 00:28:36 +00001159 self.write("Python %s on %s\n%s\n%s" %
1160 (sys.version, sys.platform, self.COPYRIGHT, nosub))
Terry Jan Reedyb41eada2015-09-07 01:58:13 -04001161 self.text.focus_force()
David Scherer7aced172000-08-15 01:13:23 +00001162 self.showprompt()
Serhiy Storchaka3d569fd2020-12-19 12:17:08 +02001163 # User code should use separate default Tk root window
Georg Brandl14fc4272008-05-17 18:39:55 +00001164 import tkinter
Serhiy Storchaka3d569fd2020-12-19 12:17:08 +02001165 tkinter._support_default_root = True
1166 tkinter._default_root = None
Kurt B. Kaiser76637292004-01-21 22:10:01 +00001167 return True
David Scherer7aced172000-08-15 01:13:23 +00001168
Roger Serwy6b7a5ae2013-04-03 00:42:24 -05001169 def stop_readline(self):
1170 if not self.reading: # no nested mainloop to exit.
1171 return
1172 self._stop_readline_flag = True
1173 self.top.quit()
1174
David Scherer7aced172000-08-15 01:13:23 +00001175 def readline(self):
1176 save = self.reading
1177 try:
Terry Jan Reedy34a49aa2020-03-28 12:51:21 -04001178 self.reading = True
Kurt B. Kaiser5c3df352004-12-23 04:20:59 +00001179 self.top.mainloop() # nested mainloop()
David Scherer7aced172000-08-15 01:13:23 +00001180 finally:
1181 self.reading = save
Roger Serwy6b7a5ae2013-04-03 00:42:24 -05001182 if self._stop_readline_flag:
1183 self._stop_readline_flag = False
1184 return ""
David Scherer7aced172000-08-15 01:13:23 +00001185 line = self.text.get("iomark", "end-1c")
Kurt B. Kaiser5c3df352004-12-23 04:20:59 +00001186 if len(line) == 0: # may be EOF if we quit our mainloop with Ctrl-C
1187 line = "\n"
David Scherer7aced172000-08-15 01:13:23 +00001188 self.resetoutput()
1189 if self.canceled:
Terry Jan Reedy34a49aa2020-03-28 12:51:21 -04001190 self.canceled = False
Kurt B. Kaiser5c3df352004-12-23 04:20:59 +00001191 if not use_subprocess:
1192 raise KeyboardInterrupt
David Scherer7aced172000-08-15 01:13:23 +00001193 if self.endoffile:
Terry Jan Reedy34a49aa2020-03-28 12:51:21 -04001194 self.endoffile = False
Kurt B. Kaiser5c3df352004-12-23 04:20:59 +00001195 line = ""
David Scherer7aced172000-08-15 01:13:23 +00001196 return line
1197
1198 def isatty(self):
Kurt B. Kaiser837d15c2002-09-18 02:29:59 +00001199 return True
David Scherer7aced172000-08-15 01:13:23 +00001200
Kurt B. Kaiser003091c2003-02-17 18:57:16 +00001201 def cancel_callback(self, event=None):
David Scherer7aced172000-08-15 01:13:23 +00001202 try:
1203 if self.text.compare("sel.first", "!=", "sel.last"):
1204 return # Active selection -- always use default binding
1205 except:
1206 pass
1207 if not (self.executing or self.reading):
1208 self.resetoutput()
Kurt B. Kaiser7f38ec02003-05-15 03:19:42 +00001209 self.interp.write("KeyboardInterrupt\n")
David Scherer7aced172000-08-15 01:13:23 +00001210 self.showprompt()
1211 return "break"
Terry Jan Reedy34a49aa2020-03-28 12:51:21 -04001212 self.endoffile = False
1213 self.canceled = True
Kurt B. Kaiser5c3df352004-12-23 04:20:59 +00001214 if (self.executing and self.interp.rpcclt):
Kurt B. Kaiser67fd0ea2003-05-24 20:59:15 +00001215 if self.interp.getdebugger():
1216 self.interp.restart_subprocess()
1217 else:
1218 self.interp.interrupt_subprocess()
Kurt B. Kaiser5c3df352004-12-23 04:20:59 +00001219 if self.reading:
1220 self.top.quit() # exit the nested mainloop() in readline()
David Scherer7aced172000-08-15 01:13:23 +00001221 return "break"
1222
1223 def eof_callback(self, event):
1224 if self.executing and not self.reading:
1225 return # Let the default binding (delete next char) take over
1226 if not (self.text.compare("iomark", "==", "insert") and
1227 self.text.compare("insert", "==", "end-1c")):
1228 return # Let the default binding (delete next char) take over
1229 if not self.executing:
David Scherer7aced172000-08-15 01:13:23 +00001230 self.resetoutput()
1231 self.close()
1232 else:
Terry Jan Reedy34a49aa2020-03-28 12:51:21 -04001233 self.canceled = False
1234 self.endoffile = True
David Scherer7aced172000-08-15 01:13:23 +00001235 self.top.quit()
1236 return "break"
1237
David Scherer7aced172000-08-15 01:13:23 +00001238 def linefeed_callback(self, event):
1239 # Insert a linefeed without entering anything (still autoindented)
1240 if self.reading:
1241 self.text.insert("insert", "\n")
1242 self.text.see("insert")
1243 else:
Kurt B. Kaiser822a77f2002-12-16 02:07:11 +00001244 self.newline_and_indent_event(event)
David Scherer7aced172000-08-15 01:13:23 +00001245 return "break"
1246
1247 def enter_callback(self, event):
1248 if self.executing and not self.reading:
1249 return # Let the default binding (insert '\n') take over
1250 # If some text is selected, recall the selection
1251 # (but only if this before the I/O mark)
1252 try:
1253 sel = self.text.get("sel.first", "sel.last")
1254 if sel:
1255 if self.text.compare("sel.last", "<=", "iomark"):
Kurt B. Kaisera7daba62005-06-19 18:56:15 +00001256 self.recall(sel, event)
David Scherer7aced172000-08-15 01:13:23 +00001257 return "break"
1258 except:
1259 pass
1260 # If we're strictly before the line containing iomark, recall
1261 # the current line, less a leading prompt, less leading or
1262 # trailing whitespace
1263 if self.text.compare("insert", "<", "iomark linestart"):
Tal Einat15d38612021-04-29 01:27:55 +03001264 # Check if there's a relevant stdin range -- if so, use it.
1265 # Note: "stdin" blocks may include several successive statements,
1266 # so look for "console" tags on the newline before each statement
1267 # (and possibly on prompts).
David Scherer7aced172000-08-15 01:13:23 +00001268 prev = self.text.tag_prevrange("stdin", "insert")
Tal Einat15d38612021-04-29 01:27:55 +03001269 if (
1270 prev and
1271 self.text.compare("insert", "<", prev[1]) and
1272 # The following is needed to handle empty statements.
1273 "console" not in self.text.tag_names("insert")
1274 ):
1275 prev_cons = self.text.tag_prevrange("console", "insert")
1276 if prev_cons and self.text.compare(prev_cons[1], ">=", prev[0]):
1277 prev = (prev_cons[1], prev[1])
1278 next_cons = self.text.tag_nextrange("console", "insert")
1279 if next_cons and self.text.compare(next_cons[0], "<", prev[1]):
1280 prev = (prev[0], self.text.index(next_cons[0] + "+1c"))
Kurt B. Kaisera7daba62005-06-19 18:56:15 +00001281 self.recall(self.text.get(prev[0], prev[1]), event)
David Scherer7aced172000-08-15 01:13:23 +00001282 return "break"
1283 next = self.text.tag_nextrange("stdin", "insert")
1284 if next and self.text.compare("insert lineend", ">=", next[0]):
Tal Einat15d38612021-04-29 01:27:55 +03001285 next_cons = self.text.tag_nextrange("console", "insert lineend")
1286 if next_cons and self.text.compare(next_cons[0], "<", next[1]):
1287 next = (next[0], self.text.index(next_cons[0] + "+1c"))
Kurt B. Kaisera7daba62005-06-19 18:56:15 +00001288 self.recall(self.text.get(next[0], next[1]), event)
David Scherer7aced172000-08-15 01:13:23 +00001289 return "break"
Kurt B. Kaiser4ada7ad2002-12-29 22:03:38 +00001290 # No stdin mark -- just get the current line, less any prompt
Kurt B. Kaiserb1754452005-11-18 22:05:48 +00001291 indices = self.text.tag_nextrange("console", "insert linestart")
1292 if indices and \
1293 self.text.compare(indices[0], "<=", "insert linestart"):
1294 self.recall(self.text.get(indices[1], "insert lineend"), event)
1295 else:
1296 self.recall(self.text.get("insert linestart", "insert lineend"), event)
David Scherer7aced172000-08-15 01:13:23 +00001297 return "break"
Kurt B. Kaiser822a77f2002-12-16 02:07:11 +00001298 # If we're between the beginning of the line and the iomark, i.e.
Kurt B. Kaiser4ada7ad2002-12-29 22:03:38 +00001299 # in the prompt area, move to the end of the prompt
Kurt B. Kaiser822a77f2002-12-16 02:07:11 +00001300 if self.text.compare("insert", "<", "iomark"):
Kurt B. Kaiser4ada7ad2002-12-29 22:03:38 +00001301 self.text.mark_set("insert", "iomark")
David Scherer7aced172000-08-15 01:13:23 +00001302 # If we're in the current input and there's only whitespace
1303 # beyond the cursor, erase that whitespace first
1304 s = self.text.get("insert", "end-1c")
Kurt B. Kaiser837d15c2002-09-18 02:29:59 +00001305 if s and not s.strip():
David Scherer7aced172000-08-15 01:13:23 +00001306 self.text.delete("insert", "end-1c")
1307 # If we're in the current input before its last line,
1308 # insert a newline right at the insert point
1309 if self.text.compare("insert", "<", "end-1c linestart"):
Kurt B. Kaiser822a77f2002-12-16 02:07:11 +00001310 self.newline_and_indent_event(event)
David Scherer7aced172000-08-15 01:13:23 +00001311 return "break"
1312 # We're in the last line; append a newline and submit it
1313 self.text.mark_set("insert", "end-1c")
1314 if self.reading:
1315 self.text.insert("insert", "\n")
1316 self.text.see("insert")
1317 else:
Kurt B. Kaiser822a77f2002-12-16 02:07:11 +00001318 self.newline_and_indent_event(event)
David Scherer7aced172000-08-15 01:13:23 +00001319 self.text.update_idletasks()
1320 if self.reading:
Neal Norwitzce96f692006-03-17 06:49:51 +00001321 self.top.quit() # Break out of recursive mainloop()
David Scherer7aced172000-08-15 01:13:23 +00001322 else:
1323 self.runit()
1324 return "break"
1325
Kurt B. Kaisera7daba62005-06-19 18:56:15 +00001326 def recall(self, s, event):
Thomas Wouters00ee7ba2006-08-21 19:07:27 +00001327 # remove leading and trailing empty or whitespace lines
Tal Einat15d38612021-04-29 01:27:55 +03001328 s = re.sub(r'^\s*\n', '', s)
Thomas Wouters00ee7ba2006-08-21 19:07:27 +00001329 s = re.sub(r'\n\s*$', '', s)
1330 lines = s.split('\n')
Kurt B. Kaisera7daba62005-06-19 18:56:15 +00001331 self.text.undo_block_start()
1332 try:
1333 self.text.tag_remove("sel", "1.0", "end")
1334 self.text.mark_set("insert", "end-1c")
Thomas Wouters00ee7ba2006-08-21 19:07:27 +00001335 prefix = self.text.get("insert linestart", "insert")
1336 if prefix.rstrip().endswith(':'):
Kurt B. Kaiser8fa7eb52005-06-21 02:42:17 +00001337 self.newline_and_indent_event(event)
Thomas Wouters00ee7ba2006-08-21 19:07:27 +00001338 prefix = self.text.get("insert linestart", "insert")
Tal Einat15d38612021-04-29 01:27:55 +03001339 self.text.insert("insert", lines[0].strip(),
1340 self.user_input_insert_tags)
Kurt B. Kaiser8fa7eb52005-06-21 02:42:17 +00001341 if len(lines) > 1:
Thomas Wouters00ee7ba2006-08-21 19:07:27 +00001342 orig_base_indent = re.search(r'^([ \t]*)', lines[0]).group(0)
1343 new_base_indent = re.search(r'^([ \t]*)', prefix).group(0)
Kurt B. Kaiser8fa7eb52005-06-21 02:42:17 +00001344 for line in lines[1:]:
Thomas Wouters00ee7ba2006-08-21 19:07:27 +00001345 if line.startswith(orig_base_indent):
1346 # replace orig base indentation with new indentation
1347 line = new_base_indent + line[len(orig_base_indent):]
Tal Einat15d38612021-04-29 01:27:55 +03001348 self.text.insert('insert', '\n' + line.rstrip(),
1349 self.user_input_insert_tags)
Kurt B. Kaisera7daba62005-06-19 18:56:15 +00001350 finally:
1351 self.text.see("insert")
1352 self.text.undo_block_stop()
David Scherer7aced172000-08-15 01:13:23 +00001353
Tal Einat15d38612021-04-29 01:27:55 +03001354 _last_newline_re = re.compile(r"[ \t]*(\n[ \t]*)?\Z")
David Scherer7aced172000-08-15 01:13:23 +00001355 def runit(self):
Tal Einat15d38612021-04-29 01:27:55 +03001356 index_before = self.text.index("end-2c")
David Scherer7aced172000-08-15 01:13:23 +00001357 line = self.text.get("iomark", "end-1c")
1358 # Strip off last newline and surrounding whitespace.
1359 # (To allow you to hit return twice to end a statement.)
Tal Einat15d38612021-04-29 01:27:55 +03001360 line = self._last_newline_re.sub("", line)
1361 input_is_complete = self.interp.runsource(line)
1362 if not input_is_complete:
1363 if self.text.get(index_before) == '\n':
1364 self.text.tag_remove(self.user_input_insert_tags, index_before)
1365 self.shell_sidebar.update_sidebar()
David Scherer7aced172000-08-15 01:13:23 +00001366
David Scherer7aced172000-08-15 01:13:23 +00001367 def open_stack_viewer(self, event=None):
Chui Tey5d2af632002-05-26 13:36:41 +00001368 if self.interp.rpcclt:
1369 return self.interp.remote_stack_viewer()
David Scherer7aced172000-08-15 01:13:23 +00001370 try:
1371 sys.last_traceback
1372 except:
Terry Jan Reedy879986d2021-01-25 06:33:18 -05001373 messagebox.showerror("No stack trace",
David Scherer7aced172000-08-15 01:13:23 +00001374 "There is no stack trace yet.\n"
1375 "(sys.last_traceback is not defined)",
Terry Jan Reedy3be2e542015-09-25 22:22:55 -04001376 parent=self.text)
David Scherer7aced172000-08-15 01:13:23 +00001377 return
Terry Jan Reedy6fa5bdc2016-05-28 13:22:31 -04001378 from idlelib.stackviewer import StackBrowser
Terry Jan Reedy038c16b2015-05-15 23:03:17 -04001379 StackBrowser(self.root, self.flist)
David Scherer7aced172000-08-15 01:13:23 +00001380
Kurt B. Kaiser1061e722003-01-04 01:43:53 +00001381 def view_restart_mark(self, event=None):
1382 self.text.see("iomark")
1383 self.text.see("restart")
1384
1385 def restart_shell(self, event=None):
Terry Jan Reedyda4c4672012-01-31 02:26:32 -05001386 "Callback for Run/Restart Shell Cntl-F6"
1387 self.interp.restart_subprocess(with_cwd=True)
Kurt B. Kaiser1061e722003-01-04 01:43:53 +00001388
David Scherer7aced172000-08-15 01:13:23 +00001389 def showprompt(self):
1390 self.resetoutput()
Tal Einat15d38612021-04-29 01:27:55 +03001391
1392 prompt = self.prompt
1393 if self.sys_ps1 and prompt.endswith(self.sys_ps1):
1394 prompt = prompt[:-len(self.sys_ps1)]
1395 self.text.tag_add("console", "iomark-1c")
1396 self.console.write(prompt)
1397
1398 self.shell_sidebar.update_sidebar()
David Scherer7aced172000-08-15 01:13:23 +00001399 self.text.mark_set("insert", "end-1c")
Chui Tey5d2af632002-05-26 13:36:41 +00001400 self.set_line_and_column()
Kurt B. Kaiserdc1e7092002-07-11 04:33:41 +00001401 self.io.reset_undo()
David Scherer7aced172000-08-15 01:13:23 +00001402
Tal Einat9ebe8792018-12-07 08:32:21 +02001403 def show_warning(self, msg):
1404 width = self.interp.tkconsole.width
1405 wrapper = TextWrapper(width=width, tabsize=8, expand_tabs=True)
1406 wrapped_msg = '\n'.join(wrapper.wrap(msg))
1407 if not wrapped_msg.endswith('\n'):
1408 wrapped_msg += '\n'
1409 self.per.bottom.insert("iomark linestart", wrapped_msg, "stderr")
1410
David Scherer7aced172000-08-15 01:13:23 +00001411 def resetoutput(self):
1412 source = self.text.get("iomark", "end-1c")
1413 if self.history:
Terry Jan Reedy0a01ac42013-08-13 19:51:04 -04001414 self.history.store(source)
David Scherer7aced172000-08-15 01:13:23 +00001415 if self.text.get("end-2c") != "\n":
1416 self.text.insert("end-1c", "\n")
1417 self.text.mark_set("iomark", "end-1c")
Chui Tey5d2af632002-05-26 13:36:41 +00001418 self.set_line_and_column()
Zackery Spytzbfdeaa32020-01-30 18:55:42 -07001419 self.ctip.remove_calltip_window()
David Scherer7aced172000-08-15 01:13:23 +00001420
1421 def write(self, s, tags=()):
Kurt B. Kaiser003091c2003-02-17 18:57:16 +00001422 try:
1423 self.text.mark_gravity("iomark", "right")
Martin v. Löwis30d5e6c2012-07-25 11:32:26 +02001424 count = OutputWindow.write(self, s, tags, "iomark")
Kurt B. Kaiser003091c2003-02-17 18:57:16 +00001425 self.text.mark_gravity("iomark", "left")
1426 except:
Kurt B. Kaisercbbe98f2007-08-12 01:52:35 +00001427 raise ###pass # ### 11Aug07 KBK if we are expecting exceptions
1428 # let's find out what they are and be specific.
David Scherer7aced172000-08-15 01:13:23 +00001429 if self.canceled:
Terry Jan Reedy34a49aa2020-03-28 12:51:21 -04001430 self.canceled = False
Kurt B. Kaiser7f38ec02003-05-15 03:19:42 +00001431 if not use_subprocess:
1432 raise KeyboardInterrupt
Martin v. Löwis30d5e6c2012-07-25 11:32:26 +02001433 return count
David Scherer7aced172000-08-15 01:13:23 +00001434
Andrew Svetlovd1837672012-11-01 22:41:19 +02001435 def rmenu_check_cut(self):
1436 try:
1437 if self.text.compare('sel.first', '<', 'iomark'):
1438 return 'disabled'
1439 except TclError: # no selection, so the index 'sel.first' doesn't exist
1440 return 'disabled'
1441 return super().rmenu_check_cut()
1442
1443 def rmenu_check_paste(self):
1444 if self.text.compare('insert','<','iomark'):
1445 return 'disabled'
1446 return super().rmenu_check_paste()
1447
Tal Einat15d38612021-04-29 01:27:55 +03001448 def squeeze_current_text_event(self, event=None):
1449 self.squeezer.squeeze_current_text()
1450 self.shell_sidebar.update_sidebar()
1451
1452 def on_squeezed_expand(self, index, text, tags):
1453 self.shell_sidebar.update_sidebar()
1454
Kurt B. Kaiser969de452002-06-12 03:28:57 +00001455
Terry Jan Reedy5ad85992016-06-11 02:06:26 -04001456def fix_x11_paste(root):
1457 "Make paste replace selection on x11. See issue #5124."
1458 if root._windowingsystem == 'x11':
1459 for cls in 'Text', 'Entry', 'Spinbox':
1460 root.bind_class(
1461 cls,
1462 '<<Paste>>',
1463 'catch {%W delete sel.first sel.last}\n' +
1464 root.bind_class(cls, '<<Paste>>'))
1465
1466
David Scherer7aced172000-08-15 01:13:23 +00001467usage_msg = """\
David Scherer7aced172000-08-15 01:13:23 +00001468
Kurt B. Kaiser11659ad2003-05-15 23:23:21 +00001469USAGE: idle [-deins] [-t title] [file]*
1470 idle [-dns] [-t title] (-c cmd | -r file) [arg]*
1471 idle [-dns] [-t title] - [arg]*
Kurt B. Kaiser6655e4b2002-12-31 16:03:23 +00001472
Kurt B. Kaiserf06eed02002-12-11 04:42:04 +00001473 -h print this help message and exit
Andrew Svetlov9b1e59a2012-10-05 22:12:44 +03001474 -n run IDLE without a subprocess (DEPRECATED,
1475 see Help/IDLE Help for details)
David Scherer7aced172000-08-15 01:13:23 +00001476
Kurt B. Kaiserf06eed02002-12-11 04:42:04 +00001477The following options will override the IDLE 'settings' configuration:
Kurt B. Kaiser96d88422001-07-17 04:59:01 +00001478
Kurt B. Kaiserf06eed02002-12-11 04:42:04 +00001479 -e open an edit window
1480 -i open a shell window
1481
1482The following options imply -i and will open a shell:
1483
1484 -c cmd run the command in a shell, or
1485 -r file run script from file
1486
1487 -d enable the debugger
1488 -s run $IDLESTARTUP or $PYTHONSTARTUP before anything else
1489 -t title set title of shell window
1490
1491A default edit window will be bypassed when -c, -r, or - are used.
1492
1493[arg]* are passed to the command (-c) or script (-r) in sys.argv[1:].
1494
1495Examples:
1496
1497idle
1498 Open an edit window or shell depending on IDLE's configuration.
1499
1500idle foo.py foobar.py
1501 Edit the files, also open a shell if configured to start with shell.
1502
1503idle -est "Baz" foo.py
1504 Run $IDLESTARTUP or $PYTHONSTARTUP, edit foo.py, and open a shell
1505 window with the title "Baz".
1506
Neal Norwitz752abd02008-05-13 04:55:24 +00001507idle -c "import sys; print(sys.argv)" "foo"
Kurt B. Kaiserf06eed02002-12-11 04:42:04 +00001508 Open a shell window and run the command, passing "-c" in sys.argv[0]
1509 and "foo" in sys.argv[1].
1510
1511idle -d -s -r foo.py "Hello World"
1512 Open a shell window, run a startup script, enable the debugger, and
1513 run foo.py, passing "foo.py" in sys.argv[0] and "Hello World" in
1514 sys.argv[1].
1515
Neal Norwitz752abd02008-05-13 04:55:24 +00001516echo "import sys; print(sys.argv)" | idle - "foobar"
Kurt B. Kaiserf06eed02002-12-11 04:42:04 +00001517 Open a shell window, run the script piped in, passing '' in sys.argv[0]
1518 and "foobar" in sys.argv[1].
David Scherer7aced172000-08-15 01:13:23 +00001519"""
1520
Kurt B. Kaiser969de452002-06-12 03:28:57 +00001521def main():
Serhiy Storchaka7e4db2f2017-05-04 08:17:47 +03001522 import getopt
1523 from platform import system
1524 from idlelib import testing # bool value
1525 from idlelib import macosx
1526
Kurt B. Kaiserf06eed02002-12-11 04:42:04 +00001527 global flist, root, use_subprocess
David Scherer7aced172000-08-15 01:13:23 +00001528
Terry Jan Reedy95a3f112013-06-28 23:50:12 -04001529 capture_warnings(True)
Kurt B. Kaiser8f570a72003-05-15 18:52:51 +00001530 use_subprocess = True
Roger Serwyc35151c2013-03-31 23:28:55 -05001531 enable_shell = False
Kurt B. Kaiserf06eed02002-12-11 04:42:04 +00001532 enable_edit = False
1533 debug = False
1534 cmd = None
1535 script = None
1536 startup = False
Kurt B. Kaiser969de452002-06-12 03:28:57 +00001537 try:
Kurt B. Kaiser8f570a72003-05-15 18:52:51 +00001538 opts, args = getopt.getopt(sys.argv[1:], "c:deihnr:st:")
Guido van Rossumb940e112007-01-10 16:19:56 +00001539 except getopt.error as msg:
Terry Jan Reedy81b062f2014-09-19 22:38:41 -04001540 print("Error: %s\n%s" % (msg, usage_msg), file=sys.stderr)
Kurt B. Kaiser969de452002-06-12 03:28:57 +00001541 sys.exit(2)
Kurt B. Kaiser969de452002-06-12 03:28:57 +00001542 for o, a in opts:
1543 if o == '-c':
1544 cmd = a
Kurt B. Kaiserf06eed02002-12-11 04:42:04 +00001545 enable_shell = True
Kurt B. Kaiser969de452002-06-12 03:28:57 +00001546 if o == '-d':
Kurt B. Kaiserf06eed02002-12-11 04:42:04 +00001547 debug = True
1548 enable_shell = True
Kurt B. Kaiser969de452002-06-12 03:28:57 +00001549 if o == '-e':
Kurt B. Kaiserf06eed02002-12-11 04:42:04 +00001550 enable_edit = True
1551 if o == '-h':
1552 sys.stdout.write(usage_msg)
1553 sys.exit()
1554 if o == '-i':
1555 enable_shell = True
Kurt B. Kaiser8f570a72003-05-15 18:52:51 +00001556 if o == '-n':
Andrew Svetlov9b1e59a2012-10-05 22:12:44 +03001557 print(" Warning: running IDLE without a subprocess is deprecated.",
1558 file=sys.stderr)
Kurt B. Kaiser8f570a72003-05-15 18:52:51 +00001559 use_subprocess = False
Kurt B. Kaiser969de452002-06-12 03:28:57 +00001560 if o == '-r':
1561 script = a
Kurt B. Kaiserf06eed02002-12-11 04:42:04 +00001562 if os.path.isfile(script):
1563 pass
1564 else:
Guido van Rossumbe19ed72007-02-09 05:37:30 +00001565 print("No script file: ", script)
Kurt B. Kaiserf06eed02002-12-11 04:42:04 +00001566 sys.exit()
1567 enable_shell = True
Kurt B. Kaiser969de452002-06-12 03:28:57 +00001568 if o == '-s':
Kurt B. Kaiserf06eed02002-12-11 04:42:04 +00001569 startup = True
1570 enable_shell = True
Kurt B. Kaiser969de452002-06-12 03:28:57 +00001571 if o == '-t':
1572 PyShell.shell_title = a
Kurt B. Kaiserf06eed02002-12-11 04:42:04 +00001573 enable_shell = True
1574 if args and args[0] == '-':
1575 cmd = sys.stdin.read()
1576 enable_shell = True
Kurt B. Kaiserf06eed02002-12-11 04:42:04 +00001577 # process sys.argv and sys.path:
Kurt B. Kaiser969de452002-06-12 03:28:57 +00001578 for i in range(len(sys.path)):
1579 sys.path[i] = os.path.abspath(sys.path[i])
Kurt B. Kaiserf06eed02002-12-11 04:42:04 +00001580 if args and args[0] == '-':
1581 sys.argv = [''] + args[1:]
1582 elif cmd:
1583 sys.argv = ['-c'] + args
1584 elif script:
1585 sys.argv = [script] + args
1586 elif args:
1587 enable_edit = True
1588 pathx = []
Kurt B. Kaiser969de452002-06-12 03:28:57 +00001589 for filename in args:
1590 pathx.append(os.path.dirname(filename))
Kurt B. Kaiserf06eed02002-12-11 04:42:04 +00001591 for dir in pathx:
1592 dir = os.path.abspath(dir)
1593 if not dir in sys.path:
1594 sys.path.insert(0, dir)
Kurt B. Kaiserff002b92002-12-21 21:39:11 +00001595 else:
Neal Norwitzd082b6e2007-08-26 23:37:53 +00001596 dir = os.getcwd()
1597 if dir not in sys.path:
Kurt B. Kaiserff002b92002-12-21 21:39:11 +00001598 sys.path.insert(0, dir)
Kurt B. Kaiserf06eed02002-12-11 04:42:04 +00001599 # check the IDLE settings configuration (but command line overrides)
1600 edit_start = idleConf.GetOption('main', 'General',
Kurt B. Kaiser6655e4b2002-12-31 16:03:23 +00001601 'editor-on-startup', type='bool')
Kurt B. Kaiserf06eed02002-12-11 04:42:04 +00001602 enable_edit = enable_edit or edit_start
Roger Serwyc35151c2013-03-31 23:28:55 -05001603 enable_shell = enable_shell or not enable_edit
Terry Jan Reedy5ad85992016-06-11 02:06:26 -04001604
Terry Jan Reedyf35bb9e2016-08-16 00:10:14 -04001605 # Setup root. Don't break user code run in IDLE process.
1606 # Don't change environment when testing.
1607 if use_subprocess and not testing:
Terry Jan Reedyb60adc52016-06-21 18:41:38 -04001608 NoDefaultRoot()
Kurt B. Kaiser969de452002-06-12 03:28:57 +00001609 root = Tk(className="Idle")
Terry Jan Reedy5ad85992016-06-11 02:06:26 -04001610 root.withdraw()
Serhiy Storchakaa96c96f2017-09-21 11:20:06 +03001611 from idlelib.run import fix_scaling
1612 fix_scaling(root)
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001613
Terry Jan Reedya7c07d32014-02-08 09:02:26 -05001614 # set application icon
1615 icondir = os.path.join(os.path.dirname(__file__), 'Icons')
1616 if system() == 'Windows':
1617 iconfile = os.path.join(icondir, 'idle.ico')
1618 root.wm_iconbitmap(default=iconfile)
Ned Deily7eebbbd2019-03-01 17:53:50 -05001619 elif not macosx.isAquaTk():
Miro Hrončok3a69f3c2020-04-22 09:21:44 +02001620 if TkVersion >= 8.6:
1621 ext = '.png'
1622 sizes = (16, 32, 48, 256)
1623 else:
1624 ext = '.gif'
1625 sizes = (16, 32, 48)
Terry Jan Reedya7c07d32014-02-08 09:02:26 -05001626 iconfiles = [os.path.join(icondir, 'idle_%d%s' % (size, ext))
Miro Hrončok3a69f3c2020-04-22 09:21:44 +02001627 for size in sizes]
Serhiy Storchakad3af1562016-06-25 23:52:51 +03001628 icons = [PhotoImage(master=root, file=iconfile)
1629 for iconfile in iconfiles]
Terry Jan Reedya7c07d32014-02-08 09:02:26 -05001630 root.wm_iconphoto(True, *icons)
1631
Terry Jan Reedyb60adc52016-06-21 18:41:38 -04001632 # start editor and/or shell windows:
Kurt B. Kaiser969de452002-06-12 03:28:57 +00001633 fixwordbreaks(root)
Terry Jan Reedy5ad85992016-06-11 02:06:26 -04001634 fix_x11_paste(root)
Kurt B. Kaiser969de452002-06-12 03:28:57 +00001635 flist = PyShellFileList(root)
Terry Jan Reedy6fa5bdc2016-05-28 13:22:31 -04001636 macosx.setupApp(root, flist)
Terry Jan Reedyf94471c2015-09-22 21:13:28 -04001637
Kurt B. Kaiserf06eed02002-12-11 04:42:04 +00001638 if enable_edit:
1639 if not (cmd or script):
Andrew Svetlov6b6e4372012-03-20 23:03:26 +02001640 for filename in args[:]:
1641 if flist.open(filename) is None:
1642 # filename is a directory actually, disconsider it
1643 args.remove(filename)
Kurt B. Kaiserf06eed02002-12-11 04:42:04 +00001644 if not args:
1645 flist.new()
Ned Deilyf3c65892013-12-10 16:24:01 -08001646
Kurt B. Kaiseraf3eb872004-01-21 18:54:30 +00001647 if enable_shell:
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001648 shell = flist.open_shell()
1649 if not shell:
Kurt B. Kaiseraf3eb872004-01-21 18:54:30 +00001650 return # couldn't open shell
Terry Jan Reedy6fa5bdc2016-05-28 13:22:31 -04001651 if macosx.isAquaTk() and flist.dict:
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001652 # On OSX: when the user has double-clicked on a file that causes
1653 # IDLE to be launched the shell window will open just in front of
1654 # the file she wants to see. Lower the interpreter window when
1655 # there are open files.
1656 shell.top.lower()
Ned Deilyf3c65892013-12-10 16:24:01 -08001657 else:
1658 shell = flist.pyshell
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001659
Ned Deilyf3c65892013-12-10 16:24:01 -08001660 # Handle remaining options. If any of these are set, enable_shell
1661 # was set also, so shell must be true to reach here.
Kurt B. Kaiserf06eed02002-12-11 04:42:04 +00001662 if debug:
1663 shell.open_debugger()
Kurt B. Kaiser969de452002-06-12 03:28:57 +00001664 if startup:
1665 filename = os.environ.get("IDLESTARTUP") or \
1666 os.environ.get("PYTHONSTARTUP")
1667 if filename and os.path.isfile(filename):
Kurt B. Kaiserf06eed02002-12-11 04:42:04 +00001668 shell.interp.execfile(filename)
Ned Deilyf3c65892013-12-10 16:24:01 -08001669 if cmd or script:
Kurt B. Kaiserf06eed02002-12-11 04:42:04 +00001670 shell.interp.runcommand("""if 1:
1671 import sys as _sys
Walter Dörwald70a6b492004-02-12 17:35:32 +00001672 _sys.argv = %r
Kurt B. Kaiserf06eed02002-12-11 04:42:04 +00001673 del _sys
Walter Dörwald70a6b492004-02-12 17:35:32 +00001674 \n""" % (sys.argv,))
Kurt B. Kaiserf06eed02002-12-11 04:42:04 +00001675 if cmd:
1676 shell.interp.execsource(cmd)
1677 elif script:
Kurt B. Kaiser11659ad2003-05-15 23:23:21 +00001678 shell.interp.prepend_syspath(script)
Kurt B. Kaiserf06eed02002-12-11 04:42:04 +00001679 shell.interp.execfile(script)
Ned Deilyf3c65892013-12-10 16:24:01 -08001680 elif shell:
1681 # If there is a shell window and no cmd or script in progress,
Tal Einat9ebe8792018-12-07 08:32:21 +02001682 # check for problematic issues and print warning message(s) in
1683 # the IDLE shell window; this is less intrusive than always
1684 # opening a separate window.
1685
1686 # Warn if using a problematic OS X Tk version.
Terry Jan Reedy6fa5bdc2016-05-28 13:22:31 -04001687 tkversionwarning = macosx.tkVersionWarning(root)
Ned Deilyf3c65892013-12-10 16:24:01 -08001688 if tkversionwarning:
Tal Einat9ebe8792018-12-07 08:32:21 +02001689 shell.show_warning(tkversionwarning)
1690
1691 # Warn if the "Prefer tabs when opening documents" system
1692 # preference is set to "Always".
1693 prefer_tabs_preference_warning = macosx.preferTabsPreferenceWarning()
1694 if prefer_tabs_preference_warning:
1695 shell.show_warning(prefer_tabs_preference_warning)
Ned Deily4ce92b22011-01-15 04:37:12 +00001696
Terry Jan Reedycd6b8c62012-05-26 20:23:45 -04001697 while flist.inversedict: # keep IDLE running while files are open.
1698 root.mainloop()
Kurt B. Kaiser969de452002-06-12 03:28:57 +00001699 root.destroy()
Terry Jan Reedy95a3f112013-06-28 23:50:12 -04001700 capture_warnings(False)
Kurt B. Kaiser969de452002-06-12 03:28:57 +00001701
David Scherer7aced172000-08-15 01:13:23 +00001702if __name__ == "__main__":
David Scherer7aced172000-08-15 01:13:23 +00001703 main()
Terry Jan Reedy95a3f112013-06-28 23:50:12 -04001704
1705capture_warnings(False) # Make sure turned off; see issue 18081