There are a ton of examples out there on how to utilize ldap in python. I recently worked on a simple project which required active directory authentication using ldap. The following is simple script I've used hundreds of times.
#!/usr/bin/env python
#binds to ldap, queries for a specific AD account
import ldap
def Search(server,port, auth_user, auth_pass, ldap_user,attrs):
#base_dn should reflect your domain
base_dn="dc=yourcompanydomain,dc=com"
found_results=""
l = ldap.initialize('ldap://%s:%s' % (server, port) )
l.simple_bind_s(auth_user, auth_pass)
try:
search_result= l.search(base_dn,ldap.SCOPE_SUBTREE,'sAMAccountName='+ldap_user,attrs)
result_set =[]
while 1:
result_type, result_data = l.result(search_result,0)
if (result_data ==[]):
break
else:
if result_type == ldap.RES_SEARCH_ENTRY:
result_set.append(result_data)
print len(result_set)
for line in result_set:
print line
except ldap.LDAPError, e:
print e
results = Search("192.168.5.10",3268,"akonkol@yourcompanydomain.com","mysecretpassword","akonkol",['givenName','sn','mail'])
This will return the attributes you supplied (givenName, sn, mail)
A few months ago I decided to write a web application that would essentially run like RANCID, I named it "tratto." Since then I decided that it would be better to nail down a basic python framework first, then integrate it into say.. a django application. In all of my years as an engineer I had never leveraged expect to accomplish simple and repetitive tasks. I am a recent python "convert" and wanted to write a simple app that could be used as a framework for managing and monitoring network connected devices and hosts. Tratto uses pexpectto connect and parse ssh and telnet sessions. This framework provides an easy way to connect to remote devices and issue commands and store the output. I also wanted an easy way to "extend" this framework and be able to add ways to connect to any operating system (or at least use default shell behavior as a baseline). I manage a wide variety of devices and I wanted to support at least the default implementations of Cisco IOS, OpenBSD, Mac OS X, and Aruba OS. Here is how you would add an operating system's parameters to Tratto (Systems.py):
class ArubaOS(OperatingSystem):
'''aruba configs'''
PROMPTLINE ='#'
PAGINATES =True
DISABLE_PAGINATION = 'terminal length 0'
GET_CONFIG ="show run"
There are 3 files included in Tratto right now:
- Connectivity.py -- This is a class which manages sessions using pexpect
- Systems.py -- This is the class which manages all the operating parameters
- driver.py -- This is an example file of how to use Tratto to fetch whatever info you want
Here is an example of how to use the framework to connect to devices and issue commands:
#!/usr/bin/env python
import Connectivity
import Systems
#telnet to a cisco switch
m = Systems.OperatingSystems['IOS']
s = Connectivity.Session("192.168.6.1",23,telnet,m)
s.login("akonkol", "mypass")
s.sendcommand("show ver")
s.sendcommand("show clock")
s.sendcommand("show run")
s.logout()
#ssh to a apple machine
m = Systems.OperatingSystems['OSX']
s = Connectivity.Session("127.0.0.1",22,"ssh",m)
s.login("akonkol", "mypass")
#sendcommand will echo response by default, you can store that
#response in a variable if you wish
result = s.sendcommand("df -h")
print result
s.getversion()
s.logout()
#ssh to openbsd box
m = Systems.OperatingSystems['OBSD']
s = Connectivity.Session("192.168.5.1",22,"ssh",m)
s.login("akonkol", "mypass")
print s.sendcommand("cat /etc/passwd")
print s.sendcommand("arp -a")
s.logout()
The Future
With Tratto you can technically pull information from any networked device and use that data for whatever you please. Current ideas are integrating Tratto into
- a config repository application with a web frontend (like RANCID)
- a network mapping application using cdp neighbors
- monitoring platform which performs different commands based on certain scenarios ("show interfaces" when IP SLA shows latency)
This is my first attempt at releasing python software, so if you think something could be better let me know. Download Tratto
A lot of talk about the lottery today at lunch.. i've never bought a ticket.
#!/usr/bin/python
import random
for i in range(1,6):
print random.randrange(1,56)
print random.randrange(1,46)
Graphs are a pain in the ass. I first attempted to get PYOFC2 to work about a year ago and gave up, I just didn't care enough. Now I want graphs, and have figured out how to use it (at least for a pie chart). urls.py
from django.conf.urls.defaults import *
from django.conf import settings
# the chart data views
urlpatterns = patterns('yourapp.views',
('^data/$','chart_data_two'),
)
urlpatterns += patterns('django.views.generic.simple',
(r'^$', 'direct_to_template', {'template': 'index.html'}),
)
templates/index.html
Chart!
<h1>Pie Chart!</h1>
<div id="chart_demo" class="chart"> </div>
ls -l static/*
-rwxr-xr-x 1 akonkol akonkol 773 2011-03-02 16:08 expressInstall.swf
-rw-r--r-- 1 akonkol akonkol 263109 2011-03-02 16:08 open-flash-chart.swf
-rw-r--r-- 1 akonkol akonkol 9759 2011-03-02 16:08 swfobject.js
yourapp/views.py
from django.http import HttpResponse
from pyofc2 import *
import random
import time
def chart_data_two(request):
t = title(text='Crapface Graph')
p1 = pie()
values = [ pie_value(label="crap", value=4), pie_value(label="face", value=20)]
p1.values = values
chart = open_flash_chart()
chart.title = t
chart.add_element(p1)
return HttpResponse(chart.render())
Produces: 
References demodjofc2_demo
About a month ago I wrote what I call hatch, which helps you create templates and generate configurations on the fly. Originally the idea was to allow people to store configuration files, but I nixed that idea and it reduced my amount of code by about 50%. Instead I added a few convenient features such as a copy-to-clipboard button, comments, markdown syntax, tagging, and user profiles with avatar support. On top of all the new features, hatch went through a re-design: 
To make hatch better, I've started eating my own dog food and so far so good.
I'm working on a new project and decided to use slugs to access object information. The built-in in "slugify" does not generate unique slugs for objects, but I found a solution. Overriding the save method allows us to check to see if there are any other objects which have the same slug such as "how-to-make-a-table" and if there are we append a number to it such as "how-to-make-a-table-2." Below is a real-world example of a model "Project" which has pretty typical fields in it and the code to generate a unique slug. file: /myproject/myapp/models.py
from django.contrib.auth.models import User
from sorl.thumbnail import ImageField
import datetime
import re
class Project(models.Model):
title = models.CharField(max_length=255,blank=True)
description = models.TextField(blank=True)
photo = ImageField(upload_to=get_project_upload_path, blank=True, null=True)
author = models.ForeignKey(User, editable=False,blank=True, null=True)
pub_date = models.DateTimeField(editable=False, blank=True)
slug = models.SlugField(unique=True,blank=True)
def save(self, *args, **kwargs):
#set pub_date as right now
self.pub_date=datetime.datetime.now()
#As long as this object does NOT have a slug
if not self.slug:
from django.template.defaultfilters import slugify
#Take the title and replace spaces with hypens, make lowercase
potential_slug = slugify(self.title)
self.slug = potential_slug
while True:
try:
#try to save the object
super(Project, self).save(*args, **kwargs)
#if this slug already exists we get an error
except IntegrityError:
#match the slug or look for a trailing number
match_obj = re.match(r'^(.*)-(\d+)$', self.slug)
#if we find a match
if match_obj:
#take the found number and increment it by 1
next_int = int(match_obj.group(2)) + 1
self.slug = match_obj.group(1) + "-" + str(next_int)
else:
#There are no matches for -# so create one with -2
self.slug += '-2'
#different error than IntegrityError
else:
break
def __unicode__(self):
return self.title
apt-get install gcc apt-get install libperl-dev apt-get install python2.7-dev Download the netsnmp source: wget http://downloads.sourceforge.net/project/net-snmp/net-snmp/5.7.1/net-snmp-5.7.1.tar.gz?r=http%3A%2F%2Fsourceforge.net%2Fprojects%2Fnet-snmp%2Ffiles%2Fnet-snmp%2F5.7.1%2F&ts=1318446955&use_mirror=surfnet tar xzvf net-snmp-5.7.1.tar.gz cd net-snmp-5.7.1 ./configure --with-python-modules Answer the questions make make install ldconfig akonkol@cmg005:/home/akonkol/net-snmp-5.7.1# python Python 2.7.1+ (r271:86832, Apr 11 2011, 18:13:53) [GCC 4.5.2] on linux2 Type "help", "copyright", "credits" or "license" for more information. >>> import netsnmp >>>
I've written a ton of snmp monitoring scripts and they all suck because they are blocking and take "too long" to return results for a large amount of hosts. So how would we make this process faster and make us happier?
multiprocessing is a package that supports spawning processes using an API similar to the threading module. The multiprocessing package offers both local and remote concurrency, effectively side-stepping the Global Interpreter Lock by using subprocesses instead of threads. Due to this, the multiprocessing module allows the programmer to fully leverage multiple processors on a given machine. It runs on both Unix and Windows.
IBM has published a wonderful article on Multiprocessing with Python in which I have modified their snippets for my purposes below. To leverage NetSNMP in python you need to compile it with python modules, I wrote a post about this yesterday in Compiling Netsnmp for Ubuntu. After battling through the compiling bit you should be able to get the example in the IBM post to work. I modified this example *slightly* to accommodate instance identifier (iid). Hopefully this will be suitable for a new project I'm working on which polls thousands of snmp oids to measure latency between routers using the concept I posted on hatch - Configure RTR/SLA Ping in IOS template.
!/usr/bin/env python
"""
This is a multiprocessing wrapper for Net-SNMP.
This makes a synchronous API asynchronous by combining
it with Python2.6
"""
import netsnmp
from multiprocessing import Process, Queue, current_process
class HostRecord():
"""This creates a host record"""
def __init__(self,
hostname = None,
query = None):
self.hostname = hostname
self.query = query
class SnmpSession():
"""A SNMP Session"""
def __init__(self,
oid = "sysDescr",
iid="0",
Version = 2,
DestHost = "localhost",
Community = "public",
Verbose = True,
):
self.oid = oid
self.Version = Version
self.DestHost = DestHost
self.Community = Community
self.Verbose = Verbose
self.var = netsnmp.Varbind(oid, iid)
self.hostrec = HostRecord()
self.hostrec.hostname = self.DestHost
def query(self):
"""Creates SNMP query
Fills out a Host Object and returns result
"""
try:
result = netsnmp.snmpget(self.var,
Version = self.Version,
DestHost = self.DestHost,
Community = self.Community)
self.hostrec.query = result
except Exception, err:
if self.Verbose:
print err
self.hostrec.query = None
finally:
return self.hostrec
def make_query(host):
"""This does the actual snmp query
This is a bit fancy as it accepts both instances
of SnmpSession and host/ip addresses. This
allows a user to customize mass queries with
subsets of different hostnames and community strings
"""
if isinstance(host,SnmpSession):
return host.query()
else:
s = SnmpSession(DestHost=host)
return s.query()
# Function run by worker processes
def worker(input, output):
for func in iter(input.get, 'STOP'):
result = make_query(func)
output.put(result)
def main():
"""Runs everything"""
#clients
hosts = [
SnmpSession(DestHost="10.71.1.1", Community="my-pub-community", oid="1.3.6.1.4.1.9.9.42.1.2.10.1.1", iid="1"),
SnmpSession(DestHost="10.81.1.1", Community="my-pub-community", oid="1.3.6.1.4.1.9.9.42.1.2.10.1.1", iid="123")
]
NUMBER_OF_PROCESSES = len(hosts)
# Create queues
task_queue = Queue()
done_queue = Queue()
#submit tasks
for host in hosts:
task_queue.put(host)
#Start worker processes
for i in range(NUMBER_OF_PROCESSES):
Process(target=worker, args=(task_queue, done_queue)).start()
# Get and print results
print 'Unordered results:'
for i in range(len(hosts)):
print '\t', done_queue.get().query
# Tell child processes to stop
for i in range(NUMBER_OF_PROCESSES):
task_queue.put('STOP')
#print "Stopping Process #%s" % i
if __name__ == "__main__":
main()
Two years ago I wrote a script that you could use to send commands to telnet/ssh enabled devices called tratto. Since then I have changed jobs and with new jobs comes new enviornments. I can no longer send commands like "show run" without sending the enable command (enable is like the 'su' of unix). I re-visited my code... which is always an entertaining. Anyway, I added an "escalateprivileges" command and added a string to the Systems object so you can store what the escalation command is for different operating systems.
You can download it here or via github
Connectivity.py
def escalateprivileges(self, escalated_password=None):
escalated_password = escalated_password
if self.connected:
self.connection.sendline(self.operatingsystem.ESCALATE_COMMAND)
i = self.connection.expect(r"(?i)password[\s:]+")
if i==0:
self.connection.sendline(escalated_password)
i = self.connection.expect(self.operatingsystem.PROMPTLINE)
if i==0:
if("denied" in self.connection.before):
print "***Escalation FAILED***"
print self.connection.before
else:
print "***Escalation Successful***"
else:
raise SessionError("***Not Connected***")
Systems.py
class OperatingSystem(object):
ESCALATE_COMMAND=''
PAGINATES =False
VERSION =''
PROMPTLINE =''
..
class CiscoIOS(OperatingSystem):
'''cisco ios'''
PROMPTLINE = r'[-\w]+[>#]'
GET_CONFIG ='show running-config'
PAGINATES =True
VERSION ='show version'
DISABLE_PAGINATION = 'terminal length 0'
ESCALATE_COMMAND='enable'
...
Example usage:
#!/usr/bin/env python
import Connectivity
import Systems
#telnet to a cisco switch
m = Systems.OperatingSystems['IOS']
s = Connectivity.Session("10.10.1.1",23,"telnet",m)
s.login("akonkol", "mypassword")
s.escalateprivileges('myenablepassword')
#s.sendcommand("show ver")
s.sendcommand("show clock")
s.sendcommand("show run")
s.sendcommand("show start")
s.logout()
I needed to handle errors better so I looked into creating my own exceptions. It easy, and this is a dead simple example.
#!/usr/bin/python
class AdditionError(Exception):
pass
x = 2
y = 'i'
try:
print x + y
except:
raise AdditionError("Couldn't add x + y ")
root@echo:/home/akonkol/Code/exception# ./addition.py
Traceback (most recent call last):
File "./addition.py", line 14, in
raise AdditionError("Couldn't add x + y ")
__main__.AdditionError: Couldn't add x + y
I recenly wrote a django app which needed to store regex strings in a model. Since I couldn't find a way to store raw strings in models I wrote a function to convert unicode or string into string literals.
def to_raw_string(s):
if isinstance(s,str):
s = s.encode('string-escape')
if isinstance(s,unicode):
s = s.encode('unicode-escape')
return s