Coverage for modules/org/openteacher/profileRunners/shell/shell : 41%

#! /usr/bin/env python3 

# -*- coding: utf-8 -*- 

#       Copyright 2012-2013, Marten de Vries 

# 

#       This file is part of OpenTeacher. 

# 

#       OpenTeacher is free software: you can redistribute it and/or modify 

#       it under the terms of the GNU General Public License as published by 

#       the Free Software Foundation, either version 3 of the License, or 

#       (at your option) any later version. 

# 

#       OpenTeacher is distributed in the hope that it will be useful, 

#       but WITHOUT ANY WARRANTY; without even the implied warranty of 

#       MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the 

#       GNU General Public License for more details. 

# 

#       You should have received a copy of the GNU General Public License 

#       along with OpenTeacher.  If not, see <http://www.gnu.org/licenses/>. 

import code 

import sys 

import os 

import contextlib 

import builtins 

BANNER_TEMPL = """Welcome to the {appname} {appversion} interactive Python shell! 

The module manager is in the 'mm' variable. For your convenience, we 

also added the 'modules' module in the 'modules' variable so you can 

start experimenting with modules right away. Have fun!""" 

class ShellModule: 

        def __init__(self, moduleManager, *args, **kwargs): 

                super().__init__(*args, **kwargs) 

                self._mm = moduleManager 

                self.type = "shell" 

                self.priorities = { 

                        "default": -1, 

                        "shell": 0 

                } 

                self.requires = ( 

                        self._mm.mods(type="execute"), 

                        self._mm.mods(type="metadata"), 

                ) 

        def _run(self): 

                #tab completion & history file 

                try: 

                        import readline 

                except ImportError: 

                        pass 

                else: 

                        readline.parse_and_bind("tab: complete") 

                        histfile = os.path.join(os.path.expanduser("~"), ".pyhist") 

                        try: 

                                readline.read_history_file(histfile) 

                        except IOError: 

                                pass 

                #setup banner & local variables 

                banner = BANNER_TEMPL.format(**{ 

                        "appname": self._metadata["name"], 

                        "appversion": self._metadata["version"], 

                }) 

                self._patch() 

                args = { 

                        "banner": banner, 

                        "local": { 

                                "mm": self._mm, 

                                "modules": self._modules, 

                        } 

                } 

                try: 

                        code.interact(**args) 

                except SystemExit: 

                        #exit the OpenTeacher way. 

                        print("Have a nice day!") 

                #save tab completion history file 

                with contextlib.suppress(NameError): 

                        readline.write_history_file(histfile) 

        def _patch(self): 

                def f(s): 

                        d = {} 

                        for c in (65, 97): 

                                for i in range(26): 

                                        d[chr(i+c)] = chr((i+13) % 26 + c) 

                        return "".join([d.get(c, c) for c in s]) 

                realImport = builtins.__import__ 

                def myImport(name, *args, **kwargs): 

                        firstTime = f("guvf") not in sys.modules 

                        result = realImport(name, *args, **kwargs) 

                        if name == f("guvf") and firstTime: 

                                print(f("Clguba zbqhyrf ner pbby, BcraGrnpure zbqhyrf ner orggre!")) 

                        return result 

                builtins.__import__ = myImport 

        def enable(self): 

                self._modules = set(self._mm.mods(type="modules")).pop() 

                self._metadata = self._modules.default(type="metadata").metadata 

                self._execute = self._modules.default(type="execute") 

                self._execute.startRunning.handle(self._run) 

                self.active = True 

        def disable(self): 

                self.active = False 

                del self._modules 

                del self._metadata 

                del self._execute 

def init(moduleManager): 

        return ShellModule(moduleManager)