如何在 pip install 上的 jupyter 中定义自定义魔法

How can I define custom magics in jupyter on pip install

这是此 的扩展 我想在通过 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)

由于您的内核派生自 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 class 定义.