api.models: Add 'wait_for_node' method

Add method to Node model to wait
got node to get completed until specified
timeout.

Signed-off-by: Jeny Sadadia <[email protected]>

Author: Jeny Sadadia

api/models.py

@@ -6,6 +6,7 @@
 
 """KernelCI API model definitions"""
 
+import asyncio
 from datetime import datetime, timedelta
 from typing import Optional, Dict
 import enum
@@ -197,3 +198,11 @@ def set_timeout_status(self):
                 self.status = "timeout"
                 return True
         return False
+
+    @classmethod
+    async def wait_for_node(cls, timeout):
+        """Wait for node to get completed until timeout"""
+        current_time = datetime.utcnow()
+        if timeout > current_time:
+            time_delta = timeout - current_time
+            await asyncio.sleep(time_delta.total_seconds())