mavutil: wait_heartbeat takes a timeout

ArduPilot · Jul 22, 2018 · 9143db6 · 9143db6
1 parent 52ec0bd
commit 9143db6
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/mavutil.py b/mavutil.py
@@ -389,9 +389,9 @@ def setup_logfile_raw(self, logfile, mode='w'):
         '''start logging raw bytes to the given logfile, without timestamps'''
         self.logfile_raw = open(logfile, mode=mode)
 
-    def wait_heartbeat(self, blocking=True):
+    def wait_heartbeat(self, blocking=True, timeout=None):
         '''wait for a heartbeat so we know the target system IDs'''
-        return self.recv_match(type='HEARTBEAT', blocking=blocking)
+        return self.recv_match(type='HEARTBEAT', blocking=blocking, timeout=timeout)
 
     def param_fetch_all(self):
         '''initiate fetch of all parameters'''