A try to fix imports

Author: HardNorth

reportportal_client/aio/__init__.py

@@ -12,8 +12,6 @@
 #  limitations under the License
 
 """Common package for Asynchronous I/O clients and utilities."""
-import asyncio
-from typing import Coroutine, Optional, TypeVar, Union
 
 from reportportal_client.aio.client import (
     DEFAULT_SHUTDOWN_TIMEOUT,
@@ -26,25 +24,7 @@
 )
 from reportportal_client.aio.tasks import BlockingOperationError, EmptyTask, Task
 
-_T = TypeVar("_T")
-
-
-async def await_if_necessary(obj: Union[_T, Task[_T], Coroutine[_T, None, None]]) -> Optional[_T]:
-    """Await Coroutine, Feature or coroutine Function if given argument is one of them, or return immediately.
-
-    :param obj: value, Coroutine, Feature or coroutine Function
-    :return: result which was returned by Coroutine, Feature or coroutine Function
-    """
-    if obj:
-        if asyncio.isfuture(obj) or asyncio.iscoroutine(obj):
-            return await obj
-        elif asyncio.iscoroutinefunction(obj):
-            return await obj()
-    return obj
-
-
 __all__ = [
-    "await_if_necessary",
     "Task",
     "EmptyTask",
     "BlockingOperationError",

reportportal_client/aio/client.py

@@ -56,10 +56,10 @@
 # noinspection PyProtectedMember
 from reportportal_client._internal.static.abstract import AbstractBaseClass, abstractmethod
 from reportportal_client._internal.static.defines import DEFAULT_LOG_LEVEL
-from reportportal_client.aio import EmptyTask, await_if_necessary
 
 # noinspection PyProtectedMember
-from reportportal_client.aio.tasks import Task
+from reportportal_client.aio.tasks import EmptyTask, Task
+from reportportal_client.aio.util import await_if_necessary
 from reportportal_client.client import RP, OutputType
 from reportportal_client.core.rp_issues import Issue
 from reportportal_client.core.rp_requests import (

reportportal_client/aio/util.py

@@ -0,0 +1,34 @@
+#  Copyright 2026 EPAM Systems
+#
+#  Licensed under the Apache License, Version 2.0 (the "License");
+#  you may not use this file except in compliance with the License.
+#  You may obtain a copy of the License at
+#
+#      https://www.apache.org/licenses/LICENSE-2.0
+#
+#  Unless required by applicable law or agreed to in writing, software
+#  distributed under the License is distributed on an "AS IS" BASIS,
+#  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+#  See the License for the specific language governing permissions and
+#  limitations under the License.
+
+import asyncio
+from typing import Coroutine, Optional, TypeVar, Union
+
+from reportportal_client.aio.tasks import Task
+
+_T = TypeVar("_T")
+
+
+async def await_if_necessary(obj: Union[_T, Task[_T], Coroutine[_T, None, None]]) -> Optional[_T]:
+    """Await Coroutine, Feature or coroutine Function if given argument is one of them, or return immediately.
+
+    :param obj: value, Coroutine, Feature or coroutine Function
+    :return: result which was returned by Coroutine, Feature or coroutine Function
+    """
+    if obj:
+        if asyncio.isfuture(obj) or asyncio.iscoroutine(obj):
+            return await obj
+        elif asyncio.iscoroutinefunction(obj):
+            return await obj()
+    return obj

reportportal_client/core/rp_requests.py

@@ -37,7 +37,7 @@
 
 # noinspection PyProtectedMember
 from reportportal_client._internal.static.defines import DEFAULT_LOG_LEVEL, DEFAULT_PRIORITY, LOW_PRIORITY, Priority
-from reportportal_client.aio import await_if_necessary
+from reportportal_client.aio.util import await_if_necessary
 from reportportal_client.core.rp_file import RPFile
 from reportportal_client.core.rp_issues import Issue
 from reportportal_client.core.rp_responses import AsyncRPResponse, RPResponse