fetch resources info and set choices in job submission form

[qcg-portal.git] / qcg / utils.py

diff --git a/qcg/utils.py b/qcg/utils.py
index 37a5b1a..da8f83c 100644 (file)
--- a/qcg/utils.py
+++ b/qcg/utils.py
@@ -7,6 +7,7 @@ import string
 import random
 
 from django.contrib.auth.decorators import login_required
+from django.core.cache import caches
 from django.core.paginator import Paginator
 from django.utils.formats import date_format
 from django.utils.timezone import localtime
@@ -17,8 +18,12 @@ from pyqcg.description import JobDescription
 from filex.ftp import FTPOperation
 from qcg import constants
 
+
 from django.utils import encoding
 
+resources_cache = caches['resources']
+
+
 def get_attributes(obj, attrs):
     return {name: getattr(obj, name) for name in attrs if getattr(obj, name) is not None}
 
@@ -194,3 +199,24 @@ def to_form_data(xml):
 
 def restricted(view):
     return wraps(view)(cache_control(no_cache=True, must_revalidate=True, no_store=True)(login_required(view)))
+
+
+def cached_resources(proxy):
+    hosts = resources_cache.get('hosts')
+    if hosts is None:
+        # prevent circular import errors
+        from qcg.service import fetch_resources
+
+        hosts, _, applications, modules = map(make_choices, fetch_resources(proxy))
+        resources_cache.set('hosts', hosts)
+        resources_cache.set('applications', applications)
+        resources_cache.set('modules', modules)
+    else:
+        applications = resources_cache.get('applications')
+        modules = resources_cache.get('modules')
+
+    return hosts, applications, modules
+
+
+def make_choices(iterable):
+    return ((None, ''),) + tuple((item, item) for item in sorted(iterable))