软硬件环境

  • ubuntu 18.04 64bit
  • anaconda3 with python 3.7.1
  • sh 1.12.14
  • fire

sh

sh是一个完美替代subprocess的方案,它能帮助你方便的启动其它程序。

安装sh

pip install sh

sh中封装了很多的shell命令,但是官方宣称sh is not a collection of system commands implemented in Python.

In[2]: import sh
In[3]: sh.ifconfig('enp0s25')
Out[3]: 
enp0s25: flags=4163<UP,BROADCAST,RUNNING,MULTICAST>  mtu 1500
        inet 192.168.1.182  netmask 255.255.255.0  broadcast 192.168.1.255
        inet6 fe80::9f51:6b1e:2a44:524c  prefixlen 64  scopeid 0x20<link>
        ether 50:7b:9d:d5:13:e6  txqueuelen 1000  (Ethernet)
        RX packets 2107249  bytes 2099954336 (2.0 GB)
        RX errors 0  dropped 0  overruns 0  frame 0
        TX packets 1343257  bytes 1630798979 (1.6 GB)
        TX errors 0  dropped 0 overruns 0  carrier 0  collisions 0
        device interrupt 20  memory 0xe1300000-e1320000

再看看grep的一个示例,比如我的系统中跑了多个ffmpeg的进程,然后需要找到其中的某一个,获取到进程的pid

xugaoxiang@ubuntu:~$ ps ax | grep ffmpeg
15133 pts/1    S+     1:54 ffmpeg -i rtsp://admin:rm123456@192.168.2.118:554 -c:a copy -c:v copy -y -f flv rtmp://192.168.1.97:2935/live/192.168.2.118:554/index
19252 pts/1    S+     0:00 ffmpeg -i rtsp://admin:rm123456@192.168.2.119:554 -c:a copy -c:v copy -y -f flv rtmp://192.168.1.97:2935/live/192.168.2.119:554/index
19491 pts/6    S+     0:00 grep --color=auto ffmpeg
import sh

def get_pid(expression):
    """

    :param expression: 字符串辨识符,如这里的ip地址 192.168.2.118:554
    :return: 进程的pid
    """

    p = sh.grep(sh.ps("ax"), expression)

    return p.stdout.decode('utf8').lstrip().split(' ')[0]

fire

firepython实用模块推荐(1)_click_pathlib中介绍过click模块功能类似,但是fire使用起来更加的方便简单(基本上不用写代码及文档注释),功能更加强大。

安装fire

pip install fire
import fire

def hello(name):
  return 'Hello {name}!'.format(name=name)

if __name__ == '__main__':
  fire.Fire()

如上面的示例所示,只需要fire.Fire()就完成了命令行参数的构建,运行方法是

python example.py hello world

其中example.py是文件名,hello是方法名,world是传递给hello方法的参数;另外你可以明确的给方法构建,如下调用fire.Fire(hello),那么运行的时候就可以省略掉hello方法了,python example.py world

fire.Fire()的参数可以是方法,对象,类,字典,来看个字典的例子

import fire

def add(x, y):
  return x + y

def multiply(x, y):
  return x * y

if __name__ == '__main__':
  fire.Fire({
      'add': add,
      'multiply': multiply,
  })

运行

$ python example.py add 1 2
3
$ python example.py multiply 3 4
12

如果参数是类的话,跟字典传递的功能类似,即类的每个方法都可以接受命令行参数。

参考资料


Comments

comments powered by Disqus