obscvty: Implement readlines() generator function.

This commit is contained in:
Alexander Chemeris 2015-05-30 10:07:53 -04:00 committed by Holger Hans Peter Freyther
parent b819b57a98
commit 2f48313ed6
1 changed files with 19 additions and 0 deletions

View File

@ -127,6 +127,25 @@ class VTYInteract(object):
self._close_socket()
return res[len(request) + 2: -end]
"""A generator function yielding lines separated by delim.
Behaves similar to a file readlines() method.
Example of use:
for line in vty.readlines():
print line
"""
def readlines(self, recv_buffer=4096, delim='\n'):
buffer = ''
data = True
while data:
data = self.socket.recv(recv_buffer)
buffer += data
while buffer.find(delim) != -1:
line, buffer = buffer.split('\n', 1)
yield line
return
# There's no close parameter, as close=True makes this useless
def enable(self):
self.command("enable")