#!/usr/bin/env python
# -*- mode: python -*-
# This script imports arbitary lists of data. The input is a file with 
# the form of:
#  uid: <data>

import string, re, time, ldap, getopt, sys;
from userdir_ldap import *;

# Process options
(options, arguments) = getopt.getopt(sys.argv[1:], "u:m:n")
for (switch, val) in options:
   if (switch == '-u'):
      AdminUser = val
   elif (switch == '-m'):
       LoadOverride(val);
   elif (switch == '-n'):
       NoAct = 1;
if len(arguments) == 0:
   print "Give the key to assignt to then the file to import";
   os.exit(0);

# Main program starts here
l = passwdAccessLDAP(LDAPServer, BaseDn, AdminUser)

List = open(arguments[1],"r");
Set = [];
User = None;
while(1):
   Line = List.readline();
   if Line != "":
      # Glob similar lines
      Split = re.split("[:\n]",Line);
      if User == None:
         User = Split[0];
      if Split[0] == User:
         Set.append(string.strip(Split[1]));
         continue;
   else:
      if len(Set) == 0:
         break;
   
   # Generate the command..
   Rec = [(ldap.MOD_REPLACE,arguments[0],Set[0])];
   for x in Set[1:]:
      Rec.append((ldap.MOD_ADD,arguments[0],x))

   Dn = "uid=" + User + "," + BaseDn;
   try:
      print Dn,Rec;
      l.modify_s(Dn,Rec);
   except:
      print "Failed",Dn;
   
   # Out of data..
   if Line == "":
      break;   
   User = Split[0];
   Set = [string.strip(Split[1])];