File: //lib/python3.6/site-packages/glances/__pycache__/snmp.cpython-36.pyc
3
Eo�cQ � @ sb d dl Z d dlmZ yd dlmZ W n( ek
rL ejd� e jd� Y nX G dd� de�Z dS )� N)�logger)�cmdgenz;PySNMP library not found. To install it: pip install pysnmp� c sJ e Zd ZdZd� fdd � Zd
d � Zd d
� Zdd� Zdd� Zdd� Z � Z
S )�GlancesSNMPClientz,SNMP client class (based on pysnmp library).� localhost� �2c�public�private� c s@ t t| �j� tj� | _|| _|| _|| _|| _ || _
|| _ d S )N) �superr �__init__r ZCommandGenerator�cmdGen�version�host�port� community�user�auth)�selfr r r r r r )� __class__� �/usr/lib/python3.6/snmp.pyr
s
zGlancesSNMPClient.__init__c C sr i }xh|D ]`\}}t |�dkr,d||j� < q
|j� ||j� < ||j� jd�r
||j� dd� ||j� < q
W |S )zBuild the results.r zb'r � ���)�strZ prettyPrint�
startswith)r �varBinds�ret�name�valr r r Z
__buid_result( s z GlancesSNMPClient.__buid_resultc C s i }| s| r| j |�}|S )zPut results in table.)� _GlancesSNMPClient__buid_result)r �errorIndication� errorStatus�
errorIndexr r r r r �__get_result__5 s
z GlancesSNMPClient.__get_result__c G s� | j dkrB| jjtj| j| j�tj| j| j f�f|�� \}}}}n2| jjtj
| j �tj| j| j f�f|�� \}}}}| j ||||�S )z�SNMP simple request (list of OID).
One request per OID list.
:param oid: oid list
:return: a dict
�3)
r r �getCmdr � UsmUserDatar r �UdpTransportTargetr r �
CommunityDatar r% )r �oidr" r# r$ r r r r �
get_by_oid< s
2,z GlancesSNMPClient.get_by_oidc C s2 g }| s| r.x |D ]}|j | j|�� qW |S )N)�appendr! )r r" r# r$ � varBindTabler ZvarBindTableRowr r r �__bulk_result__N s
z!GlancesSNMPClient.__bulk_result__c G s� | j jd�rF| jjtj| j| j�tj| j | j
f�||f|�� \}}}}| j jd�r�| jj tj | j
�tj| j | j
f�||f|�� \}}}}ng S | j||||�S )a� SNMP getbulk request.
In contrast to snmpwalk, this information will typically be gathered in
a single transaction with the agent, rather than one transaction per
variable found.
* non_repeaters: This specifies the number of supplied variables that
should not be iterated over.
* max_repetitions: This specifies the maximum number of iterations over
the repeating variables.
* oid: oid list
> Return a list of dicts
r& �2)r r r r' r r( r r r) r r ZbulkCmdr* r r/ ) r Z
non_repeatersZmax_repetitionsr+ r" r# r$ r r. r r r �getbulk_by_oidU s
z GlancesSNMPClient.getbulk_by_oid)r r r r r
r ) �__name__�
__module__� __qualname__�__doc__r
r! r% r, r/ r1 �
__classcell__r r )r r r s
r )
�sysZglances.loggerr Z pysnmp.entity.rfc3413.onelinerr � ImportErrorZcritical�exit�objectr r r r r �<module>
s