sage-list-packages

#!/usr/bin/env python3
r"""
Script to list the Sage packages

This is script can be called with one argument which might be either
"all", "standard", "optional", or "experimental". It is mostly a
script interface to sage_setup.packages.list_packages.
"""
import os
import argparse

if "SAGE_ROOT" not in os.environ:
    raise RuntimeError("The environment variable SAGE_ROOT must be set.")
SAGE_ROOT = os.environ["SAGE_ROOT"]

from sage.misc.package import list_packages

# Input parsing #
#################

parser = argparse.ArgumentParser(description="List Sage's packages")
parser.add_argument('category', choices=['all', 'standard', 'optional',
                                         'experimental', 'installed'],
                    metavar="category",
                    help="The type of packages. Can be 'all', 'standard', "
                    "'optional', or 'experimental'.")
parser.add_argument('--installed-only', dest='installed_only',
                    default=False, action='store_true',
                    help='only display installed packages')
parser.add_argument('--not-installed-only', dest='not_installed_only',
                    default=False, action='store_true',
                    help='only display non installed packages')
parser.add_argument('--dump', dest='dump', default=False, action='store_true',
                    help='computer-friendly format')
parser.add_argument('--no-version', dest='version', default=True,
                    action='store_false',
                    help='no version number')
parser.add_argument('--local', dest='local', default=False,
                    action='store_true',
                    help='only read local data')

args = vars(parser.parse_args())

# Get the data #
################

if args['category'] == 'installed':
    WARN = "*" * 60 + "\n"
    WARN += 'The "installed" category is deprecated. Use\n'
    WARN += '"list-packages all --installed-only" instead\n'
    WARN += "*" * 60

    args['category'] = 'all'
    args['installed_only'] = True
else:
    WARN = None

if args['installed_only'] and args['not_installed_only']:
    raise ValueError("only one of --installed-only or --not-installed-only can be specified")

# set the output format
if args['version']:
    if args['category'] == 'installed':
        format_string = "{installed_version}"
    else:
        format_string = "{remote_version} ({installed_version})"
else:
    args['dump'] = True
    format_string = ''

if args['dump']:
    format_string = "{name} " + format_string
else:
    format_string = "{name:.<40}" + format_string
    print(format_string.format(name="[package]", installed_version="[version]",
                               remote_version="[latest version]"))
    print()

# make the list of packages
if args['category'] == 'all':
    L = list(list_packages(local=True, ignore_URLError=True).values())
elif args['category'] == 'optional':
    L = list(list_packages('optional', local=args['local'], ignore_URLError=True).values())
else:
    L = list(list_packages(args['category'], local=args['local'], ignore_URLError=True).values())

# possible filter by installed/not installed
if WARN:
    print(WARN)
if args['installed_only']:
    L = [pkg for pkg in L if pkg.is_installed()]
elif args['not_installed_only']:
    L = [pkg for pkg in L if not pkg.is_installed()]

L.sort(key=lambda pkg: pkg.name)

# print (while getting rid of None in versions)
for pkg in L:
    pkg = pkg._replace(installed_version=pkg.installed_version or 'not_installed',
                       remote_version=pkg.remote_version or '?')
    print(format_string.format(**pkg._asdict()))
if WARN:
    print(WARN)