Windows下的bat, linux 下的shell 用来做批处理都很好用,可惜不通用
用 Python 来做就简单多了,不过一条条写代码来调用系统命令也够烦的了
程序员都很懒, 不愿做机械无谓的重复性工作, 干脆自己实现一个.
用法超级简单, 默认会执行一个自定义的 batch.json, 按顺序一条条执行其中的步骤
1
2
3
4
5
6
7
|
{ "steps" : [ { "step" : "df -h" , "desc" : "display disk space usage" }, { "step" : "date" , "desc" : "display the current dater" }, { "step" : "time" , "desc" : "display the current time" } ] } |
用法:
1
|
python batch.py |
当然也可以指定不同的步骤文件 , 例如
1
|
python batch.py xxx.json |
运行结果以markdown形式输出, 例如
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
|
$ python batch.py Usage: python batch.py <batch_json_file> note: execute the batch.json by default # Execute batch.json begin - - - - - - - - - - - - - - - - - - - - - - - - - - - ## Will execute 3 steps ~~~~~~~~~~~~~~~~~~~~~~~~~~~ 0. [df - h]: display disk space usage 1. [date]: display the current dater 2. [time]: display the current time * 0. [df - h]: display disk space usage Filesystem Size Used Avail Capacity iused ifree % iused Mounted on / dev / disk1 233Gi 208Gi 24Gi 90 % 54622825 6364694 90 % / devfs 329Ki 329Ki 0Bi 100 % * 1. [date]: display the current dater Thu Mar 3 22 : 50 : 21 CST 2016 * 2. [time]: display the current time real 0m0 . 001s user 0m0 . 000s sys 0m0 . 000s ## Done the following steps ~~~~~~~~~~~~~~~~~~~~~~~~~~~ 0. [df - h]: display disk space usage 1. [date]: display the current dater # Execute batch.json end. |
Python源代码如下, 希望有人能用得上
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
|
''' like bat file, execute the steps in batch.json ''' import os,sys,subprocess import time,thread import codecs import json from datetime import datetime from subprocess import call from pprint import pprint def execute_json(json_file): print "# Execute {0} begin\n---------------------------" . format (json_file) json_data = open (json_file) data = json.load(json_data) cnt = len (data[ 'steps' ]) i = 0 print "\n## Will execute {0} steps \n~~~~~~~~~~~~~~~~~~~~~~~~~~~" . format (cnt) for i in range ( 0 , cnt): print "{0}. [{1}]: {2}" . format (i, data[ 'steps' ][i][ 'step' ], data[ 'steps' ][i][ 'desc' ]) #pprint(data) #print("cnt=", cnt) for i in range ( 0 , cnt): cmd = data[ 'steps' ][i][ 'step' ] desc = data[ 'steps' ][i][ 'desc' ] print "\n* {0}. [{1}]: {2} " . format (i, cmd, desc) if (cmd.startswith( 'cd' )): cmd = cmd.replace( "cd " , "") os.chdir(cmd) else : ret = os.system(cmd) if (ret ! = 0 ): print "Encounter error of step {0}. {1}, error code={2}" . format (i, cmd, ret) break print "\n## Done the following steps\n~~~~~~~~~~~~~~~~~~~~~~~~~~~" for j in range ( 0 , i): print "{0}. [{1}]: {2}" . format (j, data[ 'steps' ][j][ 'step' ], data[ 'steps' ][j][ 'desc' ]) json_data.close() print "# Execute {0} end." . format (json_file) if __name__ = = "__main__" : argc = len (sys.argv) step_file = 'batch.json' if ( argc > 1 ): idx = 1 while (idx < argc): step_file = sys.argv[idx] execute_json(step_file) idx = idx + 1 else : print "Usage: python {0} <batch_json_file>" . format (sys.argv[ 0 ]) print "note: execute the batch.json by default" execute_json(step_file) |
以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持服务器之家。
原文链接:https://blog.csdn.net/fanyamin/article/details/50792426