Fix Py file borrower to become functional

Apparently, `PyFileBorrower` uses default `FileReader` which implies
using MIB files extensions rather than .py. This commit makes
`FileReader` objects using alternative extensions when being run
by `PyFileBorrower`.

CHANGES.rst

@@ -5,6 +5,7 @@ Revision 0.3.4, XX-01-2019
 - Added `implied` key to JSON SNMP table index structure
 - Rebased MIB importing code onto `importlib` because `imp` is long
   deprecated
+- Fixed Py file borrower to become functional
 
 Revision 0.3.3, 29-12-2018
 --------------------------

pysmi/borrower/base.py

@@ -50,4 +50,4 @@ def getData(self, mibname, **kwargs):
             debug.logger('trying to borrow file %s from %s' % (mibname, self._reader))
         )
 
-        return self._reader.getData(mibname)
+        return self._reader.getData(mibname, exts=self.exts)

pysmi/reader/base.py

@@ -22,7 +22,7 @@ def setOptions(self, **kwargs):
             setattr(self, k, kwargs[k])
         return self
 
-    def getMibVariants(self, mibname):
+    def getMibVariants(self, mibname, exts=None):
         filenames = []
 
         if self.originalMatching:
@@ -45,7 +45,7 @@ def getMibVariants(self, mibname):
                 filenames.append(suffixed.upper())
                 filenames.append(suffixed.lower())
 
-        return ((x, x + y) for x in filenames for y in self.exts)
+        return ((x, x + y) for x in filenames for y in exts or self.exts)
 
-    def getData(self, filename):
+    def getData(self, filename, exts=None):
         raise NotImplementedError()

pysmi/reader/callback.py

@@ -33,7 +33,7 @@ def __init__(self, cbFun, cbCtx=None):
     def __str__(self):
         return '%s{"%s"}' % (self.__class__.__name__, self._cbFun)
 
-    def getData(self, mibname):
+    def getData(self, mibname, exts=None):
         debug.logger & debug.flagReader and debug.logger('calling user callback %s for MIB %s' % (self._cbFun, mibname))
 
         res = self._cbFun(mibname, self._cbCtx)

pysmi/reader/ftpclient.py

@@ -50,7 +50,7 @@ def __init__(self, host, locationTemplate, timeout=5, ssl=False, port=21,
     def __str__(self):
         return '%s{"ftp://%s%s"}' % (self.__class__.__name__, self._host, self._locationTemplate)
 
-    def getData(self, mibname):
+    def getData(self, mibname, exts=None):
         if self._ssl:
             conn = ftplib.FTP_TLS()
         else:
@@ -74,7 +74,7 @@ def getData(self, mibname):
 
         debug.logger & debug.flagReader and debug.logger('looking for MIB %s' % mibname)
 
-        for mibalias, mibfile in self.getMibVariants(mibname):
+        for mibalias, mibfile in self.getMibVariants(mibname, exts):
             location = self._locationTemplate.replace('@mib@', mibfile)
 
             mtime = time.time()

pysmi/reader/httpclient.py

@@ -62,7 +62,7 @@ def __init__(self, host, port, locationTemplate, timeout=5, ssl=False):
     def __str__(self):
         return self._url
 
-    def getData(self, mibname):
+    def getData(self, mibname, exts=None):
         headers = {
             'Accept': 'text/plain',
             'User-Agent': self._user_agent
@@ -72,7 +72,7 @@ def getData(self, mibname):
 
         debug.logger & debug.flagReader and debug.logger('looking for MIB %s' % mibname)
 
-        for mibalias, mibfile in self.getMibVariants(mibname):
+        for mibalias, mibfile in self.getMibVariants(mibname, exts):
             if self.MIB_MAGIC in self._url:
                 url = self._url.replace(self.MIB_MAGIC, mibfile)
             else:

pysmi/reader/localfile.py

@@ -83,7 +83,7 @@ def loadIndex(indexFile):
 
         return mibIndex
 
-    def getMibVariants(self, mibname):
+    def getMibVariants(self, mibname, exts=None):
         if self.useIndexFile:
             if not self._indexLoaded:
                 self._mibIndex = self.loadIndex(
@@ -96,7 +96,7 @@ def getMibVariants(self, mibname):
                     'found %s in MIB index: %s' % (mibname, self._mibIndex[mibname]))
                 return [(mibname, self._mibIndex[mibname])]
 
-        return super(FileReader, self).getMibVariants(mibname)
+        return super(FileReader, self).getMibVariants(mibname, exts)
 
     def getData(self, mibname):
         debug.logger & debug.flagReader and debug.logger(

pysmi/reader/zipreader.py

@@ -156,7 +156,7 @@ def _readZipFile(self, refs):
     def __str__(self):
         return '%s{"%s"}' % (self.__class__.__name__, self._name)
 
-    def getData(self, mibname, zipBlob=None):
+    def getData(self, mibname, exts=None, zipBlob=None):
         debug.logger & debug.flagReader and debug.logger('looking for MIB %s at %s' % (mibname, self._name))
 
         if self._pendingError:
@@ -165,7 +165,7 @@ def getData(self, mibname, zipBlob=None):
         if not self._members:
             raise error.PySmiReaderFileNotFoundError('source MIB %s not found' % mibname, reader=self)
 
-        for mibalias, mibfile in self.getMibVariants(mibname):
+        for mibalias, mibfile in self.getMibVariants(mibname, exts):
 
             debug.logger & debug.flagReader and debug.logger('trying MIB %s' % mibfile)