| lothiraldan | |
| yuja |
| hg-reviewers |
So it can be used outside of instantiated classes. This is
needed to support pipe creation before init is called.
| Automatic diff as part of commit; lint not applicable. |
| Automatic diff as part of commit; unit tests not applicable. |
| return meth(data) | return meth(data) | ||||
| def close(self): | def close(self): | ||||
| return self._main.close() | return self._main.close() | ||||
| def flush(self): | def flush(self): | ||||
| return self._main.flush() | return self._main.flush() | ||||
| def _cleanuppipes(ui, pipei, pipeo, pipee): | |||||
| """Clean up pipes used by an SSH connection.""" | |||||
| if pipeo: | |||||
| pipeo.close() | |||||
| if pipei: | |||||
| pipei.close() | |||||
| if pipee: | |||||
| # Try to read from the err descriptor until EOF. | |||||
| try: | |||||
| for l in pipee: | |||||
| ui.status(_('remote: '), l) | |||||
| except (IOError, ValueError): | |||||
| pass | |||||
| pipee.close() | |||||
| class sshpeer(wireproto.wirepeer): | class sshpeer(wireproto.wirepeer): | ||||
| def __init__(self, ui, path, create=False, sshstate=None): | def __init__(self, ui, path, create=False, sshstate=None): | ||||
| self._url = path | self._url = path | ||||
| self._ui = ui | self._ui = ui | ||||
| self._pipeo = self._pipei = self._pipee = None | self._pipeo = self._pipei = self._pipee = None | ||||
| u = util.url(path, parsequery=False, parsefragment=False) | u = util.url(path, parsequery=False, parsefragment=False) | ||||
| self._path = u.path or '.' | self._path = u.path or '.' | ||||
| def _readerr(self): | def _readerr(self): | ||||
| _forwardoutput(self.ui, self._pipee) | _forwardoutput(self.ui, self._pipee) | ||||
| def _abort(self, exception): | def _abort(self, exception): | ||||
| self._cleanup() | self._cleanup() | ||||
| raise exception | raise exception | ||||
| def _cleanup(self): | def _cleanup(self): | ||||
| if self._pipeo is None: | _cleanuppipes(self.ui, self._pipei, self._pipeo, self._pipee) | ||||
| return | |||||
| self._pipeo.close() | |||||
| self._pipei.close() | |||||
| try: | |||||
| # read the error descriptor until EOF | |||||
| for l in self._pipee: | |||||
| self.ui.status(_("remote: "), l) | |||||
| except (IOError, ValueError): | |||||
| pass | |||||
| self._pipee.close() | |||||
| __del__ = _cleanup | __del__ = _cleanup | ||||
| def _submitbatch(self, req): | def _submitbatch(self, req): | ||||
| rsp = self._callstream("batch", cmds=wireproto.encodebatchcmds(req)) | rsp = self._callstream("batch", cmds=wireproto.encodebatchcmds(req)) | ||||
| available = self._getamount() | available = self._getamount() | ||||
| # TODO this response parsing is probably suboptimal for large | # TODO this response parsing is probably suboptimal for large | ||||
| # batches with large responses. | # batches with large responses. | ||||