Support PNF In NSLCM
[vfc/nfvo/lcm.git] / lcm / ns_pnfs / biz / get_pnf.py
1 # Copyright 2018 ZTE Corporation.
2 #
3 # Licensed under the Apache License, Version 2.0 (the "License");
4 # you may not use this file except in compliance with the License.
5 # You may obtain a copy of the License at
6 #
7 #         http://www.apache.org/licenses/LICENSE-2.0
8 #
9 # Unless required by applicable law or agreed to in writing, software
10 # distributed under the License is distributed on an "AS IS" BASIS,
11 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12 # See the License for the specific language governing permissions and
13 # limitations under the License.
14
15 from lcm.pub.database.models import PNFInstModel
16
17
18 class GetPnf(object):
19     def __init__(self, filter=None, isIndividual=False):
20         self.filter = filter
21         self.isIndividual = isIndividual
22
23     def do_biz(self):
24         if self.filter and "pnfId" in self.filter:
25             pnf_instances = PNFInstModel.objects.filter(pnfId=self.filter["pnfId"])
26             if pnf_instances and self.isIndividual:
27                 return pnf_instances[0]
28             else:
29                 return pnf_instances
30         elif self.filter and "nsInstanceId" in self.filter:
31             return PNFInstModel.objects.filter(nsInstances__contains=self.filter["nsInstanceId"])
32         else:
33             return PNFInstModel.objects.all()