from django.utils.functional import SimpleLazyObject
from django.utils.timezone import now
from pyqcg import QCG
-from pyqcg.description import JobDescription
from pyqcg.service import Registry
from pyqcg.utils import Credential, TimePeriod, JobStatus, TaskStatus
@transaction.atomic
def update_job(job, proxy):
- if job.terminated:
- return
-
ts = time.time()
QCG.start()
elapsed, job.job_id, elapsed_job, job.tasks.count(), elapsed_tasks, elapsed_py)
-def submit_job(params, proxy):
+def cancel(obj, proxy):
+ ts = time.time()
QCG.start()
- desc = JobDescription(Credential(proxy))
-
- direct_map = ('env_variables', 'executable', 'arguments', 'note', 'grant', 'hosts', 'properties', 'queue', 'procs',
- 'wall_time', 'memory', 'memory_per_slot', 'modules', 'input', 'stage_in', 'native', 'notify',
- 'preprocess', 'postprocess', 'persistent')
-
- for name in direct_map:
- if params[name]:
- setattr(desc, name, params[name])
- if params['application']:
- desc.set_application(*params['application'])
- if params['nodes']:
- desc.set_nodes(*params['nodes'])
- if params['reservation']:
- desc.set_reservation(params['reservation'])
- if params['watch_output']:
- desc.set_watch_output(params['watch_output'], params['watch_output_pattern'])
- # TODO script
- # TODO monitoring
-
- # for prop in direct_map + ('application', 'nodes', 'env_variables', 'reservation', 'watch_output'):
- # print prop, type(getattr(desc, prop)), repr(getattr(desc, prop))
-
- # print desc.xml_description
+ qcg_obj = obj.qcg_job if isinstance(obj, Job) else obj.qcg_task
+ qcg_obj.credential = Credential(proxy)
- job = desc.submit()
+ jts = time.time()
+ qcg_obj.cancel()
+ elapsed_cancel = time.time() - jts
- return job.job_id
+ elapsed = time.time() - ts
+ elapsed_py = elapsed - elapsed_cancel
+ logger.info('(%.3f) OBJ = %s (%.3f), TIME = %.3f', elapsed, obj, elapsed_cancel, elapsed_py)
-def cancel(obj, proxy):
+def clean(obj, proxy):
ts = time.time()
QCG.start()
qcg_obj.credential = Credential(proxy)
jts = time.time()
- qcg_obj.cancel()
- elapsed_cancel = time.time() - jts
+ qcg_obj.clean()
+ elapsed_clean = time.time() - jts
elapsed = time.time() - ts
- elapsed_py = elapsed - elapsed_cancel
- logger.info('(%.3f) OBJ = %s (%.3f), TIME = %.3f', elapsed, obj, elapsed_cancel, elapsed_py)
+ elapsed_py = elapsed - elapsed_clean
+ logger.info('(%.3f) OBJ = %s (%.3f), TIME = %.3f', elapsed, obj, elapsed_clean, elapsed_py)