From b360e0e417f787e0266268596d630b87e88283d1 Mon Sep 17 00:00:00 2001 From: Parker Berberian Date: Tue, 3 Mar 2020 12:30:36 -0500 Subject: Fixing and adding tests for the new resource models. Change-Id: I4dd0569411b415a3e8a8de43d4b99e927e5b7821 Signed-off-by: Parker Berberian --- src/resource_inventory/resource_manager.py | 7 +++++++ 1 file changed, 7 insertions(+) (limited to 'src/resource_inventory') diff --git a/src/resource_inventory/resource_manager.py b/src/resource_inventory/resource_manager.py index e14218b..c8b2b05 100644 --- a/src/resource_inventory/resource_manager.py +++ b/src/resource_inventory/resource_manager.py @@ -7,11 +7,13 @@ # http://www.apache.org/licenses/LICENSE-2.0 ############################################################################## import re +from django.db.models import Q from dashboard.exceptions import ResourceAvailabilityException from resource_inventory.models import ( ResourceBundle, + ResourceTemplate, Network, Vlan, PhysicalNetwork, @@ -31,6 +33,11 @@ class ResourceManager: ResourceManager.instance = ResourceManager() return ResourceManager.instance + def getAvailableResourceTemplates(self, lab, user): + templates = ResourceTemplate.objects.filter(lab=lab) + templates.filter(Q(owner=user) | Q(public=True)) + return templates + def templateIsReservable(self, resource_template): """ Check if the required resources to reserve this template is available. -- cgit 1.2.3-korg