Can I define optional packages in setuptools?

前端 未结 2 1808
星月不相逢
星月不相逢 2021-02-14 03:33

Currently one of my packages requires a JSON parser/encoder, and is designed to use simplejson if available falling back to the json module (in the sta

2条回答
  •  一向
    一向 (楼主)
    2021-02-14 04:21

    optional packages at installation time.

    I am assuming you are talking about your setup.py script. You could change it to have:

    # mypackage/setup.py
    
    extras = {
       'with_simplejson': ['simplejson>=3.5.3']
    }
    
    setup(
        ...
        extras_require=extras,
        ...)
    

    then you can do either of:

    • pip install mypackage,
    • pip install mypackage[with_simplejson]

    with the latter installing simplejson>=3.5.3.

    Instead of trying to install everything and fallback to a known good version, you would want to install the subset of packages you know work.

    optional packages at execution time.

    Once you have two different sets of packages that could be installed, you need to make sure you can use them if they are available. E.g. for your json import:

    try:
        # helpful comment saying this should be faster.
        import simplejson as json
    except ImportError:
        import json
    

    Another more complex example:

    try:
        # xml is dangerous
        from defusedxml.cElementTree import parse
    except ImportError:
        try:
            # cElementTree is not available in older python
            from xml.cElementTree import parse
        except ImportError:
            from xml.ElementTree import parse
    

    But you can also find this pattern in some packages:

    try:
        optional_package = None
        import optional.package as optional_package
    except ImportError:
        pass
    
    ...
    
    if optional_package:
        # do addtional behavior
    

提交回复
热议问题