A context manager to protect command execution from UNC paths. In the Win32 API, commands can't be invoked with the cwd being a UNC path. This context manager temporarily changes directory to the 'C:' drive on entering, and restores the original working directory on exit. The conte
| 150 | LocalFree.restype = HLOCAL |
| 151 | |
| 152 | class AvoidUNCPath(object): |
| 153 | """A context manager to protect command execution from UNC paths. |
| 154 | |
| 155 | In the Win32 API, commands can't be invoked with the cwd being a UNC path. |
| 156 | This context manager temporarily changes directory to the 'C:' drive on |
| 157 | entering, and restores the original working directory on exit. |
| 158 | |
| 159 | The context manager returns the starting working directory *if* it made a |
| 160 | change and None otherwise, so that users can apply the necessary adjustment |
| 161 | to their system calls in the event of a change. |
| 162 | |
| 163 | Examples |
| 164 | -------- |
| 165 | :: |
| 166 | cmd = 'dir' |
| 167 | with AvoidUNCPath() as path: |
| 168 | if path is not None: |
| 169 | cmd = '"pushd %s &&"%s' % (path, cmd) |
| 170 | os.system(cmd) |
| 171 | """ |
| 172 | def __enter__(self): |
| 173 | self.path = os.getcwd() |
| 174 | self.is_unc_path = self.path.startswith(r"\\") |
| 175 | if self.is_unc_path: |
| 176 | # change to c drive (as cmd.exe cannot handle UNC addresses) |
| 177 | os.chdir("C:") |
| 178 | return self.path |
| 179 | else: |
| 180 | # We return None to signal that there was no change in the working |
| 181 | # directory |
| 182 | return None |
| 183 | |
| 184 | def __exit__(self, exc_type, exc_value, traceback): |
| 185 | if self.is_unc_path: |
| 186 | os.chdir(self.path) |
| 187 | |
| 188 | |
| 189 | class Win32ShellCommandController(object): |