X-Git-Url: http://mmka.chem.univ.gda.pl/gitweb/?a=blobdiff_plain;f=qcg%2Fservice.py;h=c87f40a83920a2280803a6817793ac7c948663b6;hb=0b053edee9793cef21c62b3ccb379eb304a22ccd;hp=f950aacc9a9c4057e71fe296d7525692166e6f96;hpb=e70a3ed3dfc6556ddad1bda7accb446ee5136604;p=qcg-portal.git

diff --git a/qcg/service.py b/qcg/service.py
index f950aac..c87f40a 100644
--- a/qcg/service.py
+++ b/qcg/service.py
@@ -85,7 +85,7 @@ def update_user_data(user, proxy):
 
 @transaction.atomic
 def update_job(job, proxy):
-    if job.get_status_display() in [JobStatus.FINISHED, JobStatus.FAILED, JobStatus.CANCELED]:
+    if job.terminated:
         return
 
     ts = time.time()
@@ -164,3 +164,19 @@ def submit_job(params, proxy):
     job = desc.submit()
 
     return job.job_id
+
+
+def cancel(obj, proxy):
+    ts = time.time()
+    QCG.start()
+
+    qcg_obj = obj.qcg_job if isinstance(obj, Job) else obj.qcg_task
+    qcg_obj.credential = Credential(proxy)
+
+    jts = time.time()
+    qcg_obj.cancel()
+    elapsed_cancel = 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)