custom_dhcpd.py

custom_dhcpd.py

from datadog_checks.checks import AgentCheck
from isc_dhcp_leases import IscDhcpLeases

__version__ = "0.0.1"

class DhcpCheck(AgentCheck):
  def check(self, instance):
    name = instance['name']
    lease_file_path = '/var/lib/dhcp/dhcpd.leases'
    location = None

    if instance.has_key('location'):
      location = instance['location']

    if instance.has_key('lease_file_path'):
      location = instance['lease_file_path']

    leases = IscDhcpLeases(lease_file_path)
    lease_count = len(leases.get_current())

    self.gauge('custom_dhcpd.lease_count', lease_count, tags = ["name:%s" % name, "location:%s" % location])

custom_dhcpd.yaml

init_config:

instances:
  - name: scala-matsuri