rename.py

Go to the documentation of this file.

#!/usr/bin/env python3
from __future__ import print_function
from builtins import range
import sys,os,re
from CommonMethods import *
def main():
#    sourcePath = "/castor/cern.ch/cms/store/caf/user/uplegger/Workflows/Repro_2011Feb24/"
    sourcePath = "LatestRuns/"
#    sourcePath = "Files/"
    sourceDirList = [sourcePath+"2010A",sourcePath+"2010B",sourcePath+"Commissioning10"]
    destDirList   = ["2010A","2010B","Commissioning10"]
    path = "LatestRuns/"
    finalDir = path + "Results/"

    if not os.path.isdir(path):
        error = "WARNING: path directory doesn't exist! Creating it..."
        print(error)
        os.mkdir(path)

    if not os.path.isdir(finalDir):
        error = "WARNING: final dir directory doesn't exist! Creating it..."
        print(error)
        os.mkdir(finalDir)

    #for n in range(0,3):
    for n in range(0,len(sourceDirList)):
        sourceDir = sourceDirList[n] + '/'
        destDir   = path + destDirList[n] + '/'
        if not dirExists(sourceDir):
            print(sourceDir + " doesn't exist!")
            continue
        fileList = ls(sourceDir)
        if not os.path.isdir(destDir):
            error = "WARNING: destination directory doesn't exist! Creating it..."
            print(error)
            os.mkdir(destDir)
        copiedFiles = cp(sourceDir,destDir,fileList,False,False)

        #if len(copiedFiles) != len(fileList):
        #    error = "ERROR: I couldn't copy all files"
        #    exit(error)
#        exit("ok")
        for fileName in copiedFiles:
            #regExp = re.search('(\D+)(\d+)_(\d+)_.txt',fileName)
            regExp = re.search('(\D+)(\d+)_(\d+)_[a-zA-Z0-9]+.txt',fileName)
            #if regExp:
                #print regExp.group(1) + regExp.group(2) + "_" + str(1) + "_" + regExp.group(3) + ".txt" 

            fullFileName = destDir + fileName
            print(fullFileName)
            runNumber = -1
            with open(fullFileName,'r') as file:
                allTxt = '' 
                for line in file:
                    if line.find("Runnumber") != -1:
                        tmpRun = int(line.split(' ')[1])
                        if runNumber != -1 and tmpRun != runNumber:
                            error = "This file (" + fileName + ") contains more than 1 run number!"
                            if regExp:
                                newFileName = regExp.group(1) + regExp.group(2) + "_" + str(runNumber) + "_1.txt" 
                                with open(finalDir+newFileName,'a') as outFile:
                                    outFile.writelines(allTxt)
                                outFile.close()
                                allTxt = ''
                                #aCmd = "cp " + destDir + fileName + " " + finalDir + newFileName
                                #print aCmd
                                #output =  subprocess.getstatusoutput(aCmd)
                                #if output[0] != 0:
                                    #print output[1]
                            else:
                                print("WARNING: I can't match the regular espression for file: " + fileName)
                        runNumber = int(line.split(' ')[1])
                    allTxt += line
            file.close()
            if regExp:
                newFileName = regExp.group(1) + regExp.group(2) + "_" + str(runNumber) + "_" + regExp.group(3) + ".txt" 
                with open(finalDir+newFileName,'a') as outFile:
                    outFile.writelines(allTxt)
                outFile.close()





if __name__ == "__main__":
    main()