轻松使用pip（基于sh）

简介

1. 有时候直接使用pip不方便，我就想做个简单的餐单
2. 同时实现加速、sudo切换、失败后轻松重新下载
3. 基于sh. 觉得subprocess有点麻烦

代码

#!/usr/bin/env python3
# -*- coding: utf-8 -*-


import sh
import collections

out = sh.pip3.list('--format', 'columns', '--outdate', '-i', 'https://pypi.tuna.tsinghua.edu.cn/simple')

Package = collections.namedtuple('package', ['name', 'version', 'latest', 'type_'])

packages = [Package._make(package.split()) for package in out.split('\n')[2:-1] if package]
display(packages)

def display(packages):
    print('''
        Name  Version  Latest
        ----------------------
        ''')
    for package in packages:
        print('    ', package.name, package.version, package.latest)

sudo = False
current = []
def install(source='tsinghua'):
    global packages, sudo, current
    print('''====Type one or some names of packages wantted.====
        q: quit
        s: sudo mode
        u: usual mode
        l: list the packages
        m: install the previous packages again
        ''')
    while True:
        ps = input('>=> ')
        if ps == 'q':
            print('Bye.')
            break
        elif ps == 's':
            sudo = 1
            print('in sudo mode')
        elif ps == 'u':
            sudo = 0
            print('not in sudo mode')
        elif ps == 'l':
            display(packages)
        elif ps == 'm':
            if not current:
                print('m could not be used now!')
                continue
            params = 'install', '--upgrade', '-i', 'https://pypi.tuna.tsinghua.edu.cn/simple'
            cmd = sh.pip3
            if sudo:
                cmd = sh.contrib.sudo.pip3
            for p in current:
                try:
                    out = cmd(*params, p)
                except Exception as e:
                    print(e)
                else:
                    packages = [package for package in packages if package.name not in ps.split()]
        else:
            current = ps.split()
            for c in current:
                if not any(c == package.name for package in packages):
                    print(f'{c} is not in the list of packages!')
                    current.remove(c)
                else:
                    print(f'{c} will be installed.')

            params = 'install', '--upgrade', '-i', 'https://pypi.tuna.tsinghua.edu.cn/simple'
            cmd = sh.pip3
            if sudo:
                cmd = sh.contrib.sudo.pip3
            for p in current:
                try:
                    out = cmd(*params, p)
                    print(f'{p} has been installed.')
                except Exception as e:
                    print(e)
                else:
                    packages = [package for package in packages if package.name not in ps.split()]

install()

脚本运行效果

     ...
     pymongo 3.9.0 3.10.1
     PyOpenGL 3.1.4 3.1.5
     pyrsistent 0.15.6 0.15.7
     ...
     zope.interface 4.6.0 4.7.1
====Type one or some names of packages wantted.====
        q: quit
        s: sudo mode
        u: usual mode
        l: list the packages
        m: install the previous packages again

>=>

shell 版本（DEPRECATED）

#!/bin/sh

:<<EOF
update the packages of python

s: sudo, u: not sudo -- udo, l: show list again, m: install last module
h: help
EOF

list=`pip3 list --format columns --outdate`  # execute pip, and get the output
echo "$list"

last_module=""
mode=u

echo "Input q to quit, input h for help."

while true
do
    read -p "please input module name: " module
    if [ $module == m -o -z $module ]
        # upgrade the last module again
    then
        if [ -n $last_module ]
        then
            echo "Upgrade $last_module again."
            pip3 install --upgrade $last_module -i https://pypi.tuna.tsinghua.edu.cn/simple
        else
            echo "Please input a valid string."
        fi
    elif [ $module == q ]   # to quit
    then
        echo "Finished. Have a good time ^_^"
        break
    elif [ $module == s ]
    then
        echo "sudo mode on"
        mode=s
    elif [ $module == u ]
    then
        echo "sudo mode off"
        mode=u
    elif [ $module == l ]
    then
        echo "$list"
    elif [ $module == h ]
    then
        echo "s: sudo, u: not sudo -- udo, l: show list again, m: install last module, h: help"
    else
        last_module=$module
        echo "Start to upgrade $last_module."
        if [ $mode == u ]
        then
            pip3 install --upgrade $module -i https://pypi.tuna.tsinghua.edu.cn/simple
        else
            sudo pip3 install --upgrade $module -i https://pypi.tuna.tsinghua.edu.cn/simple
        fi
    fi
done