pgadmin-org
/
pgadmin4
mirror of https://github.com/pgadmin-org/pgadmin4.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Added the verification of region id for BigAnimal deployment.

pull/5349/head
Khushboo Vashi 2022-09-14 15:10:21 +05:30 committed by Akshay Joshi
parent 16bcb7d4a2
commit 40e0175ee5
1 changed files with 11 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
web/pgadmin/misc/cloud/biganimal/__init__.py
View File

@ -89,6 +89,7 @@ def biganimal_regions():
"""Get Regions.""" """Get Regions."""
biganimal_obj = pickle.loads(session['biganimal']['provider_obj']) biganimal_obj = pickle.loads(session['biganimal']['provider_obj'])
status, regions = biganimal_obj.get_regions() status, regions = biganimal_obj.get_regions()
session['biganimal']['provider_obj'] = pickle.dumps(biganimal_obj, -1)
return make_json_response(data=regions) return make_json_response(data=regions)
@ -162,6 +163,7 @@ class BigAnimalProvider():
self.access_token = None self.access_token = None
self.token_error = {} self.token_error = {}
self.token_status = -1 self.token_status = -1
self.regions = []
self.get_auth_provider() self.get_auth_provider()
def _get_headers(self): def _get_headers(self):
@ -272,6 +274,7 @@ class BigAnimalProvider():
'label': value['regionName'], 'label': value['regionName'],
'value': value['regionId'] 'value': value['regionId']
}) })
self.regions.append(value['regionId'])
return True, regions return True, regions
elif resp.content: elif resp.content:
regions_resp = json.loads(resp.content) regions_resp = json.loads(resp.content)
@ -315,6 +318,8 @@ class BigAnimalProvider():
def get_instance_types(self, region_id): def get_instance_types(self, region_id):
"""GEt Instance Types.""" """GEt Instance Types."""
if region_id not in self.regions:
return []
_url = "{0}/{1}".format( _url = "{0}/{1}".format(
self.BASE_URL, self.BASE_URL,
'cloud-providers/azure/regions/' 'cloud-providers/azure/regions/'
@ -327,6 +332,9 @@ class BigAnimalProvider():
def get_volume_types(self, region_id): def get_volume_types(self, region_id):
"""Get Volume Types.""" """Get Volume Types."""
if region_id not in self.regions:
return []
_url = "{0}/{1}".format( _url = "{0}/{1}".format(
self.BASE_URL, self.BASE_URL,
'cloud-providers/azure/regions/{0}/volume-types'.format(region_id)) 'cloud-providers/azure/regions/{0}/volume-types'.format(region_id))
@ -343,6 +351,9 @@ class BigAnimalProvider():
def get_volume_properties(self, region_id, volume_type): def get_volume_properties(self, region_id, volume_type):
"""Get Volume Properties.""" """Get Volume Properties."""
if region_id not in self.regions:
return []
_url = "{0}/{1}".format( _url = "{0}/{1}".format(
self.BASE_URL, self.BASE_URL,
'cloud-providers/azure/regions/{0}/volume-types' 'cloud-providers/azure/regions/{0}/volume-types'