def test_Procheck(self):
cingDirTmpTest = os.path.join(cingDirTmp, getCallerName())
mkdirs(cingDirTmpTest)
self.failIf(os.chdir(cingDirTmpTest),
msg="Failed to change to test directory for files: " +
cingDirTmpTest)
runAqua = True
showProcheckResults = False
#entryId = "1ai0" # Most complex molecular system in any PDB NMR entry
entryId = "1bus"
# entryId = "1brv_1model" # Small much studied PDB NMR entry
ranges = None
if entryId.startswith("1brv"):
ranges = "173-186"
if entryId.startswith("1ai0"):
ranges = "2-20"
if entryId == "2hgh":
# in the case of 2hgh this is not a problem because the residue numbering doesn't
# overlap between the chain A protein and chain B RNA.
ranges = "2-11,13-33,35-54"
# 1 and 55 are 5' and 3' terminii which are a little looser.
# 12, and 34 are bases that are not basepaired.
ranges += ",104-105,115-136,145-190"
# 106-114 is a loop
# 137-144 is a loop
# 191-193 are 3 Zn ions.
#This leads to a procheck ranges file like:
# RESIDUES 2 B 11 B
# RESIDUES 13 B 33 B
# RESIDUES 35 B 54 B
# RESIDUES 104 A 105 A
# RESIDUES 115 A 136 A
# RESIDUES 145 A 190 A
cingDirTmpTest = os.path.join(cingDirTmp, getCallerName())
mkdirs(cingDirTmpTest)
self.failIf(os.chdir(cingDirTmpTest),
msg="Failed to change to test directory for files: " +
cingDirTmpTest)
project = Project(entryId)
self.failIf(project.removeFromDisk())
project = Project.open(entryId, status='new')
cyanaFile = os.path.join(cingDirTestsData, "cyana",
entryId + ".cyana.tgz")
self.assertTrue(project.initCyana(cyanaFolder=cyanaFile))
project.molecule.setRanges(ranges)
project.save()
self.failIf(
project.runProcheck(createPlots=True, runAqua=runAqua) is None)
if showProcheckResults:
for res in project.molecule.allResidues():
nTdebug(repr(res) + " " + repr(res.procheck.secStruct))
def testMolgrapRunFromPdbFile(self):
# SETUP FIRST
# entryId = "1ai0" # Most complex molecular system in any PDB NMR entry
# entryId = "1a4d" # Small much studied PDB NMR entry
# entryId = "1zwj" # X-ray entry of CESG interest.
entryId = "1brv" # Small much studied PDB NMR entry
# entryId = "2hgh_1model"
# does it matter to import it just now?
project = Project(entryId)
self.failIf(project.removeFromDisk())
project = Project.open(entryId, status='new')
cyanaFile = os.path.join(cingDirTestsData, "cyana",
entryId + ".cyana.tgz")
self.assertTrue(project.initCyana(cyanaFolder=cyanaFile))
project.save()
gifFileName = entryId + ".gif"
pathGif = os.path.join(self.cingDirTmpTest, gifFileName)
self.assertFalse(project.molecule.export2gif(pathGif, project=project))
self.assertTrue(os.path.exists(pathGif))
pathMolGifPinup = pathGif[:-4] + '_pin.gif'
self.assertTrue(os.path.exists(pathMolGifPinup))
pathGifDefault = os.path.join(cingPythonCingDir, 'PluginCode',
DATA_STR, 'UnknownImage.gif')
self.assertFalse(
os.path.getsize(pathGif) == os.path.getsize(pathGifDefault))
nTmessage("Created new molecular imagery at: %s" % self.cingDirTmpTest)
def test_Procheck(self):
cingDirTmpTest = os.path.join( cingDirTmp, getCallerName() )
mkdirs( cingDirTmpTest )
self.failIf(os.chdir(cingDirTmpTest), msg =
"Failed to change to test directory for files: " + cingDirTmpTest)
runAqua = True
showProcheckResults = False
#entryId = "1ai0" # Most complex molecular system in any PDB NMR entry
entryId = "1bus"
# entryId = "1brv_1model" # Small much studied PDB NMR entry
ranges = None
if entryId.startswith("1brv"):
ranges = "173-186"
if entryId.startswith("1ai0"):
ranges = "2-20"
if entryId == "2hgh":
# in the case of 2hgh this is not a problem because the residue numbering doesn't
# overlap between the chain A protein and chain B RNA.
ranges = "2-11,13-33,35-54"
# 1 and 55 are 5' and 3' terminii which are a little looser.
# 12, and 34 are bases that are not basepaired.
ranges += ",104-105,115-136,145-190"
# 106-114 is a loop
# 137-144 is a loop
# 191-193 are 3 Zn ions.
#This leads to a procheck ranges file like:
# RESIDUES 2 B 11 B
# RESIDUES 13 B 33 B
# RESIDUES 35 B 54 B
# RESIDUES 104 A 105 A
# RESIDUES 115 A 136 A
# RESIDUES 145 A 190 A
cingDirTmpTest = os.path.join( cingDirTmp, getCallerName() )
mkdirs( cingDirTmpTest )
self.failIf(os.chdir(cingDirTmpTest), msg =
"Failed to change to test directory for files: " + cingDirTmpTest)
project = Project( entryId )
self.failIf( project.removeFromDisk() )
project = Project.open( entryId, status='new' )
cyanaFile = os.path.join(cingDirTestsData, "cyana", entryId + ".cyana.tgz")
self.assertTrue(project.initCyana(cyanaFolder = cyanaFile))
project.molecule.setRanges(ranges)
project.save()
self.failIf(project.runProcheck(createPlots=True, runAqua=runAqua) is None)
if showProcheckResults:
for res in project.molecule.allResidues():
nTdebug(repr(res) +" "+ repr(res.procheck.secStruct))
def test_Whatif(self):
#entryId = "1ai0" # Most complex molecular system in any PDB NMR entry
# entryId = "2hgh" # Small much studied PDB NMR entry; 48 models
# entryId = "1bus" # Small much studied PDB NMR entry: 5 models of 57 AA.: 285 residues.
entryId = "1brv" # DEFAULT is not to do more than 2 models because it takes quite a while.
# entryId = "1brv_cs_pk_2mdl"
# entryId = "1tgq_1model"
# pdbConvention = IUPAC
parseOnly = False # normal is False
showValues = True
ranges='cv'
# ranges='172-177'
# ranges='6-13,29-45'
cingDirTmpTest = os.path.join( cingDirTmp, getCallerName() )
mkdirs( cingDirTmpTest )
self.failIf(os.chdir(cingDirTmpTest), msg =
"Failed to change to test directory for files: " + cingDirTmpTest)
project = Project( entryId )
if not parseOnly:
project.removeFromDisk()
project = Project.open( entryId, status='new' )
inputArchiveDir = os.path.join(cingDirTestsData, "ccpn")
ccpnFile = os.path.join(inputArchiveDir, entryId + ".tgz")
if not os.path.exists(ccpnFile):
ccpnFile = os.path.join(inputArchiveDir, entryId + ".tar.gz")
if not os.path.exists(ccpnFile):
self.fail("Neither %s or the .tgz exist" % ccpnFile)
self.assertTrue(project.initCcpn(ccpnFolder = ccpnFile, modelCount=2))
self.assertFalse(runWhatif(project, ranges=ranges, parseOnly=False))
else:
project = Project.open( entryId, status='old' )
# print project.cingPaths.format()
project.save()
if showValues:
for res in project.molecule.allResidues():
nTdebug(repr(res))
whatifResDict = res.getDeepByKeys(WHATIF_STR)
if not whatifResDict:
continue
# checkIDList = whatifResDict.keys()
checkIDList = 'RAMCHK'.split()
for checkID in checkIDList:
valueList = whatifResDict.getDeepByKeys(checkID,VALUE_LIST_STR)
qualList = whatifResDict.getDeepByKeys(checkID,QUAL_LIST_STR)
nTdebug("%10s valueList: %-80s qualList: %-80s" % ( checkID, valueList, qualList))
# Do not leave the old CCPN directory laying around since it might get added to by another test.
if os.path.exists(entryId):
self.assertFalse(shutil.rmtree(entryId))
def testPdbFile(self):
nTwarning(
"This test case will take about 5 (+3 for 1v0e) minutes and is recommended to be done before major releases."
)
# entryId = "1ai0" # Most complex molecular system in any PDB NMR entry
# entryId = "1brv" # Small much studied PDB NMR entry
# entryId = "2hgh_1model"
# entryList = "1kr8".split()
# entryList = "1otz".split() # 61 chains of which one is ' '
# entryList = "1v0e".split()
# entryList = "1a4d 1a24 1afp 1ai0 1brv 1bus 1cjg 1hue 1ieh 1iv6 1kr8 1otz 2hgh 2k0e".split()
entryList = "1a4d 1ai0 1brv 1bus 1hue 1iv6 1kr8".split()
for entryId in entryList:
pdbDirectory = os.path.join(cingDirTestsData, "pdb", entryId)
pdbFileName = "pdb" + entryId + ".ent"
pdbFilePath = os.path.join(pdbDirectory, pdbFileName)
cingDirTmpTest = os.path.join(cingDirTmp, 'test2_pdb')
mkdirs(cingDirTmpTest)
os.chdir(cingDirTmpTest)
# does it matter to import it just now?
project = Project(entryId)
self.failIf(project.removeFromDisk())
project = Project.open(entryId, status='new')
self.assertTrue(
project.initPDB(pdbFile=pdbFilePath,
convention=IUPAC,
allowNonStandardResidue=True))
self.assertTrue(project.save())
def testPdbFile(self):
nTwarning("This test case will take about 5 (+3 for 1v0e) minutes and is recommended to be done before major releases.")
# entryId = "1ai0" # Most complex molecular system in any PDB NMR entry
# entryId = "1brv" # Small much studied PDB NMR entry
# entryId = "2hgh_1model"
# entryList = "1kr8".split()
# entryList = "1otz".split() # 61 chains of which one is ' '
# entryList = "1v0e".split()
# entryList = "1a4d 1a24 1afp 1ai0 1brv 1bus 1cjg 1hue 1ieh 1iv6 1kr8 1otz 2hgh 2k0e".split()
entryList = "1a4d 1ai0 1brv 1bus 1hue 1iv6 1kr8".split()
for entryId in entryList:
pdbDirectory = os.path.join(cingDirTestsData,"pdb", entryId)
pdbFileName = "pdb"+entryId+".ent"
pdbFilePath = os.path.join( pdbDirectory, pdbFileName)
cingDirTmpTest = os.path.join( cingDirTmp, 'test2_pdb' )
mkdirs( cingDirTmpTest )
os.chdir(cingDirTmpTest)
# does it matter to import it just now?
project = Project( entryId )
self.failIf( project.removeFromDisk())
project = Project.open( entryId, status='new' )
self.assertTrue( project.initPDB( pdbFile=pdbFilePath, convention=IUPAC, allowNonStandardResidue=True ))
self.assertTrue( project.save() )
def test_xplor_nih(self):
cingDirTmpTest = os.path.join( cingDirTmp, getCallerName() )
mkdirs( cingDirTmpTest )
self.failIf(os.chdir(cingDirTmpTest), msg =
"Failed to change to test directory for files: " + cingDirTmpTest)
entryId = "gb1"
pdbDirectory = os.path.join(cingDirTestsData,"xplor", entryId)
pdbFileName = entryId+".pdb"
pdbFilePath = os.path.join( pdbDirectory, pdbFileName)
project = Project( entryId )
self.failIf( project.removeFromDisk())
project = Project.open( entryId, status='new' )
project.initPDB( pdbFile=pdbFilePath, convention = XPLOR )
# project.validate(ranges, parseOnly, htmlOnly, doProcheck, doWhatif, doWattos, doTalos)
project.validate(htmlOnly=True, doProcheck=False, doWhatif=False, doWattos=False, doTalos=False)
project.save()
def _testCreateCcpn(self):
'Disabled test because...'
cingDirTmpTest = os.path.join( cingDirTmp, getCallerName() )
mkdirs( cingDirTmpTest )
self.failIf(os.chdir(cingDirTmpTest), msg =
"Failed to change to test directory for files: " + cingDirTmpTest)
for entryId in AllChecks.entryList:
# Allow pdb files to be of different naming systems for this test.
project = Project(entryId)
self.failIf(project.removeFromDisk())
project = Project.open(entryId, status = 'new')
cyanaFolder = os.path.join(cingDirTestsData,"cyana", entryId+".cyana.tgz")
nTdebug("Reading files from: " + cyanaFolder)
project.initCyana( cyanaFolder)
project.save()
nTmessage( "Project: %s" % project)
ccpnFolder = entryId + "New"
self.assertTrue(project.saveCcpn(ccpnFolder))
def _testCreateCcpn(self):
'Disabled test because...'
cingDirTmpTest = os.path.join(cingDirTmp, getCallerName())
mkdirs(cingDirTmpTest)
self.failIf(os.chdir(cingDirTmpTest),
msg="Failed to change to test directory for files: " +
cingDirTmpTest)
for entryId in AllChecks.entryList:
# Allow pdb files to be of different naming systems for this test.
project = Project(entryId)
self.failIf(project.removeFromDisk())
project = Project.open(entryId, status='new')
cyanaFolder = os.path.join(cingDirTestsData, "cyana",
entryId + ".cyana.tgz")
nTdebug("Reading files from: " + cyanaFolder)
project.initCyana(cyanaFolder)
project.save()
nTmessage("Project: %s" % project)
ccpnFolder = entryId + "New"
self.assertTrue(project.saveCcpn(ccpnFolder))
def test_xplor_nih(self):
cingDirTmpTest = os.path.join(cingDirTmp, getCallerName())
mkdirs(cingDirTmpTest)
self.failIf(os.chdir(cingDirTmpTest),
msg="Failed to change to test directory for files: " +
cingDirTmpTest)
entryId = "gb1"
pdbDirectory = os.path.join(cingDirTestsData, "xplor", entryId)
pdbFileName = entryId + ".pdb"
pdbFilePath = os.path.join(pdbDirectory, pdbFileName)
project = Project(entryId)
self.failIf(project.removeFromDisk())
project = Project.open(entryId, status='new')
project.initPDB(pdbFile=pdbFilePath, convention=XPLOR)
# project.validate(ranges, parseOnly, htmlOnly, doProcheck, doWhatif, doWattos, doTalos)
project.validate(htmlOnly=True,
doProcheck=False,
doWhatif=False,
doWattos=False,
doTalos=False)
project.save()
def testMolgrapRunFromPdbFile(self):
# SETUP FIRST
# entryId = "1ai0" # Most complex molecular system in any PDB NMR entry
# entryId = "1a4d" # Small much studied PDB NMR entry
# entryId = "1zwj" # X-ray entry of CESG interest.
entryId = "1brv" # Small much studied PDB NMR entry
# entryId = "2hgh_1model"
# does it matter to import it just now?
project = Project( entryId )
self.failIf( project.removeFromDisk())
project = Project.open( entryId, status='new' )
cyanaFile = os.path.join(cingDirTestsData, "cyana", entryId + ".cyana.tgz")
self.assertTrue(project.initCyana(cyanaFolder = cyanaFile))
project.save()
gifFileName = entryId+".gif"
pathGif = os.path.join( self.cingDirTmpTest, gifFileName)
self.assertFalse(project.molecule.export2gif(pathGif, project=project))
self.assertTrue(os.path.exists(pathGif))
pathMolGifPinup = pathGif[:-4] + '_pin.gif'
self.assertTrue(os.path.exists(pathMolGifPinup))
pathGifDefault = os.path.join( cingPythonCingDir, 'PluginCode', DATA_STR, 'UnknownImage.gif' )
self.assertFalse(os.path.getsize(pathGif) == os.path.getsize(pathGifDefault))
nTmessage("Created new molecular imagery at: %s" % self.cingDirTmpTest)
def main(entryId, *extraArgList):
"""inputDir may be a directory or a url. A url needs to start with http://.
"""
fastestTest = False # default: False
htmlOnly = False # default: False but enable it for faster runs without some actual data.
doWhatif = True # disables whatif actual run
doProcheck = True
doWattos = True
doQueeny = True
doTalos = True
tgzCing = True # default: True # Create a tgz for the cing project. In case of a CING project input it will be overwritten.
# NB leave this set to True or modify code below.
removeCcpnDirectory = 1 # perhaps not so in the future.
modelCount = None # default setting is None
# ranges = None
if fastestTest:
modelCount = 2 # if this is more and there is only one model present it leads to an error message.
htmlOnly = True
doWhatif = False
doProcheck = False
doWattos = False
doQueeny = False
doTalos = False
forceRedo = True
forceRetrieveInput = True
nTmessage(header)
nTmessage(getStartMessage())
# Sync below code with nrgCing#createToposTokens
expectedArgumentList = """
verbosity inputDir outputDir
pdbConvention restraintsConvention archiveType projectType
storeCING2db ranges filterTopViolations filterVasco
singleCoreOperation
""".split()
expectedNumberOfArguments = len(expectedArgumentList)
if len(extraArgList) != expectedNumberOfArguments:
nTmessage("consider updating code to include all sequential parameters: %s" % str(expectedArgumentList))
if len(extraArgList) > expectedNumberOfArguments:
nTerror("Got arguments: " + str(extraArgList))
nTerror("Failed to get expected number of arguments: %d got %d" % (
expectedNumberOfArguments, len(extraArgList)))
nTerror("Expected arguments: %s" % expectedArgumentList)
return True
# end if
# end if
entryCodeChar2and3 = entryId[1:3]
cing.verbosity = int( extraArgList[IDX_VERBOSITY] )
inputDir = extraArgList[IDX_INPUT]
outputDir = os.path.join(extraArgList[IDX_OUTPUT], DATA_STR, entryCodeChar2and3, entryId)
pdbConvention = extraArgList[IDX_PDB] #@UnusedVariable
restraintsConvention = extraArgList[IDX_RESTRAINTS]
archiveType = extraArgList[IDX_ARCHIVE] # Only used for deriving the input location not the output.
projectType = extraArgList[IDX_PROJECT_TYPE]
storeCING2db = stringMeansBooleanTrue( getDeepByKeysOrAttributes(extraArgList, IDX_STORE_DB))
ranges = getDeepByKeysOrAttributes(extraArgList, IDX_RANGES)
filterTopViolations = getDeepByKeysOrAttributes(extraArgList, IDX_FILTER_TOP)
if filterTopViolations:
filterTopViolations = int(filterTopViolations) # change '0' to 0
filterVasco = getDeepByKeysOrAttributes(extraArgList, IDX_FILTER_VASCO)
if filterVasco:
filterVasco = int(filterVasco)
else:
filterVasco = 1 # Default should be True
# end if
singleCoreOperation = getDeepByKeysOrAttributes(extraArgList, IDX_SINGLE_CORE_OPERATION )
if singleCoreOperation:
singleCoreOperation = int(singleCoreOperation)
else:
singleCoreOperation = 0 # Default should be True
# end if
if archiveType == ARCHIVE_TYPE_FLAT:
pass # default
elif archiveType == ARCHIVE_TYPE_BY_ENTRY:
inputDir = os.path.join(inputDir, entryId)
elif archiveType == ARCHIVE_TYPE_BY_CH23:
inputDir = os.path.join(inputDir, entryCodeChar2and3)
elif archiveType == ARCHIVE_TYPE_BY_CH23_BY_ENTRY:
inputDir = os.path.join(inputDir, entryCodeChar2and3, entryId)
isRemoteOutputDir = False
if '@' in outputDir:
isRemoteOutputDir = True
# end if
# vc = vCing('.') # argument is a fake master_ssh_url not needed here.
archive_id = getArchiveIdFromDirectoryName( outputDir )
nTdebug("Using program arguments:")
nTdebug("inputDir: %s" % inputDir)
nTdebug("outputDir: %s" % outputDir)
nTdebug("pdbConvention: %s" % pdbConvention)
nTdebug("restraintsConvention: %s" % restraintsConvention)
nTdebug("archiveType: %s" % archiveType)
nTdebug("projectType: %s" % projectType)
nTdebug("storeCING2db: %s" % storeCING2db)
nTdebug("ranges: %s" % ranges)
nTdebug("filterTopViolations: %s" % filterTopViolations)
nTdebug("filterVasco: %s" % filterVasco)
nTdebug("singleCoreOperation: %s" % singleCoreOperation)
nTdebug("")
nTdebug("Using derived settings:")
nTdebug("modelCount: %s" % modelCount)
nTdebug("isRemoteOutputDir: %s" % isRemoteOutputDir)
nTdebug("archive_id: %s" % archive_id)
# For NMR_REDO required as most efficient.
if singleCoreOperation:
setToSingleCoreOperation()
# presume the directory still needs to be created.
cingEntryDir = entryId + ".cing"
if os.path.isdir(cingEntryDir):
if forceRedo:
nTmessage("Enforcing a redo")
rmtree(cingEntryDir)
else:
mainIndexFile = os.path.join(cingEntryDir, "index.html")
isDone = os.path.isfile(mainIndexFile)
if isDone:
nTmessage("SKIPPING ENTRY ALREADY DONE")
return
nTmessage("REDOING BECAUSE VALIDATION CONSIDERED NOT DONE.")
rmtree(cingEntryDir)
# end if.
# end if.
if isRemoteOutputDir:
os.chdir(cingDirTmp)
else:
os.chdir(outputDir)
project = Project(entryId)
if project.removeFromDisk():
nTerror("Failed to remove existing project (if present)")
return True
# end if.
formatFileName = '%s.tgz'
if projectType == PROJECT_TYPE_CING:
formatFileName = '%s.cing.tgz'
elif projectType == PROJECT_TYPE_PDB:
formatFileName = 'pdb%s.ent.gz'
fileNameTgz = formatFileName % entryId
# nTdebug("fileNameTgz: %s" % fileNameTgz)
allowedInputProtocolList = 'http file ssh'.split()
inputProtocal = string.split( inputDir, ':' )[0]
if inputProtocal in allowedInputProtocolList:
stillToRetrieve = False
if os.path.exists(fileNameTgz):
if forceRetrieveInput:
os.unlink(fileNameTgz)
stillToRetrieve = True
# end if
else:
stillToRetrieve = True
# end if
if stillToRetrieve:
retrieveTgzFromUrl(entryId, inputDir, archiveType=archiveType, formatFileName=formatFileName)
# end if
if not os.path.exists(fileNameTgz):
nTerror("Tgz should already have been present skipping entry")
return
# end if
else:
nTdebug("Entry not retrieved which might be normal in some situations.")
# end if.
if projectType == PROJECT_TYPE_CING:
# Needs to be copied because the open method doesn't take a directory argument..
# fullFileNameTgz = os.path.join(inputDir, fileNameTgz)
# shutil.copy(fullFileNameTgz, '.')
project = Project.open(entryId, status='old')
if not project:
nTerror("Failed to init old project")
return True
elif projectType == PROJECT_TYPE_CCPN:
project = Project.open(entryId, status='new')
if not project.initCcpn(ccpnFolder=fileNameTgz, modelCount=modelCount):
nTerror("Failed to init project from ccpn")
return True
elif projectType == PROJECT_TYPE_PDB:
project = Project.open(entryId, status='new')
pdbFilePath = entryId + ".pdb"
gunzip(fileNameTgz, outputFileName=pdbFilePath, removeOriginal=True)
project.initPDB(pdbFile=pdbFilePath, convention=IUPAC, nmodels=modelCount)
# if tmpPdbFile:
if True:
nTdebug("Removing tmp: %s" % pdbFilePath)
os.unlink(pdbFilePath)
# end if
if ranges is not None:
project.molecule.setRanges(ranges)
# end if
if archive_id:
project.molecule.setArchiveId(archive_id)
# end if
project.molecule.superpose(ranges=ranges)
if filterTopViolations and not project.filterHighRestraintViol():
nTerror("Failed to filterHighRestraintViol")
####> MAIN UTILITY HERE
if 0: # DEFAULT 0
project.save()
if project.validate(htmlOnly=htmlOnly, ranges=ranges, doProcheck=doProcheck, doWhatif=doWhatif,
doWattos=doWattos, doQueeny = doQueeny, doTalos=doTalos, filterVasco = filterVasco ):
nTerror("Failed to validate project read")
return True
# end if filterVasco
# Write a single PDB file containing all models
# according to IUPAC conventions
project.export2PDB()
project.save()
if storeCING2db and archive_id:
# Does require:
#from cing.PluginCode.sqlAlchemy import csqlAlchemy
# and should never crash run.
# archive_id = ARCHIVE_DEV_NRG_ID
# if isProduction:
# archive_id = ARCHIVE_NRG_ID
try:
if doStoreCING2db( entryId, archive_id, project=project):
nTerror("Failed to store CING project's data to DB but continuing.")
except:
nTtracebackError()
nTerror("Failed to store CING project's data due to above traceback error.")
if projectType == PROJECT_TYPE_CCPN:
# fileNameTgz = entryId + '.tgz'
os.unlink(fileNameTgz) # temporary ccpn tgz
if removeCcpnDirectory:
rmdir(entryId) # ccpn dir may contain vasco info.
if tgzCing:
directoryNameCing = entryId + ".cing"
tgzFileNameCing = directoryNameCing + ".tgz"
if os.path.exists(tgzFileNameCing):
nTwarning("Overwriting: " + tgzFileNameCing)
cmd = "tar -czf %s %s" % (tgzFileNameCing, directoryNameCing)
nTdebug("cmd: %s" % cmd)
# do_cmd(cmd)
status, result = commands.getstatusoutput(cmd)
if status:
nTerror("Failed to tar status: %s with result %s" % (status, result))
return True
if isRemoteOutputDir:
if putFileBySsh(tgzFileNameCing, outputDir, ntriesMax = 2):
nTerror("Failed to send File By Scp status: %s with result %s" % (status, result))
nTerror("Maintaining results.")
return True
# end if
nTmessage("Removing tgz result: %s" % tgzFileNameCing)
os.remove(tgzFileNameCing)
nTmessage("Removing cing dir itself: %s" % directoryNameCing)
rmdir(directoryNameCing)
else: # do NOT remove local copy
pass
def main(entryId, *extraArgList):
"""inputDir may be a directory or a url. A url needs to start with http://.
"""
fastestTest = True # default: False
# ranges=AUTO_STR # default is None retrieved from DBMS csv files.
htmlOnly = False # default: False but enable it for faster runs without some actual data.
doWhatif = True # disables whatif actual run
doProcheck = True
doWattos = True
doTalos = True
tgzCing = True # default: True # Create a tgz for the cing project. In case of a CING project input it will be overwritten.
modelCount = None # default setting is None
if fastestTest:
modelCount = 3
htmlOnly = True
doWhatif = False
doProcheck = False
doWattos = False
doTalos = False
force_redo = True
force_retrieve_input = True
nTmessage(cing.cingDefinitions.getHeaderString())
nTmessage(cing.systemDefinitions.getStartMessage())
expectedArgumentList = [ 'inputDir', 'outputDir', 'pdbConvention', 'restraintsConvention', 'archiveType','projectType','storeCING2db']
expectedNumberOfArguments = len(expectedArgumentList)
if len(extraArgList) != expectedNumberOfArguments:
nTerror("Got arguments: " + repr(extraArgList))
nTerror("Failed to get expected number of arguments: %d got %d" % (
expectedNumberOfArguments, len(extraArgList)))
nTerror("Expected arguments: %s" % expectedArgumentList)
return True
entryCodeChar2and3 = entryId[1:3]
inputDir = extraArgList[0]
outputDir = os.path.join(extraArgList[1], DATA_STR, entryCodeChar2and3, entryId)
pdbConvention = extraArgList[2] #@UnusedVariable
restraintsConvention = extraArgList[3]
archiveType = extraArgList[4]
projectType = extraArgList[5]
storeCING2db = False
if len(extraArgList) >= expectedNumberOfArguments:
storeCING2db = extraArgList[6]
if archiveType == ARCHIVE_TYPE_FLAT:
pass
# default
elif archiveType == ARCHIVE_TYPE_BY_ENTRY:
inputDir = os.path.join(inputDir, entryId)
elif archiveType == ARCHIVE_TYPE_BY_CH23:
inputDir = os.path.join(inputDir, entryCodeChar2and3)
elif archiveType == ARCHIVE_TYPE_BY_CH23_BY_ENTRY:
inputDir = os.path.join(inputDir, entryCodeChar2and3, entryId)
ranges = None
# targetId = getTargetForFullEntryName(entryId)
# if not targetId:
# nTerror("Failed to getTargetForFullEntryName for entryId: %s" % entryId)
# return True
# ranges = getRangesForTarget(targetId)
# if ranges == None:
# nTerror("Failed to getRangesForTarget for targetId: %s" % targetId)
# return True
nTdebug("Using:")
nTdebug("inputDir: %s" % inputDir)
nTdebug("outputDir: %s" % outputDir)
nTdebug("pdbConvention: %s" % pdbConvention)
nTdebug("restraintsConvention: %s" % restraintsConvention)
nTdebug("archiveType: %s" % archiveType)
nTdebug("projectType: %s" % projectType)
nTdebug("modelCount: %s" % modelCount)
nTdebug("storeCING2db: %s" % storeCING2db)
nTdebug("ranges: %s" % ranges)
# presume the directory still needs to be created.
cingEntryDir = entryId + ".cing"
if os.path.isdir(cingEntryDir):
if force_redo:
nTmessage("Enforcing a redo")
rmtree(cingEntryDir)
else:
mainIndexFile = os.path.join(cingEntryDir, "index.html")
isDone = os.path.isfile(mainIndexFile)
if isDone:
nTmessage("SKIPPING ENTRY ALREADY DONE")
return
nTmessage("REDOING BECAUSE VALIDATION CONSIDERED NOT DONE.")
rmtree(cingEntryDir)
# end if.
# end if.
os.chdir(outputDir)
project = Project(entryId)
if project.removeFromDisk():
nTerror("Failed to remove existing project (if present)")
return True
# end if.
# extension = '.tgz'
formatFileName = '%s.tgz'
# fileNameTgz = entryId + '.tgz'
if projectType == PROJECT_TYPE_CING:
# fileNameTgz = entryId + '.cing.tgz'
formatFileName = '%s.cing.tgz'
elif projectType == PROJECT_TYPE_PDB:
formatFileName = 'pdb%s.ent.gz'
fileNameTgz = formatFileName % entryId
# nTdebug("fileNameTgz: %s" % fileNameTgz)
# if true will do retrieveTgzFromUrl.
if inputDir.startswith("http") or inputDir.startswith("file"):
stillToRetrieve = False
if os.path.exists(fileNameTgz):
if force_retrieve_input:
os.unlink(fileNameTgz)
stillToRetrieve = True
# end if
else:
stillToRetrieve = True
# end if
if stillToRetrieve:
retrieveTgzFromUrl(entryId, inputDir, archiveType=archiveType, formatFileName=formatFileName)
# end if
if not os.path.exists(fileNameTgz):
nTerror("Tgz should already have been present skipping entry")
return
# end if
# end if.
# retrieveTgzFromUrl(entryId, inputDir)
if projectType == PROJECT_TYPE_CING:
# Needs to be copied because the open method doesn't take a directory argument..
fullFileNameTgz = os.path.join(inputDir, fileNameTgz)
shutil.copy(fullFileNameTgz, '.')
project = Project.open(entryId, status='old')
if not project:
nTerror("Failed to init old project")
return True
elif projectType == PROJECT_TYPE_CCPN:
project = Project.open(entryId, status='new')
if not project.initCcpn(ccpnFolder=fileNameTgz, modelCount=modelCount):
nTerror("Failed to init project from ccpn")
return True
elif projectType == PROJECT_TYPE_PDB:
project = Project.open(entryId, status='new')
# pdbFileFormats = [ entryId + ".pdb", "pdb" + entryId + ".ent.gz" ]
# for pdbFileName in pdbFileFormats:
# pdbFileName = "pdb" + entryId + ".ent.gz"
# # pdbFilePath = os.path.join( inputDir, pdbFileName)
# pdbFilePath = os.path.join(inputDir, pdbFileName)
# if os.path.exists(pdbFilePath):
# break
# tmpPdbFile = None
# if pdbFileName.endswith('.gz'):
pdbFilePath = entryId + ".pdb"
# tmpPdbFile = pdbFilePath
# if os.path.exists(pdbFilePath):
# os.unlink(pdbFilePath)
gunzip(fileNameTgz, outputFileName=pdbFilePath, removeOriginal=True)
project.initPDB(pdbFile=pdbFilePath, convention=IUPAC, nmodels=modelCount)
# if tmpPdbFile:
if True:
nTdebug("Removing tmp: %s" % pdbFilePath)
os.unlink(pdbFilePath)
# if inputDirOrg == inputDirCASD_NMR:
# if True: # Default is False for this is specific to CASD-NMR
# nTmessage("Renaming molecule name to entry id: %s" % entryId)
# project.molecule.name = entryId # insufficient since all data is already initialized to disk.
# project.molecule.rename( entryId )
# project.save()
# project.molecule.ranges = ranges # JFD: this doesn't seem to be set there exactly.
project.molecule.superpose(ranges=ranges)
if True:
if project.validate(htmlOnly=htmlOnly, ranges=ranges, doProcheck=doProcheck, doWhatif=doWhatif,
doWattos=doWattos, doTalos=doTalos):
nTerror("Failed to validate project read")
return True
if storeCING2db:
# Does require:
#from cing.PluginCode.sqlAlchemy import csqlAlchemy
# and should never crash run.
try:
if doStoreCING2db( entryId, ARCHIVE_CASP_ID, project=project):
nTerror("Failed to store CING project's data to DB but continuing.")
except:
nTtracebackError()
nTerror("Failed to store CING project's data due to above traceback error.")
project.save()
if projectType == PROJECT_TYPE_CCPN:
# fileNameTgz = entryId + '.tgz'
# os.unlink(fileNameTgz) # temporary ccpn tgz
rmdir(entryId) # temporary ccpn dir
if tgzCing:
directoryNameCing = entryId + ".cing"
tgzFileNameCing = directoryNameCing + ".tgz"
if os.path.exists(tgzFileNameCing):
nTwarning("Overwriting: " + tgzFileNameCing)
cmd = "tar -czf %s %s" % (tgzFileNameCing, directoryNameCing)
do_cmd(cmd)
def main(entryId, *extraArgList):
"""inputDir may be a directory or a url. A url needs to start with http://.
"""
fastestTest = False # default: False
htmlOnly = False # default: False but enable it for faster runs without some actual data.
doWhatif = True # disables whatif actual run
doProcheck = True
doWattos = True
doQueeny = True
doTalos = True
tgzCing = True # default: True # Create a tgz for the cing project. In case of a CING project input it will be overwritten.
# NB leave this set to True or modify code below.
removeCcpnDirectory = 1 # perhaps not so in the future.
modelCount = None # default setting is None
# ranges = None
if fastestTest:
modelCount = 2 # if this is more and there is only one model present it leads to an error message.
htmlOnly = True
doWhatif = False
doProcheck = False
doWattos = False
doQueeny = False
doTalos = False
forceRedo = True
forceRetrieveInput = True
nTmessage(header)
nTmessage(getStartMessage())
# Sync below code with nrgCing#createToposTokens
expectedArgumentList = """
verbosity inputDir outputDir
pdbConvention restraintsConvention archiveType projectType
storeCING2db ranges filterTopViolations filterVasco
singleCoreOperation
""".split()
expectedNumberOfArguments = len(expectedArgumentList)
if len(extraArgList) != expectedNumberOfArguments:
nTmessage(
"consider updating code to include all sequential parameters: %s" %
str(expectedArgumentList))
if len(extraArgList) > expectedNumberOfArguments:
nTerror("Got arguments: " + str(extraArgList))
nTerror("Failed to get expected number of arguments: %d got %d" %
(expectedNumberOfArguments, len(extraArgList)))
nTerror("Expected arguments: %s" % expectedArgumentList)
return True
# end if
# end if
entryCodeChar2and3 = entryId[1:3]
cing.verbosity = int(extraArgList[IDX_VERBOSITY])
inputDir = extraArgList[IDX_INPUT]
outputDir = os.path.join(extraArgList[IDX_OUTPUT], DATA_STR,
entryCodeChar2and3, entryId)
pdbConvention = extraArgList[IDX_PDB] #@UnusedVariable
restraintsConvention = extraArgList[IDX_RESTRAINTS]
archiveType = extraArgList[
IDX_ARCHIVE] # Only used for deriving the input location not the output.
projectType = extraArgList[IDX_PROJECT_TYPE]
storeCING2db = stringMeansBooleanTrue(
getDeepByKeysOrAttributes(extraArgList, IDX_STORE_DB))
ranges = getDeepByKeysOrAttributes(extraArgList, IDX_RANGES)
filterTopViolations = getDeepByKeysOrAttributes(extraArgList,
IDX_FILTER_TOP)
if filterTopViolations:
filterTopViolations = int(filterTopViolations) # change '0' to 0
filterVasco = getDeepByKeysOrAttributes(extraArgList, IDX_FILTER_VASCO)
if filterVasco:
filterVasco = int(filterVasco)
else:
filterVasco = 1 # Default should be True
# end if
singleCoreOperation = getDeepByKeysOrAttributes(extraArgList,
IDX_SINGLE_CORE_OPERATION)
if singleCoreOperation:
singleCoreOperation = int(singleCoreOperation)
else:
singleCoreOperation = 0 # Default should be True
# end if
if archiveType == ARCHIVE_TYPE_FLAT:
pass # default
elif archiveType == ARCHIVE_TYPE_BY_ENTRY:
inputDir = os.path.join(inputDir, entryId)
elif archiveType == ARCHIVE_TYPE_BY_CH23:
inputDir = os.path.join(inputDir, entryCodeChar2and3)
elif archiveType == ARCHIVE_TYPE_BY_CH23_BY_ENTRY:
inputDir = os.path.join(inputDir, entryCodeChar2and3, entryId)
isRemoteOutputDir = False
if '@' in outputDir:
isRemoteOutputDir = True
# end if
# vc = vCing('.') # argument is a fake master_ssh_url not needed here.
archive_id = getArchiveIdFromDirectoryName(outputDir)
nTdebug("Using program arguments:")
nTdebug("inputDir: %s" % inputDir)
nTdebug("outputDir: %s" % outputDir)
nTdebug("pdbConvention: %s" % pdbConvention)
nTdebug("restraintsConvention: %s" % restraintsConvention)
nTdebug("archiveType: %s" % archiveType)
nTdebug("projectType: %s" % projectType)
nTdebug("storeCING2db: %s" % storeCING2db)
nTdebug("ranges: %s" % ranges)
nTdebug("filterTopViolations: %s" % filterTopViolations)
nTdebug("filterVasco: %s" % filterVasco)
nTdebug("singleCoreOperation: %s" % singleCoreOperation)
nTdebug("")
nTdebug("Using derived settings:")
nTdebug("modelCount: %s" % modelCount)
nTdebug("isRemoteOutputDir: %s" % isRemoteOutputDir)
nTdebug("archive_id: %s" % archive_id)
# For NMR_REDO required as most efficient.
if singleCoreOperation:
setToSingleCoreOperation()
# presume the directory still needs to be created.
cingEntryDir = entryId + ".cing"
if os.path.isdir(cingEntryDir):
if forceRedo:
nTmessage("Enforcing a redo")
rmtree(cingEntryDir)
else:
mainIndexFile = os.path.join(cingEntryDir, "index.html")
isDone = os.path.isfile(mainIndexFile)
if isDone:
nTmessage("SKIPPING ENTRY ALREADY DONE")
return
nTmessage("REDOING BECAUSE VALIDATION CONSIDERED NOT DONE.")
rmtree(cingEntryDir)
# end if.
# end if.
if isRemoteOutputDir:
os.chdir(cingDirTmp)
else:
os.chdir(outputDir)
project = Project(entryId)
if project.removeFromDisk():
nTerror("Failed to remove existing project (if present)")
return True
# end if.
formatFileName = '%s.tgz'
if projectType == PROJECT_TYPE_CING:
formatFileName = '%s.cing.tgz'
elif projectType == PROJECT_TYPE_PDB:
formatFileName = 'pdb%s.ent.gz'
fileNameTgz = formatFileName % entryId
# nTdebug("fileNameTgz: %s" % fileNameTgz)
allowedInputProtocolList = 'http file ssh'.split()
inputProtocal = string.split(inputDir, ':')[0]
if inputProtocal in allowedInputProtocolList:
stillToRetrieve = False
if os.path.exists(fileNameTgz):
if forceRetrieveInput:
os.unlink(fileNameTgz)
stillToRetrieve = True
# end if
else:
stillToRetrieve = True
# end if
if stillToRetrieve:
retrieveTgzFromUrl(entryId,
inputDir,
archiveType=archiveType,
formatFileName=formatFileName)
# end if
if not os.path.exists(fileNameTgz):
nTerror("Tgz should already have been present skipping entry")
return
# end if
else:
nTdebug(
"Entry not retrieved which might be normal in some situations.")
# end if.
if projectType == PROJECT_TYPE_CING:
# Needs to be copied because the open method doesn't take a directory argument..
# fullFileNameTgz = os.path.join(inputDir, fileNameTgz)
# shutil.copy(fullFileNameTgz, '.')
project = Project.open(entryId, status='old')
if not project:
nTerror("Failed to init old project")
return True
elif projectType == PROJECT_TYPE_CCPN:
project = Project.open(entryId, status='new')
if not project.initCcpn(ccpnFolder=fileNameTgz, modelCount=modelCount):
nTerror("Failed to init project from ccpn")
return True
elif projectType == PROJECT_TYPE_PDB:
project = Project.open(entryId, status='new')
pdbFilePath = entryId + ".pdb"
gunzip(fileNameTgz, outputFileName=pdbFilePath, removeOriginal=True)
project.initPDB(pdbFile=pdbFilePath,
convention=IUPAC,
nmodels=modelCount)
# if tmpPdbFile:
if True:
nTdebug("Removing tmp: %s" % pdbFilePath)
os.unlink(pdbFilePath)
# end if
if ranges is not None:
project.molecule.setRanges(ranges)
# end if
if archive_id:
project.molecule.setArchiveId(archive_id)
# end if
project.molecule.superpose(ranges=ranges)
if filterTopViolations and not project.filterHighRestraintViol():
nTerror("Failed to filterHighRestraintViol")
####> MAIN UTILITY HERE
if 0: # DEFAULT 0
project.save()
if project.validate(htmlOnly=htmlOnly,
ranges=ranges,
doProcheck=doProcheck,
doWhatif=doWhatif,
doWattos=doWattos,
doQueeny=doQueeny,
doTalos=doTalos,
filterVasco=filterVasco):
nTerror("Failed to validate project read")
return True
# end if filterVasco
# Write a single PDB file containing all models
# according to IUPAC conventions
project.export2PDB()
project.save()
if storeCING2db and archive_id:
# Does require:
#from cing.PluginCode.sqlAlchemy import csqlAlchemy
# and should never crash run.
# archive_id = ARCHIVE_DEV_NRG_ID
# if isProduction:
# archive_id = ARCHIVE_NRG_ID
try:
if doStoreCING2db(entryId, archive_id, project=project):
nTerror(
"Failed to store CING project's data to DB but continuing."
)
except:
nTtracebackError()
nTerror(
"Failed to store CING project's data due to above traceback error."
)
if projectType == PROJECT_TYPE_CCPN:
# fileNameTgz = entryId + '.tgz'
os.unlink(fileNameTgz) # temporary ccpn tgz
if removeCcpnDirectory:
rmdir(entryId) # ccpn dir may contain vasco info.
if tgzCing:
directoryNameCing = entryId + ".cing"
tgzFileNameCing = directoryNameCing + ".tgz"
if os.path.exists(tgzFileNameCing):
nTwarning("Overwriting: " + tgzFileNameCing)
cmd = "tar -czf %s %s" % (tgzFileNameCing, directoryNameCing)
nTdebug("cmd: %s" % cmd)
# do_cmd(cmd)
status, result = commands.getstatusoutput(cmd)
if status:
nTerror("Failed to tar status: %s with result %s" %
(status, result))
return True
if isRemoteOutputDir:
if putFileBySsh(tgzFileNameCing, outputDir, ntriesMax=2):
nTerror(
"Failed to send File By Scp status: %s with result %s" %
(status, result))
nTerror("Maintaining results.")
return True
# end if
nTmessage("Removing tgz result: %s" % tgzFileNameCing)
os.remove(tgzFileNameCing)
nTmessage("Removing cing dir itself: %s" % directoryNameCing)
rmdir(directoryNameCing)
else: # do NOT remove local copy
pass
def main(entryId, *extraArgList):
"""inputDir may be a directory or a url. A url needs to start with http://.
"""
fastestTest = True # default: False
# ranges=AUTO_STR # default is None retrieved from DBMS csv files.
htmlOnly = False # default: False but enable it for faster runs without some actual data.
doWhatif = True # disables whatif actual run
doProcheck = True
doWattos = True
doTalos = True
tgzCing = True # default: True # Create a tgz for the cing project. In case of a CING project input it will be overwritten.
modelCount = None # default setting is None
if fastestTest:
modelCount = 3
htmlOnly = True
doWhatif = False
doProcheck = False
doWattos = False
doTalos = False
force_redo = True
force_retrieve_input = True
nTmessage(header)
nTmessage(getStartMessage())
expectedArgumentList = [
'inputDir', 'outputDir', 'pdbConvention', 'restraintsConvention',
'archiveType', 'projectType', 'storeCING2db'
]
expectedNumberOfArguments = len(expectedArgumentList)
if len(extraArgList) != expectedNumberOfArguments:
nTerror("Got arguments: " + repr(extraArgList))
nTerror("Failed to get expected number of arguments: %d got %d" %
(expectedNumberOfArguments, len(extraArgList)))
nTerror("Expected arguments: %s" % expectedArgumentList)
return True
entryCodeChar2and3 = entryId[1:3]
inputDir = extraArgList[0]
outputDir = os.path.join(extraArgList[1], DATA_STR, entryCodeChar2and3,
entryId)
pdbConvention = extraArgList[2] #@UnusedVariable
restraintsConvention = extraArgList[3]
archiveType = extraArgList[4]
projectType = extraArgList[5]
storeCING2db = False
if len(extraArgList) >= expectedNumberOfArguments:
storeCING2db = extraArgList[6]
if archiveType == ARCHIVE_TYPE_FLAT:
pass
# default
elif archiveType == ARCHIVE_TYPE_BY_ENTRY:
inputDir = os.path.join(inputDir, entryId)
elif archiveType == ARCHIVE_TYPE_BY_CH23:
inputDir = os.path.join(inputDir, entryCodeChar2and3)
elif archiveType == ARCHIVE_TYPE_BY_CH23_BY_ENTRY:
inputDir = os.path.join(inputDir, entryCodeChar2and3, entryId)
ranges = None
# targetId = getTargetForFullEntryName(entryId)
# if not targetId:
# nTerror("Failed to getTargetForFullEntryName for entryId: %s" % entryId)
# return True
# ranges = getRangesForTarget(targetId)
# if ranges == None:
# nTerror("Failed to getRangesForTarget for targetId: %s" % targetId)
# return True
nTdebug("Using:")
nTdebug("inputDir: %s" % inputDir)
nTdebug("outputDir: %s" % outputDir)
nTdebug("pdbConvention: %s" % pdbConvention)
nTdebug("restraintsConvention: %s" % restraintsConvention)
nTdebug("archiveType: %s" % archiveType)
nTdebug("projectType: %s" % projectType)
nTdebug("modelCount: %s" % modelCount)
nTdebug("storeCING2db: %s" % storeCING2db)
nTdebug("ranges: %s" % ranges)
# presume the directory still needs to be created.
cingEntryDir = entryId + ".cing"
if os.path.isdir(cingEntryDir):
if force_redo:
nTmessage("Enforcing a redo")
rmtree(cingEntryDir)
else:
mainIndexFile = os.path.join(cingEntryDir, "index.html")
isDone = os.path.isfile(mainIndexFile)
if isDone:
nTmessage("SKIPPING ENTRY ALREADY DONE")
return
nTmessage("REDOING BECAUSE VALIDATION CONSIDERED NOT DONE.")
rmtree(cingEntryDir)
# end if.
# end if.
os.chdir(outputDir)
project = Project(entryId)
if project.removeFromDisk():
nTerror("Failed to remove existing project (if present)")
return True
# end if.
# extension = '.tgz'
formatFileName = '%s.tgz'
# fileNameTgz = entryId + '.tgz'
if projectType == PROJECT_TYPE_CING:
# fileNameTgz = entryId + '.cing.tgz'
formatFileName = '%s.cing.tgz'
elif projectType == PROJECT_TYPE_PDB:
formatFileName = 'pdb%s.ent.gz'
fileNameTgz = formatFileName % entryId
# nTdebug("fileNameTgz: %s" % fileNameTgz)
# if true will do retrieveTgzFromUrl.
if inputDir.startswith("http") or inputDir.startswith("file"):
stillToRetrieve = False
if os.path.exists(fileNameTgz):
if force_retrieve_input:
os.unlink(fileNameTgz)
stillToRetrieve = True
# end if
else:
stillToRetrieve = True
# end if
if stillToRetrieve:
retrieveTgzFromUrl(entryId,
inputDir,
archiveType=archiveType,
formatFileName=formatFileName)
# end if
if not os.path.exists(fileNameTgz):
nTerror("Tgz should already have been present skipping entry")
return
# end if
# end if.
# retrieveTgzFromUrl(entryId, inputDir)
if projectType == PROJECT_TYPE_CING:
# Needs to be copied because the open method doesn't take a directory argument..
fullFileNameTgz = os.path.join(inputDir, fileNameTgz)
shutil.copy(fullFileNameTgz, '.')
project = Project.open(entryId, status='old')
if not project:
nTerror("Failed to init old project")
return True
elif projectType == PROJECT_TYPE_CCPN:
project = Project.open(entryId, status='new')
if not project.initCcpn(ccpnFolder=fileNameTgz, modelCount=modelCount):
nTerror("Failed to init project from ccpn")
return True
elif projectType == PROJECT_TYPE_PDB:
project = Project.open(entryId, status='new')
# pdbFileFormats = [ entryId + ".pdb", "pdb" + entryId + ".ent.gz" ]
# for pdbFileName in pdbFileFormats:
# pdbFileName = "pdb" + entryId + ".ent.gz"
# # pdbFilePath = os.path.join( inputDir, pdbFileName)
# pdbFilePath = os.path.join(inputDir, pdbFileName)
# if os.path.exists(pdbFilePath):
# break
# tmpPdbFile = None
# if pdbFileName.endswith('.gz'):
pdbFilePath = entryId + ".pdb"
# tmpPdbFile = pdbFilePath
# if os.path.exists(pdbFilePath):
# os.unlink(pdbFilePath)
gunzip(fileNameTgz, outputFileName=pdbFilePath, removeOriginal=True)
project.initPDB(pdbFile=pdbFilePath,
convention=IUPAC,
nmodels=modelCount)
# if tmpPdbFile:
if True:
nTdebug("Removing tmp: %s" % pdbFilePath)
os.unlink(pdbFilePath)
# if inputDirOrg == inputDirCASD_NMR:
# if True: # Default is False for this is specific to CASD-NMR
# nTmessage("Renaming molecule name to entry id: %s" % entryId)
# project.molecule.name = entryId # insufficient since all data is already initialized to disk.
# project.updateProject()
# project.molecule.rename( entryId )
# project.save()
# project.molecule.ranges = ranges # JFD: this doesn't seem to be set there exactly.
project.molecule.superpose(ranges=ranges)
if True:
if project.validate(htmlOnly=htmlOnly,
ranges=ranges,
doProcheck=doProcheck,
doWhatif=doWhatif,
doWattos=doWattos,
doTalos=doTalos):
nTerror("Failed to validate project read")
return True
if storeCING2db:
# Does require:
#from cing.PluginCode.sqlAlchemy import csqlAlchemy
# and should never crash run.
try:
if doStoreCING2db(entryId, ARCHIVE_CASP_ID, project=project):
nTerror(
"Failed to store CING project's data to DB but continuing."
)
except:
nTtracebackError()
nTerror(
"Failed to store CING project's data due to above traceback error."
)
project.save()
if projectType == PROJECT_TYPE_CCPN:
# fileNameTgz = entryId + '.tgz'
# os.unlink(fileNameTgz) # temporary ccpn tgz
rmdir(entryId) # temporary ccpn dir
if tgzCing:
directoryNameCing = entryId + ".cing"
tgzFileNameCing = directoryNameCing + ".tgz"
if os.path.exists(tgzFileNameCing):
nTwarning("Overwriting: " + tgzFileNameCing)
cmd = "tar -czf %s %s" % (tgzFileNameCing, directoryNameCing)
do_cmd(cmd)
