Update os.py

Check if os provides link
Tinche · Aug 17, 2023 · bdd8a7f · bdd8a7f
1 parent 1456aa9
commit bdd8a7f
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/src/aiofiles/os.py b/src/aiofiles/os.py
@@ -16,14 +16,15 @@
     "makedirs",
     "rmdir",
     "removedirs",
-    "link",
     "symlink",
     "readlink",
     "listdir",
     "scandir",
     "access",
     "wrap",
 ]
+if hasattr(os, "link"):
+    __all__ += ["link"]
 if hasattr(os, "sendfile"):
     __all__ += ["sendfile"]
 if hasattr(os, "statvfs"):
@@ -40,13 +41,14 @@
 makedirs = wrap(os.makedirs)
 rmdir = wrap(os.rmdir)
 removedirs = wrap(os.removedirs)
-link = wrap(os.link)
 symlink = wrap(os.symlink)
 readlink = wrap(os.readlink)
 listdir = wrap(os.listdir)
 scandir = wrap(os.scandir)
 access = wrap(os.access)
 
+if hasattr(os, "link"):
+    link = wrap(os.link)
 if hasattr(os, "sendfile"):
     sendfile = wrap(os.sendfile)
 if hasattr(os, "statvfs"):