sync_faculty.py

from ldap3 import Server, Connection, ALL
from peewee import *
from api.everything import *

cfg = load_config('api/config.yaml')
# skt = load_config('api/secret_config.yaml')

theStaticDB = SqliteDatabase(cfg['databases']['static'])
theStaticDB.drop_table(LDAPFaculty)

server = Server ('berea.edu', port=389, use_ssl=False, get_info='ALL')
# conn   = Connection (server, user=skt['ldap']['user'], password=skt['ldap']['pass'])
conn   = Connection (server, user="BPLP", password="Ol1v4r!Tun4bp")
if not conn.bind():
    print('error in bind', conn.result)

# search_base and search_filter are the parameters
conn.search('dc=berea,dc=edu', 
  '(description=Faculty)', 
  attributes = ['samaccountname', 'givenname', 'sn', 'employeeid']
  )

print ("Found {0} faculty.".format(len(conn.entries))) 

# print (conn.entries[0])

def safe (d, k):
  result = ""
  try:
    if k in d:
      print("Result: ", d[k])
      result = d[k]
  # If we can't find a key, skip it.
  except:
    print("Skipping: ", d, k)
    pass
  return result
    
# Recreate the table.
theStaticDB.create_table(LDAPFaculty)

faculty = conn.entries
for fac in faculty:
  try:
    if str(fac.samaccountname) != 'ptfaculty':
      print ("Faculty: {0}".format(fac.samaccountname))
      o, c = LDAPFaculty.get_or_create(
		lastname = safe (fac, 'sn'),
     		firstname = safe (fac, 'givenname'),
    		username  = fac.samaccountname,
    		bnumber   = safe (fac, 'employeeid')
    		)
  except:
    print("Error")
  #o = LDAPFaculty (
  #  lastname  = safe (fac, 'sn'),
  #  firstname = safe (fac, 'givenname'),
  #  username  = fac.samaccountname,
  #  bnumber   = safe (fac, 'employeeid'),
  #  )
  #o.save()
  
  
# Now do Staff
conn.search('dc=berea,dc=edu', 
  '(description=Staff)', 
  attributes = ['samaccountname', 'givenname', 'sn', 'employeeid']
  )

print ("Found {0} staff.".format(len(conn.entries))) 


faculty = conn.entries        #actually staff, but who cares?
for fac in faculty:
  try: 
    if str(fac.samaccountname) != 'ptfaculty':
      print ("Staff: {0}".format(fac.samaccountname))
      o, c = LDAPFaculty.get_or_create(
		lastname = safe (fac, 'sn'),
     		firstname = safe (fac, 'givenname'),
    		username  = fac.samaccountname,
    		bnumber   = safe (fac, 'employeeid')
    		) 
  except: 
    print("Error")
#o = LDAPFaculty (
#    lastname  = safe (fac, 'sn'),
#    firstname = safe (fac, 'givenname'),
#    username  = fac.samaccountname,
#    bnumber   = safe (fac, 'employeeid'),
#    )
#  o.save()