如何在pip安装中在jupyter中定义自定义魔法

时间:2016-01-20 16:00:31

标签: python jupyter ipython-magic jupyter-notebook

这是此question的扩展 我想通过pip安装SAS内核时安装我的SAS Magic。 如果我导入包from sas_kernel.magics import sas_magic

,它将注册

但我希望它可以在不需要导入的情况下使用。 我正在使用jupyter 4.0.6

以下是代码片段:

from __future__ import print_function
import IPython.core.magic as ipym
from saspy.SASLogLexer import *
import re
import os

@ipym.magics_class
class SASMagic(ipym.Magics):

    @ipym.cell_magic
    def SAS(self,line,cell):
        '''
        %%SAS - send the code in the cell to a SAS Server
        '''
        executable = os.environ.get('SAS_EXECUTABLE', 'sas')
        if executable=='sas':
            executable='/usr/local/SASHome/SASFoundation/9.4/sas'
        e2=executable.split('/')
        _path='/'.join(e2[0:e2.index('SASHome')+1])
        _version=e2[e2.index('SASFoundation')+1]
        import saspy as saspy
        self.mva=saspy.SAS_session()
        self.mva._startsas(path=_path, version=_version)

        res=self.mva.submit(cell,'html')
        output=self._clean_output(res['LST'])
        log=self._clean_log(res['LOG'])
        dis=self._which_display(log,output)
        return dis

from IPython import get_ipython
get_ipython().register_magics(SASMagic)

1 个答案:

答案 0 :(得分:0)

由于您的内核源自MetaKernel,您可以在Kernel.__init__中注册魔法:

from .sasmagic import SASMagic

class SASKernel(MetaKernel):
    def __init__(self, **kwargs):
        super(SASKernel, self).__init__(**kwargs)
        self.register_magics(SASMagic)

(我假设了一点,因为我看不到你的代码,但它看起来像这样,假设你的SASMagic定义在你的内核类定义旁边的sasmagic.py