✨ SFTPTextualPath

textual_universal_directorytree/__init__.py

@@ -7,6 +7,7 @@
 from textual_universal_directorytree.alternate_paths import (
     GitHubTextualPath,
     S3TextualPath,
+    SFTPTextualPath,
 )
 from textual_universal_directorytree.universal_directory_tree import (
     UniversalDirectoryTree,
@@ -16,6 +17,8 @@
 registry.register_implementation(protocol="github", cls=GitHubTextualPath, clobber=True)
 registry.register_implementation(protocol="s3", cls=S3TextualPath, clobber=True)
 registry.register_implementation(protocol="s3a", cls=S3TextualPath, clobber=True)
+registry.register_implementation(protocol="ssh", cls=SFTPTextualPath, clobber=True)
+registry.register_implementation(protocol="sftp", cls=SFTPTextualPath, clobber=True)
 
 __all__ = [
     "UniversalDirectoryTree",
@@ -24,4 +27,5 @@
     "GitHubTextualPath",
     "S3TextualPath",
     "UPath",
+    "SFTPTextualPath",
 ]

textual_universal_directorytree/alternate_paths.py

@@ -8,6 +8,7 @@
 import re
 from typing import Any
 
+from upath import UPath
 from upath.implementations.cloud import S3Path
 from upath.implementations.github import GitHubPath
 
@@ -116,3 +117,25 @@ def name(self) -> str:
             return f"{self._url.scheme}://{self._url.netloc}"
         else:
             return super().name
+
+
+class SFTPTextualPath(UPath):
+    """
+    SFTPTextualPath
+    """
+
+    def __str__(self) -> str:
+        """
+        String representation of the SFTPPath
+        """
+        string_representation = f"{self.protocol}://"
+        if "username" in self.storage_options:
+            string_representation += f"{self.storage_options['username']}@"
+        string_representation += f"{self.storage_options['host']}"
+        if "port" in self.storage_options:
+            string_representation += f":{self.storage_options['port']}"
+        if self.path == ".":
+            string_representation += "/"
+        else:
+            string_representation += f"/{self.path}"
+        return string_representation