Merge pull request #16 from cesbit/dev

use is_table

Author: joente

asyncsnmplib/client.py

@@ -91,7 +91,8 @@ async def get_next_multi(self, oids: Iterable[TOid]
         vbs = await self._get_next(oids)
         return [(oid, value) for oid, _, value in vbs if oid[:-1] in oids]
 
-    async def walk(self, oid: TOid) -> List[Tuple[TOid, TValue]]:
+    async def walk(self, oid: TOid, is_table: bool,
+                   ) -> List[Tuple[TOid, TValue]]:
         next_oid = oid
         prefixlen = len(oid)
         rows = []
@@ -103,7 +104,7 @@ async def walk(self, oid: TOid) -> List[Tuple[TOid, TValue]]:
                     # we're done
                     break
 
-                if next_oid[prefixlen + 1] == 0:
+                if is_table or next_oid[prefixlen + 1] == 0:
                     # this is a row we want in the result, otherwise
                     # we are in a table
                     if len(rows) == self.max_rows:
@@ -130,7 +131,8 @@ def close(self):
 class SnmpV1(Snmp):
     version = 0
 
-    async def walk(self, oid: TOid) -> List[Tuple[TOid, TValue]]:
+    async def walk(self, oid: TOid, is_table: bool,
+                   ) -> List[Tuple[TOid, TValue]]:
         next_oid = oid
         prefixlen = len(oid)
         rows = []
@@ -147,7 +149,7 @@ async def walk(self, oid: TOid) -> List[Tuple[TOid, TValue]]:
                     # we're done
                     break
 
-                if next_oid[prefixlen + 1] == 0:
+                if is_table or next_oid[prefixlen + 1] == 0:
                     # this is a row we want in the result, otherwise
                     # we are in a table
                     if len(rows) == self.max_rows: