1 from django.core.paginator import Paginator
2 from django.db import transaction
3 from django.utils.timezone import now
4 from pyqcg.service import Registry
5 from pyqcg.utils import Credential, TimePeriod
7 from qcg import constants
10 def get_attributes(obj, attrs):
11 return {name: getattr(obj, name) for name in attrs if getattr(obj, name) is not None}
14 def username_from_dn(dn):
15 _, username = dn.rsplit('=', 1)
21 def update_user_data(user, proxy):
22 from qcg.models import User, Job, Task, Allocation, NodeInfo
24 credential = Credential(proxy)
25 registry = Registry(credential)
27 # put lock on user record (hopefully..?)
28 user = User.objects.select_for_update().get(pk=user.pk)
30 changed_filter = {'changed': TimePeriod(after=user.last_update)}
32 ###################################
34 ###################################
35 for qcg_job in registry.jobs(**changed_filter):
36 params = Job.qcg_map(qcg_job, user)
37 job_id = params.pop('job_id')
39 Job.objects.update_or_create(job_id=job_id, defaults=params)
41 ###################################
43 ###################################
44 jobs_cache = {j.job_id: j for j in Job.objects.filter(owner=user)}
45 for qcg_task in registry.tasks(**changed_filter):
46 params = Task.qcg_map(qcg_task, jobs_cache)
47 task_id = params.pop('task_id')
49 task, created = Task.objects.update_or_create(job__job_id=qcg_task.job_id, task_id=task_id, defaults=params)
52 task.allocations.all().delete()
54 for qcg_alloc in qcg_task.allocations:
55 alloc = task.allocations.create(**Allocation.qcg_map(qcg_alloc))
57 for qcg_node in qcg_alloc.nodes:
58 alloc.nodes.create(**NodeInfo.qcg_map(qcg_node))
61 user.last_update = now()
65 def try_parse_int(s, default=None, base=10):
68 except (TypeError, ValueError):
72 def paginator_context(request, objects, per_page=constants.PER_PAGE):
73 paginator = Paginator(objects, per_page)
75 page_num = try_parse_int(request.GET.get('page'), 1)
76 if not (1 <= page_num <= paginator.num_pages):
79 pages_range = range(max(2, min(page_num - 2, paginator.num_pages - 4)),
80 min(max(page_num + 2, 5), paginator.num_pages) + 1)
82 return {'page': paginator.page(page_num), 'num_pages': paginator.num_pages, 'pages_range': pages_range}