from pysmi.borrower import PyFileBorrower
from pysmi.writer import PyFileWriter
from pysmi.parser import NullParser
from pysmi.codegen import NullCodeGen
from pysmi.compiler import MibCompiler
# from pysmi import debug
# debug.setLogger(debug.Debug('compiler'))
inputMibs = ['BORROWED-MIB']
httpBorrowers = [('mibs.snmplabs.com', 80, '/pysnmp/notexts/@mib@')]
dstDirectory = '.pysnmp-mibs'
# Initialize compiler infrastructure
mibCompiler = MibCompiler(NullParser(), NullCodeGen(),
PyFileWriter(dstDirectory))
# check compiled/borrowed MIBs in our own productions
mibCompiler.addSearchers(PyFileSearcher(dstDirectory))
# search for precompiled MIBs at Web sites
mibCompiler.addBorrowers(
*[PyFileBorrower(HttpReader(*x)) for x in httpBorrowers])
# run MIB compilation
results = mibCompiler.compile(*inputMibs)
print('Results: %s' % ', '.join(['%s:%s' % (x, results[x]) for x in results]))
mibStubs = [x for x in PySnmpCodeGen.baseMibs if x not in PySnmpCodeGen.fakeMibs]
if not mibBorrowers:
mibBorrowers = [('http://mibs.snmplabs.com/pysnmp/notexts/@mib@', False),
('http://mibs.snmplabs.com/pysnmp/fulltexts/@mib@', True)]
if not dstDirectory:
dstDirectory = os.path.expanduser("~")
if sys.platform[:3] == 'win':
dstDirectory = os.path.join(dstDirectory, 'PySNMP Configuration', 'mibs')
else:
dstDirectory = os.path.join(dstDirectory, '.pysnmp', 'mibs')
# Compiler infrastructure
borrowers = [PyFileBorrower(x[1], genTexts=mibBorrowers[x[0]][1])
for x in enumerate(getReadersFromUrls(*[m[0] for m in mibBorrowers], **dict(lowcaseMatching=False)))]
searchers = [PyFileSearcher(dstDirectory)]
for mibSearcher in mibSearchers:
searchers.append(PyPackageSearcher(mibSearcher))
searchers.append(StubSearcher(*mibStubs))
codeGenerator = PySnmpCodeGen()
fileWriter = PyFileWriter(dstDirectory).setOptions(pyCompile=pyCompileFlag,
pyOptimizationLevel=pyOptimizationLevel)
elif dstFormat == 'json':
inputMibs = ['BORROWED-MIB']
httpSources = [('mibs.snmplabs.com', 80, '/asn1/@mib@')]
httpBorrowers = [('mibs.snmplabs.com', 80, '/pysnmp/notexts/@mib@')]
dstDirectory = '.pysnmp-mibs'
# Initialize compiler infrastructure
mibCompiler = MibCompiler(SmiStarParser(), PySnmpCodeGen(),
PyFileWriter(dstDirectory))
# search for source MIBs at Web sites
mibCompiler.addSources(*[HttpReader(*x) for x in httpSources])
# never recompile MIBs with MACROs
mibCompiler.addSearchers(StubSearcher(*PySnmpCodeGen.baseMibs))
# check compiled/borrowed MIBs in our own productions
mibCompiler.addSearchers(PyFileSearcher(dstDirectory))
# search for compiled MIBs at Web sites if source is not available or broken
mibCompiler.addBorrowers(*[
PyFileBorrower(HttpReader(*x)).setOptions(genTexts=False)
for x in httpBorrowers
])
# run non-recursive MIB compilation
results = mibCompiler.compile(*inputMibs)
print('Results: %s' % ', '.join(['%s:%s' % (x, results[x]) for x in results]))