Skip to content
Snippets Groups Projects
Commit e22267a7 authored by Yngve Levinsen's avatar Yngve Levinsen
Browse files

moved templates to separate file

parent 010fda7b
No related branches found
No related tags found
No related merge requests found
# supporting templates..
HEAD_TEMPLATE='''
universe = Vanilla
# always transfer files to tmp directory
should_transfer_files = yes
initialdir = {args.calc_dir}/Local_TraceWin_$(Process)
transfer_input_files = {input_files}
requirements = ( OpSys == "LINUX" || OpSys == "OSX" )
log = condor.log
executable = TraceWin_exe/tracewin.$$(OpSys).$$(Arch)
'''
HEAD_MULTI_TEMPLATE='''
universe = Vanilla
# always transfer files to tmp directory
should_transfer_files = yes
requirements = ( OpSys == "LINUX" || OpSys == "OSX" )
log = condor.log
executable = TraceWin_exe/tracewin.$$(OpSys).$$(Arch)
'''
QUEUE_TEMPLATE='''
priority = {q_priority}
arguments = 1 {stat_study} {rand}
queue 1
'''
QUEUE_MULTI_TEMPLATE='''
initialdir = {args.calc_dir}/{job_folder}
transfer_input_files = {input_files}
priority = {q_priority}
arguments = 1 {stat_study} {rand}
queue 1
'''
def get_head_template(filename='head.tmp'):
return _get_template(filename,HEAD_TEMPLATE)
def get_head_multi_template(filename='head.multi.tmp'):
return _get_template(filename,HEAD_MULTI_TEMPLATE)
def get_queue_template(filename='queue.tmp'):
return _get_template(filename,QUEUE_TEMPLATE)
def get_queue_multi_template(filename='queue.multi.tmp'):
return _get_template(filename,QUEUE_MULTI_TEMPLATE)
def _get_template(filename, template):
'''
Only to be used by get_NN_template()
'''
import os
if os.path.exists(filename):
return file(filename).read()
else:
return template
#!/usr/bin/env python #!/usr/bin/env python
from ess import __queue_templates__
def define_and_handle_args(): def define_and_handle_args():
import argparse import argparse
...@@ -131,8 +131,8 @@ def setup_multi(args): ...@@ -131,8 +131,8 @@ def setup_multi(args):
seeds.append(l.split()[-1]) seeds.append(l.split()[-1])
head_template=get_head_multi_template() head_template=__queue_templates__.get_head_multi_template()
queue_template=get_queue_multi_template() queue_template=__queue_templates__.get_queue_multi_template()
with file(args.calc_dir+"/multi.job","w") as out: with file(args.calc_dir+"/multi.job","w") as out:
out.write(head_template.format(**locals())) out.write(head_template.format(**locals()))
...@@ -196,8 +196,8 @@ def setup(args): ...@@ -196,8 +196,8 @@ def setup(args):
input_files[i]=input_files[i][:-2]+'_$(Process)' input_files[i]=input_files[i][:-2]+'_$(Process)'
input_files=', '.join(input_files) input_files=', '.join(input_files)
head_template=get_head_template() head_template=__queue_templates__.get_head_template()
queue_template=get_queue_template() queue_template=__queue_templates__.get_queue_template()
with file(args.calc_dir+"/submit.job","w") as out: with file(args.calc_dir+"/submit.job","w") as out:
out.write(head_template.format(**locals())) out.write(head_template.format(**locals()))
...@@ -231,57 +231,3 @@ if __name__=="__main__": ...@@ -231,57 +231,3 @@ if __name__=="__main__":
ask(args) ask(args)
execute(args.calc_dir,args.multi>1) execute(args.calc_dir,args.multi>1)
# supporting templates..
HEAD_TEMPLATE='''
universe = Vanilla
# always transfer files to tmp directory
should_transfer_files = yes
initialdir = {args.calc_dir}/Local_TraceWin_$(Process)
transfer_input_files = {input_files}
requirements = ( OpSys == "LINUX" || OpSys == "OSX" )
log = condor.log
executable = TraceWin_exe/tracewin.$$(OpSys).$$(Arch)
'''
HEAD_MULTI_TEMPLATE='''
universe = Vanilla
# always transfer files to tmp directory
should_transfer_files = yes
requirements = ( OpSys == "LINUX" || OpSys == "OSX" )
log = condor.log
executable = TraceWin_exe/tracewin.$$(OpSys).$$(Arch)
'''
QUEUE_TEMPLATE='''
priority = {q_priority}
arguments = 1 {stat_study} {rand}
queue 1
'''
QUEUE_MULTI_TEMPLATE='''
initialdir = {args.calc_dir}/{job_folder}
transfer_input_files = {input_files}
priority = {q_priority}
arguments = 1 {stat_study} {rand}
queue 1
'''
def get_head_template(filename='head.tmp'):
return _get_template(filename,QUEUE_TEMPLATE)
def get_head_multi_template(filename='head.multi.tmp'):
return _get_template(filename,QUEUE_MULTI_TEMPLATE)
def get_queue_template(filename='queue.tmp'):
return _get_template(filename,QUEUE_TEMPLATE)
def get_queue_multi_template(filename='queue.multi.tmp'):
return _get_template(filename,QUEUE_MULTI_TEMPLATE)
def _get_template(filename, template):
'''
Only to be used by get_NN_template()
'''
import os
if os.path.exists(filename):
return file(filename).read()
else:
return template
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment