Changeset 1031 for pykota/trunk/pykota/storages/ldapstorage.py

Timestamp:

06/16/03 13:59:09 (21 years ago)

Author:

jalet

Message:

More work on LDAP

Files:

• pykota/trunk/pykota/storages/ldapstorage.py (modified) (12 diffs)

pykota/trunk/pykota/storages/ldapstorage.py

@@ -21 +21 @@
 #
 # $Log$
+# Revision 1.9  2003/06/16 11:59:09  jalet
+# More work on LDAP
+#
 # Revision 1.8  2003/06/15 22:26:52  jalet
 # More work on LDAP
@@ -86 +89 @@
         except ldap.SERVER_DOWN :    
             raise PyKotaStorageError, "LDAP backend for PyKota seems to be down !" # TODO : translate
+        except ldap.LDAPError :    
+            raise PyKotaStorageError, "Unable to connect to LDAP server %s as %s." % (host, user) # TODO : translate
         else :    
             self.closed = 0
@@ -116 +121 @@
             raise PyKotaStorageError, _("Search for %s(%s) from %s(scope=%s) returned no answer.") % (key, fields, base, scope)
         else :     
+            if self.debug :
+                self.tool.logger.log_message("QUERY : Result : %s" % result, "debug")
             return result
             
@@ -155 +162 @@
     def getPrinterName(self, printerid) :        
         """Returns a printerid given a printer id."""
-        result = self.doSearch("objectClass=*", ["pykotaPrinterName"], base=printerid, scope=ldap.SCOPE_BASE)
+        result = self.doSearch("objectClass=pykotaPrinter", ["pykotaPrinterName"], base=printerid, scope=ldap.SCOPE_BASE)
         if result :
             return result[0][1]["pykotaPrinterName"][0]
@@ -167 +174 @@
     def setPrinterPrices(self, printerid, perpage, perjob) :
         """Sets prices per job and per page for a given printer."""
-        raise PyKotaStorageError, "Not implemented !"
+        fields = { 
+                   "pykotaPricePerPage" : str(perpage),
+                   "pykotaPricePerJob" : str(perjob),
+                 } 
+        return self.doModify(printerid, fields)
     
     def getUserId(self, username) :
@@ -174 +185 @@
         if result :
             return result[0][0]
+            
+    def getUserName(self, userid) :
+        """Returns a username given a userid."""
+        result = self.doSearch("objectClass=pykotaAccount", ["pykotaUserName"], base=userid, scope=ldap.SCOPE_BASE)
+        if result :
+            return result[0][1]["pykotaUserName"][0]
             
     def getGroupId(self, groupname) :
@@ -220 +237 @@
     def getUserGroupsNames(self, userid) :        
         """Returns the list of groups' names the user is a member of."""
-        raise PyKotaStorageError, "Not implemented !"
+        username = self.getUserName(userid)
+        if username :
+            result = self.doSearch("(&(objectClass=pykotaGroup)(%s=%s))" % (self.info["groupmembers"], username), [self.info["grouprdn"]], base=self.info["groupbase"])
+            if result :
+                return [v[self.info["grouprdn"]][0] for (k, v) in result]
         
     def addPrinter(self, printername) :        
@@ -269 +290 @@
                  } 
         dn = "cn=%s,%s" % (uuid, self.info["userquotabase"])
-        return self.doAdd(dn, fields)
+        self.doAdd(dn, fields)
+        return (dn, printerid)
         
     def addGroupPQuota(self, groupname, printerid) :
@@ -283 +305 @@
                  } 
         dn = "cn=%s,%s" % (uuid, self.info["groupquotabase"])
-        return self.doAdd(dn, fields)
-        
-    def increaseUserBalance(self, userid, amount) :    
+        self.doAdd(dn, fields)
+        return (dn, printerid)
+        
+    def increaseUserBalance(self, userquotaid, amount) :    
         """Increases (or decreases) an user's account balance by a given amount."""
-        raise PyKotaStorageError, "Not implemented !"
+        balance = self.getUserBalance(userquotaid)
+        if balance :
+            (newbal, newpaid) = [(float(v) + float(amount)) for v in balance]
+            result = self.doSearch("objectClass=pykotaUserPQuota", ["pykotaUserName"], base=userquotaid, scope=ldap.SCOPE_BASE)
+            if result :
+                username = result[0][1]["pykotaUserName"][0]
+                if username :
+                    result = self.doSearch("(&(objectClass=pykotaAccountBalance)(pykotaUserName=%s))" % username , [ "pykotaBalance", "pykotaLifeTimePaid"])
+                    fields = {
+                               "pykotaBalance" : str(newbal),
+                               "pykotaLifeTimePaid" : str(newpaid),
+                             }
+                    return self.doModify(result[0][0], fields)         
         
     def getUserBalance(self, userquotaid) :    
@@ -349 +384 @@
                 return result[0][1]["pykotaLimitBy"][0]
         
-    def setUserBalance(self, userid, balance) :    
+    def setUserBalance(self, userquotaid, balance) :    
         """Sets the account balance for a given user to a fixed value."""
-        raise PyKotaStorageError, "Not implemented !"
-        
-    def limitUserBy(self, userid, limitby) :    
+        oldbalance = self.getUserBalance(userquotaid)
+        if oldbalance :
+            (oldbal, oldpaid) = oldbalance
+            difference = balance - oldbal
+            return self.increaseUserBalance(userquotaid, difference)
+        
+    def limitUserBy(self, userquotaid, limitby) :    
         """Limits a given user based either on print quota or on account balance."""
-        raise PyKotaStorageError, "Not implemented !"
-        
-    def limitGroupBy(self, groupid, limitby) :    
+        result = self.doSearch("objectClass=pykotaUserPQuota", ["pykotaUserName"], base=userquotaid, scope=ldap.SCOPE_BASE)
+        if result :
+            username = result[0][1]["pykotaUserName"][0]
+            fields = {
+                       "pykotaLimitBy" : limitby,
+                     }
+            self.doModify(self.getUserId(username), fields)
+        
+    def limitGroupBy(self, groupquotaid, limitby) :    
         """Limits a given group based either on print quota or on sum of its users' account balances."""
-        raise PyKotaStorageError, "Not implemented !"
+        result = self.doSearch("objectClass=pykotaGroupPQuota", ["pykotaGroupName"], base=groupquotaid, scope=ldap.SCOPE_BASE)
+        if result :
+            groupname = result[0][1]["pykotaGroupName"][0]
+            fields = {
+                       "pykotaLimitBy" : limitby,
+                     }
+            self.doModify(self.getGroupId(groupname), fields)
         
     def setUserPQuota(self, userquotaid, printerid, softlimit, hardlimit) :
@@ -377 +428 @@
         return self.doModify(groupquotaid, fields)
         
-    def resetUserPQuota(self, userid, printerid) :    
+    def resetUserPQuota(self, userquotaid, printerid) :    
         """Resets the page counter to zero for a user on a printer. Life time page counter is kept unchanged."""
-        raise PyKotaStorageError, "Not implemented !"
-        
-    def resetGroupPQuota(self, groupid, printerid) :    
+        fields = {
+                   "pykotaPageCounter" : "0",
+                   "pykotaDateLimit" : "None",
+                 }
+        return self.doModify(userquotaid, fields)      
+        
+    def resetGroupPQuota(self, groupquotaid, printerid) :    
         """Resets the page counter to zero for a group on a printer. Life time page counter is kept unchanged."""
-        raise PyKotaStorageError, "Not implemented !"
-        
-    def updateUserPQuota(self, userid, printerid, pagecount) :
+        fields = {
+                   "pykotaPageCounter" : "0",
+                   "pykotaDateLimit" : "None",
+                 }
+        return self.doModify(groupquotaid, fields)      
+        
+    def updateUserPQuota(self, userquotaid, printerid, pagecount) :
         """Updates the used user Quota information given (userid, printerid) and a job size in pages."""
-        raise PyKotaStorageError, "Not implemented !"
+        jobprice = self.computePrinterJobPrice(printerid, pagecount)
+        result = self.doSearch("objectClass=pykotaUserPQuota", ["pykotaPageCounter", "pykotaLifePageCounter"], base=userquotaid, scope=ldap.SCOPE_BASE)
+        if result :
+            oldfields = result[0][1]
+            fields = {
+                       "pykotaPageCounter" : str(pagecount + int(oldfields["pykotaPageCounter"][0])),
+                       "pykotaLifePageCounter" : str(pagecount + int(oldfields["pykotaLifePageCounter"][0])),
+                     }  
+            return self.doModify(userquotaid, fields)         
         
     def getUserPQuota(self, userquotaid, printerid) :
@@ -432 +499 @@
             return quota
         
-    def setUserDateLimit(self, userid, printerid, datelimit) :
+    def setUserDateLimit(self, userquotaid, printerid, datelimit) :
         """Sets the limit date for a soft limit to become an hard one given (userid, printerid)."""
-        raise PyKotaStorageError, "Not implemented !"
-        
-    def setGroupDateLimit(self, groupid, printerid, datelimit) :
+        fields = {
+                   "pykotaDateLimit" : "%04i-%02i-%02i %02i:%02i:%02i" % (datelimit.year, datelimit.month, datelimit.day, datelimit.hour, datelimit.minute, datelimit.second),
+                 }
+        return self.doModify(userquotaid, fields)
+        
+    def setGroupDateLimit(self, groupquotaid, printerid, datelimit) :
         """Sets the limit date for a soft limit to become an hard one given (groupid, printerid)."""
-        raise PyKotaStorageError, "Not implemented !"
+        fields = {
+                   "pykotaDateLimit" : "%04i-%02i-%02i %02i:%02i:%02i" % (datelimit.year, datelimit.month, datelimit.day, datelimit.hour, datelimit.minute, datelimit.second),
+                 }
+        return self.doModify(groupquotaid, fields)
         
     def addJobToHistory(self, jobid, userid, printerid, pagecounter, action) :