From 0a1c4da245e402707871e6a823959812c694f17c Mon Sep 17 00:00:00 2001
From: Tzu-Mainn Chen <tzumainn@redhat.com>
Date: Wed, 8 Oct 2014 04:01:05 +0000
Subject: [PATCH] Updated auto-discovery process with new api

Change-Id: I242779c31885725476db39c51b89db719b6bb51d
---
 tuskar_ui/api/node.py                   | 29 +++++++++++++++++++++++++
 tuskar_ui/infrastructure/nodes/forms.py |  4 +---
 2 files changed, 30 insertions(+), 3 deletions(-)

diff --git a/tuskar_ui/api/node.py b/tuskar_ui/api/node.py
index 2e445ee4c..dfd8cde87 100644
--- a/tuskar_ui/api/node.py
+++ b/tuskar_ui/api/node.py
@@ -10,8 +10,10 @@
 #    License for the specific language governing permissions and limitations
 #    under the License.
 
+import json
 import logging
 
+from django.conf import settings
 from django.utils.translation import ugettext_lazy as _
 from horizon.utils import memoized
 from ironicclient import client as ironic_client
@@ -19,6 +21,7 @@ from novaclient.v1_1.contrib import baremetal
 from openstack_dashboard.api import base
 from openstack_dashboard.api import glance
 from openstack_dashboard.api import nova
+import requests
 
 from tuskar_ui.cached_property import cached_property  # noqa
 from tuskar_ui.handle_errors import handle_errors  # noqa
@@ -189,6 +192,23 @@ class IronicNode(base.APIResourceWrapper):
         """
         return ironicclient(request).node.delete(uuid)
 
+    @classmethod
+    def discover(cls, request, uuids):
+        """Set the maintenance status of node
+
+        :param request: request object
+        :type  request: django.http.HttpRequest
+
+        :param uuids: IDs of IronicNodes
+        :type  uuids: list of str
+        """
+        url = getattr(settings, 'IRONIC_DISCOVERD_URL', None)
+        if url:
+            headers = {'content-type': 'application/json',
+                       'x-auth-token': request.user.token.id}
+            requests.post(url + "/v1/discover",
+                          data=json.dumps(uuids), headers=headers)
+
     @classmethod
     def set_maintenance(cls, request, uuid, maintenance):
         """Set the maintenance status of node
@@ -364,6 +384,11 @@ class BareMetalNode(base.APIResourceWrapper):
         """
         return baremetalclient(request).delete(uuid)
 
+    @classmethod
+    def discover(cls, request, uuids):
+        raise NotImplementedError(
+            "discover is not defined for Nova BareMetal nodes")
+
     @classmethod
     def set_maintenance(cls, request, uuid, maintenance):
         raise NotImplementedError(
@@ -533,6 +558,10 @@ class Node(base.APIResourceWrapper):
     def delete(cls, request, uuid):
         NodeClient(request).node_class.delete(request, uuid)
 
+    @classmethod
+    def discover(cls, request, uuids):
+        return NodeClient(request).node_class.discover(request, uuids)
+
     @classmethod
     def set_maintenance(cls, request, uuid, maintenance):
         node = NodeClient(request).node_class.set_maintenance(
diff --git a/tuskar_ui/infrastructure/nodes/forms.py b/tuskar_ui/infrastructure/nodes/forms.py
index 758a95432..c529a0cd8 100644
--- a/tuskar_ui/infrastructure/nodes/forms.py
+++ b/tuskar_ui/infrastructure/nodes/forms.py
@@ -64,9 +64,7 @@ def auto_discover_node(request, kwargs):
     api.node.Node.set_maintenance(request,
                                   node.uuid,
                                   True)
-    api.node.Node.set_power_state(request,
-                                  node.uuid,
-                                  'reboot')
+    api.node.Node.discover(request, [node.uuid])
 
 
 class BaseNodeForm(django.forms.Form):