from Products.CMFCore.CMFCorePermissions import SetOwnPassword
from Products.CMFCore.utils import getToolByName
from Products.CMFDefault.MembershipTool import MembershipTool as BaseTool
from Products.CMFPlone import ToolNames
from AccessControl import ClassSecurityInfo, getSecurityManager
from Globals import InitializeClass
from Acquisition import aq_base, aq_parent, aq_inner
from Products.CMFCore.CMFCorePermissions import View
from Products.CMFPlone.PloneBaseTool import PloneBaseTool
from ZODB.POSException import ConflictError
from Products.CMFCore.utils import _checkPermission

from Products.CMFPlone.MembershipTool import MembershipTool
    


def deleteMemberArea(self, member_id):
    """ Delete member area of member specified by member_id.
    """
    members = self.getMembersFolder()
    if not members:
        return 0
    if hasattr( aq_base(members), member_id ):
        members.manage_delObjects(member_id)
        return 1
    else:
        return 0


def deleteMembers(self, member_ids, delete_memberareas=1,
                    delete_localroles=1):
    """ Delete members specified by member_ids.
    """

    # Delete members in acl_users.
    acl_users = self.acl_users
    if _checkPermission('Manage Users', acl_users):
        if isinstance(member_ids, basestring):
            member_ids = (member_ids,)
        member_ids = list(member_ids)
        for member_id in member_ids[:]:
            if not acl_users.getUserById(member_id, None):
                member_ids.remove(member_id)
        try:
            acl_users.userFolderDelUsers(member_ids)
        except (NotImplementedError, 'NotImplemented'):
            raise NotImplementedError('The underlying User Folder '
                                        'doesn\'t support deleting members.')
    else:
        raise AccessControl_Unauthorized('You need the \'Manage users\' '
                                'permission for the underlying User Folder.')

    # Delete member data in portal_memberdata.
    mdtool = getToolByName(self, 'portal_memberdata', None)
    if mdtool is not None:
        for member_id in member_ids:
            members = mdtool._members
            if members.has_key(member_id):
                del members[member_id]

    # Delete members' home folders including all content items.
    if delete_memberareas:
        for member_id in member_ids:
                self.deleteMemberArea(member_id)

    # Delete members' local roles.
    if delete_localroles:
        utool = getToolByName(self, 'portal_url', None)
        self.deleteLocalRoles( utool.getPortalObject(), member_ids,
                                reindex=1 )

    return tuple(member_ids)

MembershipTool.security = ClassSecurityInfo()
MembershipTool.security.declareProtected('Manage Users', 'deleteMemberArea')
MembershipTool.deleteMemberArea = deleteMemberArea
MembershipTool.security.declareProtected('Manage Users', 'deleteMembers')
MembershipTool.deleteMembers = deleteMembers

InitializeClass(MembershipTool)