Add Hokie Stalker module

modules/hs.py

@@ -0,0 +1,52 @@
+#!/usr/bin/python2
+"""
+hs.py - hokie stalker module
+author: mutantmonkey <mutantmonkey@gmail.com>
+"""
+
+import ldap
+from urllib import quote as urlquote
+
+LDAP_URI = "ldap://directory.vt.edu"
+
+l = ldap.initialize(LDAP_URI)
+
+"""Search LDAP using the argument as a query. Argument must be a valid LDAP query."""
+def search(query):
+	result = l.search_s('ou=People,dc=vt,dc=edu', ldap.SCOPE_SUBTREE, query)
+	if len(result) <= 0:
+		return False
+
+	print "wtf"
+
+	return result
+
+def hs(phenny, input):
+	""".hs <pid/name/email> - Search for someone on Virginia Tech People Search."""
+
+	q = input.group(2)
+
+	# initially try search by PID
+	s = search('uupid=%s' % q)
+
+	# try partial search on CN if no results for PID
+	if not s:
+		s = search('cn=*%s*' % '*'.join(q.split(' ')))
+
+	# try email address if no results found for PID or CN
+	if not s:
+		s = search('mail=%s*' % q)
+
+	if s:
+		if len(s) >1:
+			phenny.reply("Multiple results found; try http://search.vt.edu/search/people.html?q=%s" % urlquote(q))
+		else:
+			for dh, entry in s:
+				phenny.reply("%s - http://search.vt.edu/search/person.html?person=%d" % (entry['cn'][0], int(entry['uid'][0])))
+	else:
+		phenny.reply("No results found")
+hs.rule = (['hs'], r'(.*)')
+
+if __name__ == '__main__':
+	print __doc__.strip()
+