Merge pull request #46 from p0dalirius/fix-missing-backslash-at-end-o…

…f-prompt [bugfix] Fix missing backslash at the end of prompt fixes #45
p0dalirius · Jun 17, 2024 · 0cd83d7 · 0cd83d7
2 parents 92d8a63 + 2d2662b
commit 0cd83d7
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/smbclientng/core/InteractiveShell.py b/smbclientng/core/InteractiveShell.py
@@ -987,7 +987,13 @@ def __prompt(self):
             else:
                 str_prompt = "%s[\x1b[1;94m\\\\%s\\\x1b[0m]> " % (connected_dot, self.smbSession.address)
         else:
-            str_path = "\\\\%s\\%s\\%s" % (self.smbSession.address, self.smbSession.smb_share, self.smbSession.smb_cwd.lstrip(ntpath.sep))
+            if len(self.smbSession.smb_cwd) == 0:
+                current_path = ""
+            else:
+                current_path = self.smbSession.smb_cwd.strip(ntpath.sep) + ntpath.sep
+
+            str_path = "\\\\%s\\%s\\%s" % (self.smbSession.address, self.smbSession.smb_share, current_path)
+
             if self.config.no_colors:
                 str_prompt = "%s[%s]> " % (connected_dot, str_path)
             else: