Kiln » Dependencies » Dulwich Read More
Clone URL:  
Pushed to one repository · View In Graph Contained in master-1, master-0, and 0.9.4

Add ParamikoSSHVendor.

Changeset fcd6aa89ee95…

Parent 1b63e1e19b5b…

committed by Jelmer Vernooij

authored by Aaron O'Mullan

Changes to 2 files · Browse files at fcd6aa89ee95 Showing diff from parent 1b63e1e19b5b Diff from another changeset...

Change 1 of 1 Show Entire File NEWS Stacked
 
24
25
26
 
 
27
28
29
 
24
25
26
27
28
29
30
31
@@ -24,6 +24,8 @@
    * Add `MemoryObjectStore.add_pack` and `MemoryObjectStore.add_thin_pack` methods.   (David Bennett) + + * Add paramiko-based SSH vendor. (Aaron O'Mullan)    0.9.0 2013-05-31  
Change 1 of 1 Show Entire File dulwich/​client.py Stacked
 
671
672
673
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
674
675
676
 
671
672
673
674
675
676
677
678
679
680
681
682
683
684
685
686
687
688
689
690
691
692
693
694
695
696
697
698
699
700
701
702
703
704
705
706
707
708
709
710
711
712
713
714
715
716
717
718
719
720
721
722
723
724
725
726
727
728
729
730
731
732
733
734
735
736
737
738
739
740
741
742
743
744
745
746
747
748
749
750
751
752
753
754
755
756
757
758
759
760
761
762
763
764
765
766
767
768
769
770
771
772
773
774
775
776
777
778
779
@@ -671,6 +671,109 @@
  return SubprocessWrapper(proc)     +try: + import paramiko +except ImportError: + pass +else: + import threading + + class ParamikoWrapper(object): + STDERR_READ_N = 2048 # 2k + + def __init__(self, client, channel, progress_stderr=None): + self.client = client + self.channel = channel + self.progress_stderr = progress_stderr + self.should_monitor = bool(progress_stderr) or True + self.monitor_thread = None + self.stderr = '' + + # Channel must block + self.channel.setblocking(True) + + # Start + if self.should_monitor: + self.monitor_thread = threading.Thread(target=self.monitor_stderr) + self.monitor_thread.start() + + def monitor_stderr(self): + while self.should_monitor: + # Block and read + data = self.read_stderr(self.STDERR_READ_N) + + # Socket closed + if not data: + self.should_monitor = False + break + + # Emit data + if self.progress_stderr: + self.progress_stderr(data) + + # Append to buffer + self.stderr += data + + def stop_monitoring(self): + # Stop StdErr thread + if self.should_monitor: + self.should_monitor = False + self.monitor_thread.join() + + # Get left over data + data = self.channel.in_stderr_buffer.empty() + self.stderr += data + + def can_read(self): + return self.channel.recv_ready() + + def write(self, data): + return self.channel.sendall(data) + + def read_stderr(self, n): + return self.channel.recv_stderr(n) + + def read(self, n=None): + data = self.channel.recv(n) + data_len = len(data) + + # Closed socket + if not data: + return + + # Read more if needed + if n and data_len < n: + diff_len = n - data_len + return data + self.read(diff_len) + return data + + def close(self): + self.channel.close() + self.stop_monitoring() + + def __del__(self): + self.close() + + class ParamikoSSHVendor(object): + + def connect_ssh(self, host, command, username=None, port=None, + progress_stderr=None, **kwargs): + client = paramiko.SSHClient() + + policy = paramiko.client.MissingHostKeyPolicy() + client.set_missing_host_key_policy(policy) + client.connect(host, username=username, port=port, **kwargs) + + # Open SSH session + channel = client.get_transport().open_session() + + # Run commands + apply(channel.exec_command, command) + + return ParamikoWrapper(client, channel, + progress_stderr=progress_stderr) + +  # Can be overridden by users  get_ssh_vendor = SubprocessSSHVendor