986 lines
35 KiB
Python
986 lines
35 KiB
Python
|
##################################################################
|
||
|
##
|
||
|
## Interactive Shell Window
|
||
|
##
|
||
|
|
||
|
import array
|
||
|
import code
|
||
|
import os
|
||
|
import string
|
||
|
import sys
|
||
|
import traceback
|
||
|
|
||
|
import __main__
|
||
|
import afxres
|
||
|
import pywin.framework.app
|
||
|
import pywin.scintilla.control
|
||
|
import pywin.scintilla.formatter
|
||
|
import pywin.scintilla.IDLEenvironment
|
||
|
import win32api
|
||
|
import win32clipboard
|
||
|
import win32con
|
||
|
import win32ui
|
||
|
|
||
|
## sequential after ID_GOTO_LINE defined in editor.py
|
||
|
ID_EDIT_COPY_CODE = 0xE2002
|
||
|
ID_EDIT_EXEC_CLIPBOARD = 0x2003
|
||
|
|
||
|
trace = pywin.scintilla.formatter.trace
|
||
|
|
||
|
import re
|
||
|
|
||
|
from . import winout
|
||
|
|
||
|
# from IDLE.
|
||
|
_is_block_opener = re.compile(r":\s*(#.*)?$").search
|
||
|
_is_block_closer = re.compile(
|
||
|
r"""
|
||
|
\s*
|
||
|
( return
|
||
|
| break
|
||
|
| continue
|
||
|
| raise
|
||
|
| pass
|
||
|
)
|
||
|
\b
|
||
|
""",
|
||
|
re.VERBOSE,
|
||
|
).match
|
||
|
|
||
|
tracebackHeader = "Traceback (".encode("ascii")
|
||
|
|
||
|
sectionProfile = "Interactive Window"
|
||
|
valueFormatTitle = "FormatTitle"
|
||
|
valueFormatInput = "FormatInput"
|
||
|
valueFormatOutput = "FormatOutput"
|
||
|
valueFormatOutputError = "FormatOutputError"
|
||
|
|
||
|
# These are defaults only. Values are read from the registry.
|
||
|
formatTitle = (-536870897, 0, 220, 0, 16711680, 184, 34, "Arial")
|
||
|
formatInput = (-402653169, 0, 200, 0, 0, 0, 49, "Courier New")
|
||
|
formatOutput = (-402653169, 0, 200, 0, 8421376, 0, 49, "Courier New")
|
||
|
formatOutputError = (-402653169, 0, 200, 0, 255, 0, 49, "Courier New")
|
||
|
|
||
|
try:
|
||
|
sys.ps1
|
||
|
except AttributeError:
|
||
|
sys.ps1 = ">>> "
|
||
|
sys.ps2 = "... "
|
||
|
|
||
|
|
||
|
def LoadPreference(preference, default=""):
|
||
|
return win32ui.GetProfileVal(sectionProfile, preference, default)
|
||
|
|
||
|
|
||
|
def SavePreference(prefName, prefValue):
|
||
|
win32ui.WriteProfileVal(sectionProfile, prefName, prefValue)
|
||
|
|
||
|
|
||
|
def GetPromptPrefix(line):
|
||
|
ps1 = sys.ps1
|
||
|
if line[: len(ps1)] == ps1:
|
||
|
return ps1
|
||
|
ps2 = sys.ps2
|
||
|
if line[: len(ps2)] == ps2:
|
||
|
return ps2
|
||
|
|
||
|
|
||
|
#############################################################
|
||
|
#
|
||
|
# Colorizer related code.
|
||
|
#
|
||
|
#############################################################
|
||
|
STYLE_INTERACTIVE_EOL = "Interactive EOL"
|
||
|
STYLE_INTERACTIVE_OUTPUT = "Interactive Output"
|
||
|
STYLE_INTERACTIVE_PROMPT = "Interactive Prompt"
|
||
|
STYLE_INTERACTIVE_BANNER = "Interactive Banner"
|
||
|
STYLE_INTERACTIVE_ERROR = "Interactive Error"
|
||
|
STYLE_INTERACTIVE_ERROR_FINALLINE = "Interactive Error (final line)"
|
||
|
|
||
|
INTERACTIVE_STYLES = [
|
||
|
STYLE_INTERACTIVE_EOL,
|
||
|
STYLE_INTERACTIVE_OUTPUT,
|
||
|
STYLE_INTERACTIVE_PROMPT,
|
||
|
STYLE_INTERACTIVE_BANNER,
|
||
|
STYLE_INTERACTIVE_ERROR,
|
||
|
STYLE_INTERACTIVE_ERROR_FINALLINE,
|
||
|
]
|
||
|
|
||
|
FormatterParent = pywin.scintilla.formatter.PythonSourceFormatter
|
||
|
|
||
|
|
||
|
class InteractiveFormatter(FormatterParent):
|
||
|
def __init__(self, scintilla):
|
||
|
FormatterParent.__init__(self, scintilla)
|
||
|
self.bannerDisplayed = False
|
||
|
|
||
|
def SetStyles(self):
|
||
|
FormatterParent.SetStyles(self)
|
||
|
Style = pywin.scintilla.formatter.Style
|
||
|
self.RegisterStyle(Style(STYLE_INTERACTIVE_EOL, STYLE_INTERACTIVE_PROMPT))
|
||
|
self.RegisterStyle(Style(STYLE_INTERACTIVE_PROMPT, formatInput))
|
||
|
self.RegisterStyle(Style(STYLE_INTERACTIVE_OUTPUT, formatOutput))
|
||
|
self.RegisterStyle(Style(STYLE_INTERACTIVE_BANNER, formatTitle))
|
||
|
self.RegisterStyle(Style(STYLE_INTERACTIVE_ERROR, formatOutputError))
|
||
|
self.RegisterStyle(
|
||
|
Style(STYLE_INTERACTIVE_ERROR_FINALLINE, STYLE_INTERACTIVE_ERROR)
|
||
|
)
|
||
|
|
||
|
def LoadPreference(self, name, default):
|
||
|
rc = win32ui.GetProfileVal("Format", name, default)
|
||
|
if rc == default:
|
||
|
rc = win32ui.GetProfileVal(sectionProfile, name, default)
|
||
|
return rc
|
||
|
|
||
|
def ColorizeInteractiveCode(self, cdoc, styleStart, stylePyStart):
|
||
|
lengthDoc = len(cdoc)
|
||
|
if lengthDoc == 0:
|
||
|
return
|
||
|
state = styleStart
|
||
|
# As per comments in Colorize(), we work with the raw utf8
|
||
|
# bytes. To avoid too muych py3k pain, we treat each utf8 byte
|
||
|
# as a latin-1 unicode character - we only use it to compare
|
||
|
# against ascii chars anyway...
|
||
|
chNext = cdoc[0:1].decode("latin-1")
|
||
|
startSeg = 0
|
||
|
i = 0
|
||
|
lastState = state # debug only
|
||
|
while i < lengthDoc:
|
||
|
ch = chNext
|
||
|
chNext = cdoc[i + 1 : i + 2].decode("latin-1")
|
||
|
|
||
|
# trace("ch=%r, i=%d, next=%r, state=%s" % (ch, i, chNext, state))
|
||
|
if state == STYLE_INTERACTIVE_EOL:
|
||
|
if ch not in "\r\n":
|
||
|
self.ColorSeg(startSeg, i - 1, state)
|
||
|
startSeg = i
|
||
|
if ch in (sys.ps1[0], sys.ps2[0]):
|
||
|
state = STYLE_INTERACTIVE_PROMPT
|
||
|
elif cdoc[i : i + len(tracebackHeader)] == tracebackHeader:
|
||
|
state = STYLE_INTERACTIVE_ERROR
|
||
|
else:
|
||
|
state = STYLE_INTERACTIVE_OUTPUT
|
||
|
elif state == STYLE_INTERACTIVE_PROMPT:
|
||
|
if ch not in sys.ps1 + sys.ps2 + " ":
|
||
|
self.ColorSeg(startSeg, i - 1, state)
|
||
|
startSeg = i
|
||
|
if ch in "\r\n":
|
||
|
state = STYLE_INTERACTIVE_EOL
|
||
|
else:
|
||
|
state = stylePyStart # Start coloring Python code.
|
||
|
elif state in (STYLE_INTERACTIVE_OUTPUT,):
|
||
|
if ch in "\r\n":
|
||
|
self.ColorSeg(startSeg, i - 1, state)
|
||
|
startSeg = i
|
||
|
state = STYLE_INTERACTIVE_EOL
|
||
|
elif state == STYLE_INTERACTIVE_ERROR:
|
||
|
if ch in "\r\n" and chNext and chNext not in string.whitespace:
|
||
|
# Everything including me
|
||
|
self.ColorSeg(startSeg, i, state)
|
||
|
startSeg = i + 1
|
||
|
state = STYLE_INTERACTIVE_ERROR_FINALLINE
|
||
|
elif i == 0 and ch not in string.whitespace:
|
||
|
# If we are coloring from the start of a line,
|
||
|
# we need this better check for the last line
|
||
|
# Color up to not including me
|
||
|
self.ColorSeg(startSeg, i - 1, state)
|
||
|
startSeg = i
|
||
|
state = STYLE_INTERACTIVE_ERROR_FINALLINE
|
||
|
elif state == STYLE_INTERACTIVE_ERROR_FINALLINE:
|
||
|
if ch in "\r\n":
|
||
|
self.ColorSeg(startSeg, i - 1, state)
|
||
|
startSeg = i
|
||
|
state = STYLE_INTERACTIVE_EOL
|
||
|
elif state == STYLE_INTERACTIVE_BANNER:
|
||
|
if ch in "\r\n" and (chNext == "" or chNext in ">["):
|
||
|
# Everything including me
|
||
|
self.ColorSeg(startSeg, i - 1, state)
|
||
|
startSeg = i
|
||
|
state = STYLE_INTERACTIVE_EOL
|
||
|
else:
|
||
|
# It is a PythonColorizer state - seek past the end of the line
|
||
|
# and ask the Python colorizer to color that.
|
||
|
end = startSeg
|
||
|
while end < lengthDoc and cdoc[end] not in "\r\n".encode("ascii"):
|
||
|
end = end + 1
|
||
|
self.ColorizePythonCode(cdoc[:end], startSeg, state)
|
||
|
stylePyStart = self.GetStringStyle(end - 1)
|
||
|
if stylePyStart is None:
|
||
|
stylePyStart = pywin.scintilla.formatter.STYLE_DEFAULT
|
||
|
else:
|
||
|
stylePyStart = stylePyStart.name
|
||
|
startSeg = end
|
||
|
i = end - 1 # ready for increment.
|
||
|
chNext = cdoc[end : end + 1].decode("latin-1")
|
||
|
state = STYLE_INTERACTIVE_EOL
|
||
|
if lastState != state:
|
||
|
lastState = state
|
||
|
i = i + 1
|
||
|
# and the rest
|
||
|
if startSeg < i:
|
||
|
self.ColorSeg(startSeg, i - 1, state)
|
||
|
|
||
|
def Colorize(self, start=0, end=-1):
|
||
|
# scintilla's formatting is all done in terms of utf, so
|
||
|
# we work with utf8 bytes instead of unicode. This magically
|
||
|
# works as any extended chars found in the utf8 don't change
|
||
|
# the semantics.
|
||
|
stringVal = self.scintilla.GetTextRange(start, end, decode=False)
|
||
|
styleStart = None
|
||
|
stylePyStart = None
|
||
|
if start > 1:
|
||
|
# Likely we are being asked to color from the start of the line.
|
||
|
# We find the last formatted character on the previous line.
|
||
|
# If TQString, we continue it. Otherwise, we reset.
|
||
|
look = start - 1
|
||
|
while look and self.scintilla.SCIGetCharAt(look) in "\n\r":
|
||
|
look = look - 1
|
||
|
if look and look < start - 1: # Did we find a char before the \n\r sets?
|
||
|
strstyle = self.GetStringStyle(look)
|
||
|
quote_char = None
|
||
|
if strstyle is not None:
|
||
|
if strstyle.name == pywin.scintilla.formatter.STYLE_TQSSTRING:
|
||
|
quote_char = "'"
|
||
|
elif strstyle.name == pywin.scintilla.formatter.STYLE_TQDSTRING:
|
||
|
quote_char = '"'
|
||
|
if quote_char is not None:
|
||
|
# It is a TQS. If the TQS is not terminated, we
|
||
|
# carry the style through.
|
||
|
if look > 2:
|
||
|
look_str = (
|
||
|
self.scintilla.SCIGetCharAt(look - 2)
|
||
|
+ self.scintilla.SCIGetCharAt(look - 1)
|
||
|
+ self.scintilla.SCIGetCharAt(look)
|
||
|
)
|
||
|
if look_str != quote_char * 3:
|
||
|
stylePyStart = strstyle.name
|
||
|
if stylePyStart is None:
|
||
|
stylePyStart = pywin.scintilla.formatter.STYLE_DEFAULT
|
||
|
|
||
|
if start > 0:
|
||
|
stylenum = self.scintilla.SCIGetStyleAt(start - 1)
|
||
|
styleStart = self.GetStyleByNum(stylenum).name
|
||
|
elif self.bannerDisplayed:
|
||
|
styleStart = STYLE_INTERACTIVE_EOL
|
||
|
else:
|
||
|
styleStart = STYLE_INTERACTIVE_BANNER
|
||
|
self.bannerDisplayed = True
|
||
|
self.scintilla.SCIStartStyling(start, 31)
|
||
|
self.style_buffer = array.array("b", (0,) * len(stringVal))
|
||
|
self.ColorizeInteractiveCode(stringVal, styleStart, stylePyStart)
|
||
|
self.scintilla.SCISetStylingEx(self.style_buffer)
|
||
|
self.style_buffer = None
|
||
|
|
||
|
|
||
|
###############################################################
|
||
|
#
|
||
|
# This class handles the Python interactive interpreter.
|
||
|
#
|
||
|
# It uses a basic EditWindow, and does all the magic.
|
||
|
# This is triggered by the enter key hander attached by the
|
||
|
# start-up code. It determines if a command is to be executed
|
||
|
# or continued (ie, emit "... ") by snooping around the current
|
||
|
# line, looking for the prompts
|
||
|
#
|
||
|
class PythonwinInteractiveInterpreter(code.InteractiveInterpreter):
|
||
|
def __init__(self, locals=None, globals=None):
|
||
|
if locals is None:
|
||
|
locals = __main__.__dict__
|
||
|
if globals is None:
|
||
|
globals = locals
|
||
|
self.globals = globals
|
||
|
code.InteractiveInterpreter.__init__(self, locals)
|
||
|
|
||
|
def showsyntaxerror(self, filename=None):
|
||
|
sys.stderr.write(
|
||
|
tracebackHeader.decode("ascii")
|
||
|
) # So the color syntaxer recognises it.
|
||
|
code.InteractiveInterpreter.showsyntaxerror(self, filename)
|
||
|
|
||
|
def runcode(self, code):
|
||
|
try:
|
||
|
exec(code, self.globals, self.locals)
|
||
|
except SystemExit:
|
||
|
raise
|
||
|
except:
|
||
|
self.showtraceback()
|
||
|
|
||
|
|
||
|
class InteractiveCore:
|
||
|
def __init__(self, banner=None):
|
||
|
self.banner = banner
|
||
|
|
||
|
# LoadFontPreferences()
|
||
|
def Init(self):
|
||
|
self.oldStdOut = self.oldStdErr = None
|
||
|
|
||
|
# self.SetWordWrap(win32ui.CRichEditView_WrapNone)
|
||
|
self.interp = PythonwinInteractiveInterpreter()
|
||
|
|
||
|
self.OutputGrab() # Release at cleanup.
|
||
|
|
||
|
if self.GetTextLength() == 0:
|
||
|
if self.banner is None:
|
||
|
suffix = ""
|
||
|
if win32ui.debug:
|
||
|
suffix = ", debug build"
|
||
|
sys.stderr.write(
|
||
|
"PythonWin %s on %s%s.\n" % (sys.version, sys.platform, suffix)
|
||
|
)
|
||
|
sys.stderr.write(
|
||
|
"Portions %s - see 'Help/About PythonWin' for further copyright information.\n"
|
||
|
% (win32ui.copyright,)
|
||
|
)
|
||
|
else:
|
||
|
sys.stderr.write(banner)
|
||
|
rcfile = os.environ.get("PYTHONSTARTUP")
|
||
|
if rcfile:
|
||
|
import __main__
|
||
|
|
||
|
try:
|
||
|
exec(
|
||
|
compile(
|
||
|
open(rcfile, "rb").read(), rcfile, "exec", dont_inherit=True
|
||
|
),
|
||
|
__main__.__dict__,
|
||
|
__main__.__dict__,
|
||
|
)
|
||
|
except:
|
||
|
sys.stderr.write(
|
||
|
">>> \nError executing PYTHONSTARTUP script %r\n" % (rcfile)
|
||
|
)
|
||
|
traceback.print_exc(file=sys.stderr)
|
||
|
self.AppendToPrompt([])
|
||
|
|
||
|
def SetContext(self, globals, locals, name="Dbg"):
|
||
|
oldPrompt = sys.ps1
|
||
|
if globals is None:
|
||
|
# Reset
|
||
|
sys.ps1 = ">>> "
|
||
|
sys.ps2 = "... "
|
||
|
locals = globals = __main__.__dict__
|
||
|
else:
|
||
|
sys.ps1 = "[%s]>>> " % name
|
||
|
sys.ps2 = "[%s]... " % name
|
||
|
self.interp.locals = locals
|
||
|
self.interp.globals = globals
|
||
|
self.AppendToPrompt([], oldPrompt)
|
||
|
|
||
|
def GetContext(self):
|
||
|
return self.interp.globals, self.interp.locals
|
||
|
|
||
|
def DoGetLine(self, line=-1):
|
||
|
if line == -1:
|
||
|
line = self.LineFromChar()
|
||
|
line = self.GetLine(line)
|
||
|
while line and line[-1] in ("\r", "\n"):
|
||
|
line = line[:-1]
|
||
|
return line
|
||
|
|
||
|
def AppendToPrompt(self, bufLines, oldPrompt=None):
|
||
|
"Take a command and stick it at the end of the buffer (with python prompts inserted if required)."
|
||
|
self.flush()
|
||
|
lastLineNo = self.GetLineCount() - 1
|
||
|
line = self.DoGetLine(lastLineNo)
|
||
|
if oldPrompt and line == oldPrompt:
|
||
|
self.SetSel(self.GetTextLength() - len(oldPrompt), self.GetTextLength())
|
||
|
self.ReplaceSel(sys.ps1)
|
||
|
elif line != str(sys.ps1):
|
||
|
if len(line) != 0:
|
||
|
self.write("\n")
|
||
|
self.write(sys.ps1)
|
||
|
self.flush()
|
||
|
self.idle.text.mark_set("iomark", "end-1c")
|
||
|
if not bufLines:
|
||
|
return
|
||
|
terms = (["\n" + sys.ps2] * (len(bufLines) - 1)) + [""]
|
||
|
for bufLine, term in zip(bufLines, terms):
|
||
|
if bufLine.strip():
|
||
|
self.write(bufLine + term)
|
||
|
self.flush()
|
||
|
|
||
|
def EnsureNoPrompt(self):
|
||
|
# Get ready to write some text NOT at a Python prompt.
|
||
|
self.flush()
|
||
|
lastLineNo = self.GetLineCount() - 1
|
||
|
line = self.DoGetLine(lastLineNo)
|
||
|
if not line or line in (sys.ps1, sys.ps2):
|
||
|
self.SetSel(self.GetTextLength() - len(line), self.GetTextLength())
|
||
|
self.ReplaceSel("")
|
||
|
else:
|
||
|
# Just add a new line.
|
||
|
self.write("\n")
|
||
|
|
||
|
def _GetSubConfigNames(self):
|
||
|
return ["interactive"] # Allow [Keys:Interactive] sections to be specific
|
||
|
|
||
|
def HookHandlers(self):
|
||
|
# Hook menu command (executed when a menu item with that ID is selected from a menu/toolbar
|
||
|
self.HookCommand(self.OnSelectBlock, win32ui.ID_EDIT_SELECT_BLOCK)
|
||
|
self.HookCommand(self.OnEditCopyCode, ID_EDIT_COPY_CODE)
|
||
|
self.HookCommand(self.OnEditExecClipboard, ID_EDIT_EXEC_CLIPBOARD)
|
||
|
mod = pywin.scintilla.IDLEenvironment.GetIDLEModule("IdleHistory")
|
||
|
if mod is not None:
|
||
|
self.history = mod.History(self.idle.text, "\n" + sys.ps2)
|
||
|
else:
|
||
|
self.history = None
|
||
|
# hack for now for event handling.
|
||
|
|
||
|
# GetBlockBoundary takes a line number, and will return the
|
||
|
# start and and line numbers of the block, and a flag indicating if the
|
||
|
# block is a Python code block.
|
||
|
# If the line specified has a Python prompt, then the lines are parsed
|
||
|
# backwards and forwards, and the flag is true.
|
||
|
# If the line does not start with a prompt, the block is searched forward
|
||
|
# and backward until a prompt _is_ found, and all lines in between without
|
||
|
# prompts are returned, and the flag is false.
|
||
|
def GetBlockBoundary(self, lineNo):
|
||
|
line = self.DoGetLine(lineNo)
|
||
|
maxLineNo = self.GetLineCount() - 1
|
||
|
prefix = GetPromptPrefix(line)
|
||
|
if prefix is None: # Non code block
|
||
|
flag = 0
|
||
|
startLineNo = lineNo
|
||
|
while startLineNo > 0:
|
||
|
if GetPromptPrefix(self.DoGetLine(startLineNo - 1)) is not None:
|
||
|
break # there _is_ a prompt
|
||
|
startLineNo = startLineNo - 1
|
||
|
endLineNo = lineNo
|
||
|
while endLineNo < maxLineNo:
|
||
|
if GetPromptPrefix(self.DoGetLine(endLineNo + 1)) is not None:
|
||
|
break # there _is_ a prompt
|
||
|
endLineNo = endLineNo + 1
|
||
|
else: # Code block
|
||
|
flag = 1
|
||
|
startLineNo = lineNo
|
||
|
while startLineNo > 0 and prefix != str(sys.ps1):
|
||
|
prefix = GetPromptPrefix(self.DoGetLine(startLineNo - 1))
|
||
|
if prefix is None:
|
||
|
break
|
||
|
# there is no prompt.
|
||
|
startLineNo = startLineNo - 1
|
||
|
endLineNo = lineNo
|
||
|
while endLineNo < maxLineNo:
|
||
|
prefix = GetPromptPrefix(self.DoGetLine(endLineNo + 1))
|
||
|
if prefix is None:
|
||
|
break # there is no prompt
|
||
|
if prefix == str(sys.ps1):
|
||
|
break # this is another command
|
||
|
endLineNo = endLineNo + 1
|
||
|
# continue until end of buffer, or no prompt
|
||
|
return (startLineNo, endLineNo, flag)
|
||
|
|
||
|
def ExtractCommand(self, lines):
|
||
|
start, end = lines
|
||
|
retList = []
|
||
|
while end >= start:
|
||
|
thisLine = self.DoGetLine(end)
|
||
|
promptLen = len(GetPromptPrefix(thisLine))
|
||
|
retList = [thisLine[promptLen:]] + retList
|
||
|
end = end - 1
|
||
|
return retList
|
||
|
|
||
|
def OutputGrab(self):
|
||
|
# import win32traceutil; return
|
||
|
self.oldStdOut = sys.stdout
|
||
|
self.oldStdErr = sys.stderr
|
||
|
sys.stdout = self
|
||
|
sys.stderr = self
|
||
|
self.flush()
|
||
|
|
||
|
def OutputRelease(self):
|
||
|
# a command may have overwritten these - only restore if not.
|
||
|
if self.oldStdOut is not None:
|
||
|
if sys.stdout == self:
|
||
|
sys.stdout = self.oldStdOut
|
||
|
if self.oldStdErr is not None:
|
||
|
if sys.stderr == self:
|
||
|
sys.stderr = self.oldStdErr
|
||
|
self.oldStdOut = None
|
||
|
self.oldStdErr = None
|
||
|
self.flush()
|
||
|
|
||
|
###################################
|
||
|
#
|
||
|
# Message/Command/Key Hooks.
|
||
|
#
|
||
|
# Enter key handler
|
||
|
#
|
||
|
def ProcessEnterEvent(self, event):
|
||
|
# If autocompletion has been triggered, complete and do not process event
|
||
|
if self.SCIAutoCActive():
|
||
|
self.SCIAutoCComplete()
|
||
|
self.SCICancel()
|
||
|
return
|
||
|
|
||
|
self.SCICancel()
|
||
|
# First, check for an error message
|
||
|
haveGrabbedOutput = 0
|
||
|
if self.HandleSpecialLine():
|
||
|
return 0
|
||
|
|
||
|
lineNo = self.LineFromChar()
|
||
|
start, end, isCode = self.GetBlockBoundary(lineNo)
|
||
|
# If we are not in a code block just go to the prompt (or create a new one)
|
||
|
if not isCode:
|
||
|
self.AppendToPrompt([])
|
||
|
win32ui.SetStatusText(win32ui.LoadString(afxres.AFX_IDS_IDLEMESSAGE))
|
||
|
return
|
||
|
|
||
|
lines = self.ExtractCommand((start, end))
|
||
|
|
||
|
# If we are in a code-block, but it isnt at the end of the buffer
|
||
|
# then copy it to the end ready for editing and subsequent execution
|
||
|
if end != self.GetLineCount() - 1:
|
||
|
win32ui.SetStatusText("Press ENTER to execute command")
|
||
|
self.AppendToPrompt(lines)
|
||
|
self.SetSel(-2)
|
||
|
return
|
||
|
|
||
|
# If SHIFT held down, we want new code here and now!
|
||
|
bNeedIndent = (
|
||
|
win32api.GetKeyState(win32con.VK_SHIFT) < 0
|
||
|
or win32api.GetKeyState(win32con.VK_CONTROL) < 0
|
||
|
)
|
||
|
if bNeedIndent:
|
||
|
self.ReplaceSel("\n")
|
||
|
else:
|
||
|
self.SetSel(-2)
|
||
|
self.ReplaceSel("\n")
|
||
|
source = "\n".join(lines)
|
||
|
while source and source[-1] in "\t ":
|
||
|
source = source[:-1]
|
||
|
self.OutputGrab() # grab the output for the command exec.
|
||
|
try:
|
||
|
if self.interp.runsource(
|
||
|
source, "<interactive input>"
|
||
|
): # Need more input!
|
||
|
bNeedIndent = 1
|
||
|
else:
|
||
|
# If the last line isnt empty, append a newline
|
||
|
if self.history is not None:
|
||
|
self.history.history_store(source)
|
||
|
self.AppendToPrompt([])
|
||
|
win32ui.SetStatusText(
|
||
|
win32ui.LoadString(afxres.AFX_IDS_IDLEMESSAGE)
|
||
|
)
|
||
|
# win32ui.SetStatusText('Successfully executed statement')
|
||
|
finally:
|
||
|
self.OutputRelease()
|
||
|
if bNeedIndent:
|
||
|
win32ui.SetStatusText("Ready to continue the command")
|
||
|
# Now attempt correct indentation (should use IDLE?)
|
||
|
curLine = self.DoGetLine(lineNo)[len(sys.ps2) :]
|
||
|
pos = 0
|
||
|
indent = ""
|
||
|
while len(curLine) > pos and curLine[pos] in string.whitespace:
|
||
|
indent = indent + curLine[pos]
|
||
|
pos = pos + 1
|
||
|
if _is_block_opener(curLine):
|
||
|
indent = indent + "\t"
|
||
|
elif _is_block_closer(curLine):
|
||
|
indent = indent[:-1]
|
||
|
# use ReplaceSel to ensure it goes at the cursor rather than end of buffer.
|
||
|
self.ReplaceSel(sys.ps2 + indent)
|
||
|
return 0
|
||
|
|
||
|
# ESC key handler
|
||
|
def ProcessEscEvent(self, event):
|
||
|
# Implement a cancel.
|
||
|
if self.SCIAutoCActive() or self.SCICallTipActive():
|
||
|
self.SCICancel()
|
||
|
else:
|
||
|
win32ui.SetStatusText("Cancelled.")
|
||
|
self.AppendToPrompt(("",))
|
||
|
return 0
|
||
|
|
||
|
def OnSelectBlock(self, command, code):
|
||
|
lineNo = self.LineFromChar()
|
||
|
start, end, isCode = self.GetBlockBoundary(lineNo)
|
||
|
startIndex = self.LineIndex(start)
|
||
|
endIndex = self.LineIndex(end + 1) - 2 # skip \r + \n
|
||
|
if endIndex < 0: # must be beyond end of buffer
|
||
|
endIndex = -2 # self.Length()
|
||
|
self.SetSel(startIndex, endIndex)
|
||
|
|
||
|
def OnEditCopyCode(self, command, code):
|
||
|
"""Sanitizes code from interactive window, removing prompts and output,
|
||
|
and inserts it in the clipboard."""
|
||
|
code = self.GetSelText()
|
||
|
lines = code.splitlines()
|
||
|
out_lines = []
|
||
|
for line in lines:
|
||
|
if line.startswith(sys.ps1):
|
||
|
line = line[len(sys.ps1) :]
|
||
|
out_lines.append(line)
|
||
|
elif line.startswith(sys.ps2):
|
||
|
line = line[len(sys.ps2) :]
|
||
|
out_lines.append(line)
|
||
|
out_code = os.linesep.join(out_lines)
|
||
|
win32clipboard.OpenClipboard()
|
||
|
try:
|
||
|
win32clipboard.SetClipboardData(
|
||
|
win32clipboard.CF_UNICODETEXT, str(out_code)
|
||
|
)
|
||
|
finally:
|
||
|
win32clipboard.CloseClipboard()
|
||
|
|
||
|
def OnEditExecClipboard(self, command, code):
|
||
|
"""Executes python code directly from the clipboard."""
|
||
|
win32clipboard.OpenClipboard()
|
||
|
try:
|
||
|
code = win32clipboard.GetClipboardData(win32clipboard.CF_UNICODETEXT)
|
||
|
finally:
|
||
|
win32clipboard.CloseClipboard()
|
||
|
|
||
|
code = code.replace("\r\n", "\n") + "\n"
|
||
|
try:
|
||
|
o = compile(code, "<clipboard>", "exec")
|
||
|
exec(o, __main__.__dict__)
|
||
|
except:
|
||
|
traceback.print_exc()
|
||
|
|
||
|
def GetRightMenuItems(self):
|
||
|
# Just override parents
|
||
|
ret = []
|
||
|
flags = 0
|
||
|
ret.append((flags, win32ui.ID_EDIT_UNDO, "&Undo"))
|
||
|
ret.append(win32con.MF_SEPARATOR)
|
||
|
ret.append((flags, win32ui.ID_EDIT_CUT, "Cu&t"))
|
||
|
ret.append((flags, win32ui.ID_EDIT_COPY, "&Copy"))
|
||
|
|
||
|
start, end = self.GetSel()
|
||
|
if start != end:
|
||
|
ret.append((flags, ID_EDIT_COPY_CODE, "Copy code without prompts"))
|
||
|
if win32clipboard.IsClipboardFormatAvailable(win32clipboard.CF_UNICODETEXT):
|
||
|
ret.append(
|
||
|
(flags, ID_EDIT_EXEC_CLIPBOARD, "Execute python code from clipboard")
|
||
|
)
|
||
|
|
||
|
ret.append((flags, win32ui.ID_EDIT_PASTE, "&Paste"))
|
||
|
ret.append(win32con.MF_SEPARATOR)
|
||
|
ret.append((flags, win32ui.ID_EDIT_SELECT_ALL, "&Select all"))
|
||
|
ret.append((flags, win32ui.ID_EDIT_SELECT_BLOCK, "Select &block"))
|
||
|
ret.append((flags, win32ui.ID_VIEW_WHITESPACE, "View &Whitespace"))
|
||
|
return ret
|
||
|
|
||
|
def MDINextEvent(self, event):
|
||
|
win32ui.GetMainFrame().MDINext(0)
|
||
|
|
||
|
def MDIPrevEvent(self, event):
|
||
|
win32ui.GetMainFrame().MDINext(0)
|
||
|
|
||
|
def WindowBackEvent(self, event):
|
||
|
parent = self.GetParentFrame()
|
||
|
if parent == win32ui.GetMainFrame():
|
||
|
# It is docked.
|
||
|
try:
|
||
|
wnd, isactive = parent.MDIGetActive()
|
||
|
wnd.SetFocus()
|
||
|
except win32ui.error:
|
||
|
# No MDI window active!
|
||
|
pass
|
||
|
else:
|
||
|
# Normal Window
|
||
|
try:
|
||
|
lastActive = self.GetParentFrame().lastActive
|
||
|
# If the window is invalid, reset it.
|
||
|
if lastActive is not None and (
|
||
|
lastActive._obj_ is None or lastActive.GetSafeHwnd() == 0
|
||
|
):
|
||
|
lastActive = self.GetParentFrame().lastActive = None
|
||
|
win32ui.SetStatusText("The last active Window has been closed.")
|
||
|
except AttributeError:
|
||
|
print("Can't find the last active window!")
|
||
|
lastActive = None
|
||
|
if lastActive is not None:
|
||
|
lastActive.MDIActivate()
|
||
|
|
||
|
|
||
|
class InteractiveView(InteractiveCore, winout.WindowOutputView):
|
||
|
def __init__(self, doc):
|
||
|
InteractiveCore.__init__(self)
|
||
|
winout.WindowOutputView.__init__(self, doc)
|
||
|
self.encoding = pywin.default_scintilla_encoding
|
||
|
|
||
|
def _MakeColorizer(self):
|
||
|
return InteractiveFormatter(self)
|
||
|
|
||
|
def OnInitialUpdate(self):
|
||
|
winout.WindowOutputView.OnInitialUpdate(self)
|
||
|
self.SetWordWrap()
|
||
|
self.Init()
|
||
|
|
||
|
def HookHandlers(self):
|
||
|
winout.WindowOutputView.HookHandlers(self)
|
||
|
InteractiveCore.HookHandlers(self)
|
||
|
|
||
|
|
||
|
class CInteractivePython(winout.WindowOutput):
|
||
|
def __init__(self, makeDoc=None, makeFrame=None):
|
||
|
self.IsFinalDestroy = 0
|
||
|
winout.WindowOutput.__init__(
|
||
|
self,
|
||
|
sectionProfile,
|
||
|
sectionProfile,
|
||
|
winout.flags.WQ_LINE,
|
||
|
1,
|
||
|
None,
|
||
|
makeDoc,
|
||
|
makeFrame,
|
||
|
InteractiveView,
|
||
|
)
|
||
|
self.Create()
|
||
|
|
||
|
def OnViewDestroy(self, view):
|
||
|
if self.IsFinalDestroy:
|
||
|
view.OutputRelease()
|
||
|
winout.WindowOutput.OnViewDestroy(self, view)
|
||
|
|
||
|
def Close(self):
|
||
|
self.IsFinalDestroy = 1
|
||
|
winout.WindowOutput.Close(self)
|
||
|
|
||
|
|
||
|
class InteractiveFrame(winout.WindowOutputFrame):
|
||
|
def __init__(self):
|
||
|
self.lastActive = None
|
||
|
winout.WindowOutputFrame.__init__(self)
|
||
|
|
||
|
def OnMDIActivate(self, bActive, wndActive, wndDeactive):
|
||
|
if bActive:
|
||
|
self.lastActive = wndDeactive
|
||
|
|
||
|
|
||
|
######################################################################
|
||
|
##
|
||
|
## Dockable Window Support
|
||
|
##
|
||
|
######################################################################
|
||
|
ID_DOCKED_INTERACTIVE_CONTROLBAR = 0xE802
|
||
|
|
||
|
DockedInteractiveViewParent = InteractiveView
|
||
|
|
||
|
|
||
|
class DockedInteractiveView(DockedInteractiveViewParent):
|
||
|
def HookHandlers(self):
|
||
|
DockedInteractiveViewParent.HookHandlers(self)
|
||
|
self.HookMessage(self.OnSetFocus, win32con.WM_SETFOCUS)
|
||
|
self.HookMessage(self.OnKillFocus, win32con.WM_KILLFOCUS)
|
||
|
|
||
|
def OnSetFocus(self, msg):
|
||
|
self.GetParentFrame().SetActiveView(self)
|
||
|
return 1
|
||
|
|
||
|
def OnKillFocus(self, msg):
|
||
|
# If we are losing focus to another in this app, reset the main frame's active view.
|
||
|
hwnd = wparam = msg[2]
|
||
|
try:
|
||
|
wnd = win32ui.CreateWindowFromHandle(hwnd)
|
||
|
reset = wnd.GetTopLevelFrame() == self.GetTopLevelFrame()
|
||
|
except win32ui.error:
|
||
|
reset = 0 # Not my window
|
||
|
if reset:
|
||
|
self.GetParentFrame().SetActiveView(None)
|
||
|
return 1
|
||
|
|
||
|
def OnDestroy(self, msg):
|
||
|
newSize = self.GetWindowPlacement()[4]
|
||
|
pywin.framework.app.SaveWindowSize("Interactive Window", newSize, "docked")
|
||
|
try:
|
||
|
if self.GetParentFrame().GetActiveView == self:
|
||
|
self.GetParentFrame().SetActiveView(None)
|
||
|
except win32ui.error:
|
||
|
pass
|
||
|
try:
|
||
|
if win32ui.GetMainFrame().GetActiveView() == self:
|
||
|
win32ui.GetMainFrame().SetActiveView(None)
|
||
|
except win32ui.error:
|
||
|
pass
|
||
|
return DockedInteractiveViewParent.OnDestroy(self, msg)
|
||
|
|
||
|
|
||
|
class CDockedInteractivePython(CInteractivePython):
|
||
|
def __init__(self, dockbar):
|
||
|
self.bFirstCreated = 0
|
||
|
self.dockbar = dockbar
|
||
|
CInteractivePython.__init__(self)
|
||
|
|
||
|
def NeedRecreateWindow(self):
|
||
|
if self.bCreating:
|
||
|
return 0
|
||
|
try:
|
||
|
frame = win32ui.GetMainFrame()
|
||
|
if frame.closing:
|
||
|
return 0 # Dieing!
|
||
|
except (win32ui.error, AttributeError):
|
||
|
return 0 # The app is dieing!
|
||
|
try:
|
||
|
cb = frame.GetControlBar(ID_DOCKED_INTERACTIVE_CONTROLBAR)
|
||
|
return not cb.IsWindowVisible()
|
||
|
except win32ui.error:
|
||
|
return 1 # Control bar does not exist!
|
||
|
|
||
|
def RecreateWindow(self):
|
||
|
try:
|
||
|
dockbar = win32ui.GetMainFrame().GetControlBar(
|
||
|
ID_DOCKED_INTERACTIVE_CONTROLBAR
|
||
|
)
|
||
|
win32ui.GetMainFrame().ShowControlBar(dockbar, 1, 1)
|
||
|
except win32ui.error:
|
||
|
CreateDockedInteractiveWindow()
|
||
|
|
||
|
def Create(self):
|
||
|
self.bCreating = 1
|
||
|
doc = InteractiveDocument(None, self.DoCreateDoc())
|
||
|
view = DockedInteractiveView(doc)
|
||
|
defRect = pywin.framework.app.LoadWindowSize("Interactive Window", "docked")
|
||
|
if defRect[2] - defRect[0] == 0:
|
||
|
defRect = 0, 0, 500, 200
|
||
|
style = win32con.WS_CHILD | win32con.WS_VISIBLE | win32con.WS_BORDER
|
||
|
id = 1050 # win32ui.AFX_IDW_PANE_FIRST
|
||
|
view.CreateWindow(self.dockbar, id, style, defRect)
|
||
|
view.OnInitialUpdate()
|
||
|
self.bFirstCreated = 1
|
||
|
|
||
|
self.currentView = doc.GetFirstView()
|
||
|
self.bCreating = 0
|
||
|
if self.title:
|
||
|
doc.SetTitle(self.title)
|
||
|
|
||
|
|
||
|
# The factory we pass to the dockable window support.
|
||
|
def InteractiveViewCreator(parent):
|
||
|
global edit
|
||
|
edit = CDockedInteractivePython(parent)
|
||
|
return edit.currentView
|
||
|
|
||
|
|
||
|
def CreateDockedInteractiveWindow():
|
||
|
# Later, the DockingBar should be capable of hosting multiple
|
||
|
# children.
|
||
|
from pywin.docking.DockingBar import DockingBar
|
||
|
|
||
|
bar = DockingBar()
|
||
|
creator = InteractiveViewCreator
|
||
|
bar.CreateWindow(
|
||
|
win32ui.GetMainFrame(),
|
||
|
creator,
|
||
|
"Interactive Window",
|
||
|
ID_DOCKED_INTERACTIVE_CONTROLBAR,
|
||
|
)
|
||
|
bar.SetBarStyle(
|
||
|
bar.GetBarStyle()
|
||
|
| afxres.CBRS_TOOLTIPS
|
||
|
| afxres.CBRS_FLYBY
|
||
|
| afxres.CBRS_SIZE_DYNAMIC
|
||
|
)
|
||
|
bar.EnableDocking(afxres.CBRS_ALIGN_ANY)
|
||
|
win32ui.GetMainFrame().DockControlBar(bar, afxres.AFX_IDW_DOCKBAR_BOTTOM)
|
||
|
|
||
|
|
||
|
######################################################################
|
||
|
#
|
||
|
# The public interface to this module.
|
||
|
#
|
||
|
######################################################################
|
||
|
# No extra functionality now, but maybe later, so
|
||
|
# publicize these names.
|
||
|
InteractiveDocument = winout.WindowOutputDocument
|
||
|
|
||
|
# We remember our one and only interactive window in the "edit" variable.
|
||
|
edit = None
|
||
|
|
||
|
|
||
|
def CreateInteractiveWindowUserPreference(makeDoc=None, makeFrame=None):
|
||
|
"""Create some sort of interactive window if the user's preference say we should."""
|
||
|
bCreate = LoadPreference("Show at startup", 1)
|
||
|
if bCreate:
|
||
|
CreateInteractiveWindow(makeDoc, makeFrame)
|
||
|
|
||
|
|
||
|
def CreateInteractiveWindow(makeDoc=None, makeFrame=None):
|
||
|
"""Create a standard or docked interactive window unconditionally"""
|
||
|
assert edit is None, "Creating second interactive window!"
|
||
|
bDocking = LoadPreference("Docking", 0)
|
||
|
if bDocking:
|
||
|
CreateDockedInteractiveWindow()
|
||
|
else:
|
||
|
CreateMDIInteractiveWindow(makeDoc, makeFrame)
|
||
|
assert edit is not None, "Created interactive window, but did not set the global!"
|
||
|
edit.currentView.SetFocus()
|
||
|
|
||
|
|
||
|
def CreateMDIInteractiveWindow(makeDoc=None, makeFrame=None):
|
||
|
"""Create a standard (non-docked) interactive window unconditionally"""
|
||
|
global edit
|
||
|
if makeDoc is None:
|
||
|
makeDoc = InteractiveDocument
|
||
|
if makeFrame is None:
|
||
|
makeFrame = InteractiveFrame
|
||
|
edit = CInteractivePython(makeDoc=makeDoc, makeFrame=makeFrame)
|
||
|
|
||
|
|
||
|
def DestroyInteractiveWindow():
|
||
|
"""Destroy the interactive window.
|
||
|
This is different to Closing the window,
|
||
|
which may automatically re-appear. Once destroyed, it can never be recreated,
|
||
|
and a complete new instance must be created (which the various other helper
|
||
|
functions will then do after making this call
|
||
|
"""
|
||
|
global edit
|
||
|
if edit is not None and edit.currentView is not None:
|
||
|
if edit.currentView.GetParentFrame() == win32ui.GetMainFrame():
|
||
|
# It is docked - do nothing now (this is only called at shutdown!)
|
||
|
pass
|
||
|
else:
|
||
|
# It is a standard window - call Close on the container.
|
||
|
edit.Close()
|
||
|
edit = None
|
||
|
|
||
|
|
||
|
def CloseInteractiveWindow():
|
||
|
"""Close the interactive window, allowing it to be re-created on demand."""
|
||
|
global edit
|
||
|
if edit is not None and edit.currentView is not None:
|
||
|
if edit.currentView.GetParentFrame() == win32ui.GetMainFrame():
|
||
|
# It is docked, just hide the dock bar.
|
||
|
frame = win32ui.GetMainFrame()
|
||
|
cb = frame.GetControlBar(ID_DOCKED_INTERACTIVE_CONTROLBAR)
|
||
|
frame.ShowControlBar(cb, 0, 1)
|
||
|
else:
|
||
|
# It is a standard window - destroy the frame/view, allowing the object itself to remain.
|
||
|
edit.currentView.GetParentFrame().DestroyWindow()
|
||
|
|
||
|
|
||
|
def ToggleInteractiveWindow():
|
||
|
"""If the interactive window is visible, hide it, otherwise show it."""
|
||
|
if edit is None:
|
||
|
CreateInteractiveWindow()
|
||
|
else:
|
||
|
if edit.NeedRecreateWindow():
|
||
|
edit.RecreateWindow()
|
||
|
else:
|
||
|
# Close it, allowing a reopen.
|
||
|
CloseInteractiveWindow()
|
||
|
|
||
|
|
||
|
def ShowInteractiveWindow():
|
||
|
"""Shows (or creates if necessary) an interactive window"""
|
||
|
if edit is None:
|
||
|
CreateInteractiveWindow()
|
||
|
else:
|
||
|
if edit.NeedRecreateWindow():
|
||
|
edit.RecreateWindow()
|
||
|
else:
|
||
|
parent = edit.currentView.GetParentFrame()
|
||
|
if parent == win32ui.GetMainFrame(): # It is docked.
|
||
|
edit.currentView.SetFocus()
|
||
|
else: # It is a "normal" window
|
||
|
edit.currentView.GetParentFrame().AutoRestore()
|
||
|
win32ui.GetMainFrame().MDIActivate(edit.currentView.GetParentFrame())
|
||
|
|
||
|
|
||
|
def IsInteractiveWindowVisible():
|
||
|
return edit is not None and not edit.NeedRecreateWindow()
|