Add error handling

hosting/models.py

@@ -1,4 +1,5 @@
 import os
+import socket
 
 import oca
 from django.db import models
@@ -13,6 +14,8 @@ from membership.models import StripeCustomer, CustomUser
 from utils.models import BillingAddress
 from utils.mixins import AssignPermissionsMixin
 from .managers import VMPlansManager
+from oca.exceptions import OpenNebulaException
+from oca.pool import WrongNameError
 
 
 class VirtualMachineType(models.Model):
@@ -273,7 +276,7 @@ class HostingBill(AssignPermissionsMixin, models.Model):
         user_email = self.customer.user.email
 
         # Connect to open nebula server
-        # TODO: handle potential connection error
+        try:
             client = oca.Client("{0}:{1}".format(settings.OPENNEBULA_USERNAME,
                                                  settings.OPENNEBULA_PASSWORD),
                                 "{protocol}://{domain}:{port}{endpoint}".format(
@@ -282,12 +285,22 @@ class HostingBill(AssignPermissionsMixin, models.Model):
                                     port=settings.OPENNEBULA_PORT,
                                     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
         user_pool = oca.UserPool(client)
         user_pool.info()
-        # TODO: handle potential name error
+        try:
             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
         vm_pool = oca.VirtualMachinePool(client)
         vm_pool.info(filter=user_id)