[2019.04.16] 由Python写成的自动解压脚本

很久很久以前（二十七天吧……大概）被要求写一个脚本来检索并解压磁盘上所有的以特定格式命名的tar文件，于是乎学习和摸鱼就一起开始了。

这次要写的脚本，针对的是这样的文件结构：

文件结构如上图所示

可以看到一个tar包里面套着两个tgz压缩包，我需要完成如下图所示的工作：

PressOn是个好东西

在Python中，有两个可以调用系统指令的包，一个叫做sys，一个叫做os。比较棒的是它们在高级语言层面上是不区分操作系统的，同样一个变量 curdir 在 linux 下和在 win下都能拿到针对此操作系统可用的路径。而os.system() 或者 os.popen() 都可以完成将括号内带的参数变成系统使用的指令的工作，只要给出的指令是正确的。

在此次脚本编写中我学到一些新鲜玩意，在这里记录下来便于自己查阅。

 #! usr/bin/python
 import sys
 import re
 import os

 errorMsg = "Usage: python unzipAutoTriage.py [File location]. If File location was ommitted, script will work at current location."

 if len(sys.argv) == 1:
     msg = raw_input("No arguments, script will work at current path(y/n):")
     if msg != "y":
         sys.exit(errorMsg)
     path = "."
 elif len(sys.argv) >= 3:
     if re.match(r'~?[a-zA-Z/\_.]*',sys.argv[1]) != None:
         print "Too many arguments, do you want to execute in", sys.argv[1], "(y/n):"
         msg = raw_input()
         if msg != "y":
             sys.exit(errorMsg)
 else:
     path = sys.argv[1]
     print path

 command = "find " + path + " -mount -name 'auto_triage*.tar'"
 filelist = os.popen(command).readlines()
 status = dict.fromkeys(filelist,"Not complete") # 创建一个字典用于保存每个文件的处理状态
 print "Found" , len(filelist) , "File"
 if len(filelist) <= 0:
     sys.exit("No such file, exiting....")

 rootPath = os.path.abspath(os.curdir)           # 保存一下运行起始目录，之后需要回来才能继续进行操作

 for path in filelist:
     command = re.sub(r'auto.*.tar$','',path).replace("\n",'')
     if os.path.abspath(os.curdir).split("/")[-1] != command.split("/")[-1]:
         os.chdir(command)                       # 更换目录不能用os.system("cd xxxx")这种方法，过不去的……
     print "unzipping", path
     command = "tar -xvf " + re.sub(r'^\..*/auto','auto',path)
     res = os.system(command)
     if res == 0:
         flag = 2
         for item in ["spa","spb"]:
             print "extracting ",item
             command = "tar -xzvf " + path + item + ".service_dc.tgz -C " + path
             command = re.sub(r'.tar\n','/',command)
             res = os.system(command)
             if res == 0:
                 flag -= flag                    # Python逻辑里面可没有什么自增自减!
             else:
                 status[path] =  str(flag) + " unzip subprogress failed"
         if flag == 0:
             status[path] = "completed"
     else:
         status[path] = "Failed"
     os.chdir(rootPath)
 print "****************\nStatus report\n****************"
 for key,value in status.items():
     print '{key}Status: {value}'.format(key = key, value = value)

具体踩到的一些坑，就记载到 https://www.cnblogs.com/jackablack/p/10614686.html 采坑合集里面吧！