72 lines
2.5 KiB
Python
72 lines
2.5 KiB
Python
from rest_framework import serializers
|
|
|
|
from orchestra.api import router
|
|
from orchestra.utils import running_syncdb
|
|
|
|
from .models import Resource, ResourceData
|
|
|
|
|
|
class ResourceSerializer(serializers.ModelSerializer):
|
|
name = serializers.SerializerMethodField('get_name')
|
|
|
|
class Meta:
|
|
model = ResourceData
|
|
fields = ('name', 'used', 'allocated')
|
|
read_only_fields = ('used',)
|
|
|
|
def get_name(self, instance):
|
|
return instance.resource.name
|
|
|
|
def get_identity(self, data):
|
|
return data.get('name')
|
|
|
|
|
|
# Monkey-patching section
|
|
|
|
if not running_syncdb():
|
|
# TODO why this is even loaded during syncdb?
|
|
for resources in Resource.group_by_content_type():
|
|
model = resources[0].content_type.model_class()
|
|
try:
|
|
router.insert(model, 'resources', ResourceSerializer, required=False, many=True)
|
|
except KeyError:
|
|
continue
|
|
|
|
def validate_resources(self, attrs, source, _resources=resources):
|
|
""" Creates missing resources """
|
|
posted = attrs.get(source, [])
|
|
result = []
|
|
resources = list(_resources)
|
|
for data in posted:
|
|
resource = data.resource
|
|
if resource not in resources:
|
|
msg = "Unknown or duplicated resource '%s'." % resource
|
|
raise serializers.ValidationError(msg)
|
|
resources.remove(resource)
|
|
if not resource.ondemand and not data.allocated:
|
|
data.allocated = resource.default_allocation
|
|
result.append(data)
|
|
for resource in resources:
|
|
data = ResourceData(resource=resource)
|
|
if not resource.ondemand:
|
|
data.allocated = resource.default_allocation
|
|
result.append(data)
|
|
attrs[source] = result
|
|
return attrs
|
|
viewset = router.get_viewset(model)
|
|
viewset.serializer_class.validate_resources = validate_resources
|
|
|
|
old_metadata = viewset.metadata
|
|
def metadata(self, request, resources=resources):
|
|
""" Provides available resources description """
|
|
ret = old_metadata(self, request)
|
|
ret['available_resources'] = [
|
|
{
|
|
'name': resource.name,
|
|
'ondemand': resource.ondemand,
|
|
'default_allocation': resource.default_allocation
|
|
} for resource in resources
|
|
]
|
|
return ret
|
|
viewset.metadata = metadata
|