def submitnew(gangOrFac):
arEdge = displayVar.get()
arRec = displayVar2.get()
ccEdge = displayVar3.get()
ccRec = displayVar4.get()
thePath = displayVar6.get()
initFolder = displayVar7.get()
delim = "\t"
if os.path.isdir(thePath+"\\"+initFolder):
tkMessageBox.showerror("Folder already exists", "You have given a project name that \
already exists as a file directory. Choose a new project name or new project directory")
cmw.mainW(thePath, initFolder, arRec,arEdge,ccRec,ccEdge,gangOrFac,delim)
displayVar5.set(thePath+"\\"+ initFolder)
submitold()
def submitnew(gangOrFac):
recordDirectory = displayVar.get()
for root, dirs, files in os.walk(recordDirectory):
if os.path.basename(root) == 'recordsFolder':
for i in dirs:
if i == 'arRelationships':
arEdge = os.path.join(recordDirectory, i)
elif i == 'arRecords':
arRec = os.path.join(recordDirectory, i)
elif i == 'ccRelationships':
ccEdge =os.path.join(recordDirectory, i)
elif i == 'ccRecords':
ccRec =os.path.join(recordDirectory, i)
thePath = displayVar6.get()
initFolder = displayVar7.get()
delim = "\t"
if os.path.isdir(thePath+"\\"+initFolder):
tkMessageBox.showerror("Folder already exists", "You have given a project name that \
already exists as a file directory. Choose a new project name or new project directory")
else:
cmw.mainW(thePath, initFolder, arRec,arEdge,ccRec,ccEdge,gangOrFac,delim)
displayVar5.set(thePath+"\\"+ initFolder)
submitold()
def submitnew(gangOrFac,ar,nonAr,removal,day,month,year, forcastFlag):
monthToIntDict = {"January":1,"February":2,"March":3,"April":4,"May":5,"June":6,"July":7,"August":8,"September":9,"October":10,"November":11,"December":12}
yearOne = datetime.strptime('110001', "%d%m%Y").date()
if day == "Day" or month == "Month" or year == "Year":
dateLimit = yearOne
else:
dateLimit = datetime.strptime(str(day)+str(monthToIntDict[month])+str(year), "%d%m%Y").date()
recordDirectory = displayVar
if os.path.basename(recordDirectory) == 'recordsFolder':
ensure_dir(recordDirectory+'\\arRelationships')
ensure_dir(recordDirectory+'\\arRecords')
ensure_dir(recordDirectory+'\\nonArRelationships')
ensure_dir(recordDirectory+'\\nonArRecords')
ensure_dir(recordDirectory+'\\RemovedNodes')
ensure_dir(recordDirectory+'\\Blank')
blankFiles = glob.glob(recordDirectory+'/Blank/*')
for f in blankFiles:
os.remove(f)
for root, dirs, files in os.walk(recordDirectory):
if os.path.basename(root) == 'records':
for i in dirs:
if i == 'arRelationships':
arEdge = os.path.join(recordDirectory, i)
elif i == 'arRecords':
arRec = os.path.join(recordDirectory, i)
elif i == 'nonArRelationships':
nonArEdge =os.path.join(recordDirectory, i)
elif i == 'nonArRecords':
nonArRec =os.path.join(recordDirectory, i)
elif i == 'RemovedNodes':
remNodes = os.path.join(recordDirectory, i)
if ar == 0:
arEdge = os.path.join(recordDirectory, 'Blank')
arRec = os.path.join(recordDirectory, 'Blank')
if nonAr == 0:
nonArEdge = os.path.join(recordDirectory, 'Blank')
nonArRec = os.path.join(recordDirectory, 'Blank')
if removal == 0:
remNodes = os.path.join(recordDirectory, 'Blank')
thePath = displayVar6
initFolder = displayVar7.get()
delim = "\t"
if initFolder == "":
tkMessageBox.showerror("Blank Entry","Enter a name for the altered project.")
return
elif os.path.isdir(thePath+"\\"+initFolder) and not forcastFlag:
if not overwriteDialog():
return
ensure_dir(thePath+'\\'+initFolder)
if day == "Day" or month == "Month" or year == "Year":
dateString = '110001'
else:
dateStringList = [str(day),str(monthToInt(month)),str(year)]
if len(dateStringList[0]) == 1:
dateStringList[0] = '0'+dateStringList[0]
if len(dateStringList[1]) == 1:
dateStringList[1] = '0'+dateStringList[1]
if len(dateStringList[2]) == 2:
dateStringList[2] = '20'+dateStringList[2]
dateString = str(dateStringList[0])+str(dateStringList[1])+str(dateStringList[2])
prefs = open(displayVar6+'\\'+displayVar7.get()+'\\DataPreferences','w')
prefs.write(str(arRec)+','+str(arEdge)+','+str(nonArRec)+','+str(nonArEdge)+','+str(remNodes)+'\n')
prefs.write(dateString+','+day+','+month+','+year+'\n')
prefs.write(str(gangOrFac)+','+str(ar)+','+str(nonAr)+','+str(removal)+','+str(ar)+'\n')
prefs.close()
newRemNodes = thePath+'\\'+initFolder+'\\projectRemoved'
if forcastFlag:
remNodes = prevDir+'\\projectRemoved'
ensure_dir(newRemNodes)
distutils.dir_util.copy_tree(remNodes, newRemNodes)
if forcastFlag:
shutil.copy(displayVar5.get()+'\\removeMore.dsv',newRemNodes)
remNodes = newRemNodes
cmw.mainW(thePath, initFolder, arRec,arEdge,nonArRec,nonArEdge,remNodes,dateLimit,gangOrFac,delim)
displayVar5.set(thePath+"\\"+ initFolder)
submitold()
thePath = r"c:\batch\experVersionCpd" #Init files (assumed to be in the above path) #Init arrest files arRecLoc = r"C:\batch\experVersionCpd\initFiles\arRec" arEdgeLoc = r"C:\batch\experVersionCpd\initFiles\arEdge" ccRecLoc = r"C:\batch\experVersionCpd\initFiles\ccRec" ccEdgeLoc = r"C:\batch\experVersionCpd\initFiles\ccEdge" #0 for gang, 1 for fac gangOrFac = 1 #Project name #initFolder="allFac4" initFolder="allGang3" #Delimiter used in input files delim = "\t" #Main wrapper that gets run cmw.mainW(thePath,initFolder,arRecLoc,arEdgeLoc,ccRecLoc,ccEdgeLoc,gangOrFac,delim) #What-if package takes a list of nodes to kill and re-does the analysis #whatIf.whatIf(thePath,initFolder,initFolder+"-copy",['2015918'],delim) print "done with all."
