mbk/pesterchum
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Console cleanup.

Browse source
This commit is contained in:
karxi 2016-12-22 17:27:06 -05:00
parent 6944cc25a0
commit 1f390ffda9
1 changed files with 11 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

17
console.py
View file

@ -124,14 +124,19 @@ class ConsoleWindow(QtGui.QDialog):
"MAINWIN": self.mainwindow,
"PCONFIG": self.mainwindow.config
}
_CUSTOM_ENV_USE = []
cenv = pchum.__dict__
for k in _CUSTOM_ENV:
if k not in env:
# Don't overwrite anything!
env[k] = _CUSTOM_ENV[k]
if k not in cenv:
cenv[k] = _CUSTOM_ENV[k]
_CUSTOM_ENV_USE.append(k)
else:
warn = "Console environment item {!r} already exists in main."
# Don't overwrite anything!
warn = "Console environment item {!r} already exists in CENV."
warn.format(k)
logging.warning(warn)
# Because all we did was change a linked AttrDict, we should be fine
# here.
# Display the input we provided
self.addMessage(scriptstr, 1)
@ -155,9 +160,9 @@ class ConsoleWindow(QtGui.QDialog):
# Try to clean us out of globals - this might be disabled
# later.
for k in _CUSTOM_ENV:
for k in _CUSTOM_ENV_USED:
# Remove the key we added.
env.pop(k, None)
cenv.pop(k, None)
def write(self, data):
# Replaces sys.stdout briefly