diff rhodecode/model/user.py @ 2330:b0fef8a77568 codereview

Added simple UI for admin to manage emails map
author Marcin Kuzminski <marcin@python-works.com>
date Wed, 23 May 2012 00:58:37 +0200
parents 24095abde696
children d3ac7491a5c8
line wrap: on
line diff
--- a/rhodecode/model/user.py	Wed May 23 00:11:45 2012 +0200
+++ b/rhodecode/model/user.py	Wed May 23 00:58:37 2012 +0200
@@ -36,7 +36,7 @@
 from rhodecode.model.db import User, UserRepoToPerm, Repository, Permission, \
     UserToPerm, UsersGroupRepoToPerm, UsersGroupToPerm, UsersGroupMember, \
     Notification, RepoGroup, UserRepoGroupToPerm, UsersGroup,\
-    UsersGroupRepoGroupToPerm
+    UsersGroupRepoGroupToPerm, UserEmailMap
 from rhodecode.lib.exceptions import DefaultUserException, \
     UserOwnsReposException
 
@@ -587,3 +587,29 @@
                 .scalar()
         if obj:
             self.sa.delete(obj)
+
+    def add_extra_email(self, user, email):
+        """
+        Adds email address to UserEmailMap
+
+        :param user:
+        :param email:
+        """
+        user = self.__get_user(user)
+        obj = UserEmailMap()
+        obj.user = user
+        obj.email = email
+        self.sa.add(obj)
+        return obj
+
+    def delete_extra_email(self, user, email_id):
+        """
+        Removes email address from UserEmailMap
+
+        :param user:
+        :param email_id:
+        """
+        user = self.__get_user(user)
+        obj = UserEmailMap.query().get(email_id)
+        if obj:
+            self.sa.delete(obj)
\ No newline at end of file