Update awaitme_niyazi_cetinkaya.py

Author: gorendes

Week05/awaitme_niyazi_cetinkaya.py

@@ -1,17 +1,11 @@
 import asyncio
 
-def awaitme(func):
-    async def wrapper(*args, **kwargs):
-        result = func(*args, **kwargs)
 
-        # Check if the result is a coroutine and await it if necessary
+
+def awaitme(func):
+    async def _wrapper(*args,**kwargs):
+        result = func(*args,**kwargs)
         if asyncio.iscoroutine(result):
             return await result
         return result
-
-    # This ensures the decorated function can be used both synchronously and asynchronously
-    if asyncio.iscoroutinefunction(func):
-        return wrapper
-    else:
-        # If it's not a coroutine function, return the original function
-        return func
+    return _wrapper