tools: punchboot: bufix: wait_for_device

Actually try to find a device within the specified timeout. Before this
commit the code would return early if no device was enumerated which is
not the expected behaviour.

tools/punchboot/python_wrapper.c

@@ -1067,13 +1067,12 @@ static PyObject *wait_for_device(PyObject *self, PyObject *args, PyObject *kwds)
 
     while (true) {
         rc = init_transport(NULL, NULL, &ctx);
-        if (rc != PB_RESULT_OK) {
-            PyErr_SetString(PyExc_TimeoutError, "No device found");
-            return NULL;
+
+        if (rc == 0) {
+            rc = get_uuid(ctx, uuid);
+            pb_api_free_context(ctx);
         }
 
-        rc = get_uuid(ctx, uuid);
-        pb_api_free_context(ctx);
         if (rc != PB_RESULT_OK) {
             if (timeout > 0) {
                 sleep(1);