aatish/dynamicweb
1
0
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity

Add error handling

Browse source
This commit is contained in:
Modulos 2017-05-08 02:47:38 +02:00
commit ae69a56ab2
1 changed files with 25 additions and 12 deletions
Download patch file Download diff file Expand all files Collapse all files

37
hosting/models.py
View file

@ -1,4 +1,5 @@
import os import os
import socket
import oca import oca
from django.db import models from django.db import models
@ -13,6 +14,8 @@ from membership.models import StripeCustomer, CustomUser
from utils.models import BillingAddress from utils.models import BillingAddress
from utils.mixins import AssignPermissionsMixin from utils.mixins import AssignPermissionsMixin
from .managers import VMPlansManager from .managers import VMPlansManager
from oca.exceptions import OpenNebulaException
from oca.pool import WrongNameError
class VirtualMachineType(models.Model): class VirtualMachineType(models.Model):
@ -273,21 +276,31 @@ class HostingBill(AssignPermissionsMixin, models.Model):
user_email = self.customer.user.email user_email = self.customer.user.email
# Connect to open nebula server # Connect to open nebula server
# TODO: handle potential connection error try:
client = oca.Client("{0}:{1}".format(settings.OPENNEBULA_USERNAME, client = oca.Client("{0}:{1}".format(settings.OPENNEBULA_USERNAME,
settings.OPENNEBULA_PASSWORD), settings.OPENNEBULA_PASSWORD),
"{protocol}://{domain}:{port}{endpoint}".format( "{protocol}://{domain}:{port}{endpoint}".format(
protocol=settings.OPENNEBULA_PROTOCOL, protocol=settings.OPENNEBULA_PROTOCOL,
domain=settings.OPENNEBULA_DOMAIN, domain=settings.OPENNEBULA_DOMAIN,
port=settings.OPENNEBULA_PORT, port=settings.OPENNEBULA_PORT,
endpoint=settings.OPENNEBULA_ENDPOINT endpoint=settings.OPENNEBULA_ENDPOINT
)) ))
except OpenNebulaException as err:
logger.error("Error : {0}".format(err))
return None
except socket.timeout:
logger.error("Socket timeout error.")
return None
# Get open nebula user id for given email # Get open nebula user id for given email
user_pool = oca.UserPool(client) user_pool = oca.UserPool(client)
user_pool.info() user_pool.info()
# TODO: handle potential name error try:
user_id = user_pool.get_by_name(user_email).id user_id = user_pool.get_by_name(user_email).id
except WrongNameError as wrong_name_err:
logger.error("User {0} does not exist.", user_email)
return None
# Get vm_pool for given user_id # Get vm_pool for given user_id
vm_pool = oca.VirtualMachinePool(client) vm_pool = oca.VirtualMachinePool(client)
vm_pool.info(filter=user_id) vm_pool.info(filter=user_id)