[1] | 1 | from Products.CMFCore.CMFCorePermissions import SetOwnPassword |
---|
| 2 | from Products.CMFCore.utils import getToolByName |
---|
| 3 | from Products.CMFDefault.MembershipTool import MembershipTool as BaseTool |
---|
| 4 | from Products.CMFPlone import ToolNames |
---|
| 5 | from Products.CMFPlone.PloneUtilities import translate |
---|
| 6 | from Products.CMFPlone.PloneUtilities import _createObjectByType |
---|
| 7 | from OFS.Image import Image |
---|
| 8 | from AccessControl import ClassSecurityInfo, getSecurityManager |
---|
| 9 | from Globals import InitializeClass |
---|
| 10 | from Acquisition import aq_base, aq_parent, aq_inner |
---|
| 11 | from Products.CMFCore.CMFCorePermissions import View |
---|
| 12 | from Products.CMFPlone.PloneBaseTool import PloneBaseTool |
---|
| 13 | |
---|
| 14 | from utils import scale_image |
---|
| 15 | from Globals import InitializeClass, DTMLFile |
---|
| 16 | from ZODB.POSException import ConflictError |
---|
| 17 | from Products.CMFCore.utils import _checkPermission |
---|
| 18 | |
---|
| 19 | from Products.CMFPlone.MembershipTool import MembershipTool |
---|
| 20 | |
---|
| 21 | def changeMemberPortrait(self, portrait, member_id=None): |
---|
| 22 | """ |
---|
| 23 | given a portrait we will modify the users portrait |
---|
| 24 | we put this method here because we do not want |
---|
| 25 | .personal or portrait in the catalog |
---|
| 26 | """ |
---|
| 27 | if not member_id: |
---|
| 28 | member_id = self.getAuthenticatedMember().getUserName() |
---|
| 29 | |
---|
| 30 | if portrait and portrait.filename: |
---|
| 31 | scaled, mimetype = scale_image(portrait) |
---|
| 32 | portrait = Image(id=member_id, file=scaled, title='') |
---|
| 33 | membertool = getToolByName(self, 'portal_memberdata') |
---|
| 34 | membertool._setPortrait(portrait, member_id) |
---|
| 35 | |
---|
| 36 | |
---|
| 37 | |
---|
| 38 | def getBadMembers(self): |
---|
| 39 | """Will search for members with bad images in the portal_memberdata |
---|
| 40 | delete their portraits and return their member ids""" |
---|
| 41 | memberdata = getToolByName(self, 'portal_memberdata') |
---|
| 42 | portraits = getattr(memberdata, 'portraits', None) |
---|
| 43 | if portraits is None: |
---|
| 44 | return [] |
---|
| 45 | bad_member_ids = [] |
---|
| 46 | TXN_THRESHOLD = 50 |
---|
| 47 | counter = 1 |
---|
| 48 | for member_id in tuple(portraits.objectIds()): |
---|
| 49 | portrait = portraits[member_id] |
---|
| 50 | portrait_data = str(portrait.data) |
---|
| 51 | if portrait_data == '': |
---|
| 52 | continue |
---|
| 53 | try: |
---|
| 54 | img = PIL.Image.open(StringIO(portrait_data)) |
---|
| 55 | except ConflictError: |
---|
| 56 | pass |
---|
| 57 | except: |
---|
| 58 | # Anything else we have a bad bad image and we destroy it |
---|
| 59 | # and ask questions later. |
---|
| 60 | portraits._delObject(member_id) |
---|
| 61 | bad_member_ids.append(member_id) |
---|
| 62 | if not counter%TXN_THRESHOLD: |
---|
| 63 | get_transaction().commit() |
---|
| 64 | counter = counter + 1 |
---|
| 65 | |
---|
| 66 | return bad_member_ids |
---|
| 67 | |
---|
| 68 | |
---|
| 69 | def deleteMemberArea(self, member_id): |
---|
| 70 | """ Delete member area of member specified by member_id. |
---|
| 71 | """ |
---|
| 72 | members = self.getMembersFolder() |
---|
| 73 | if not members: |
---|
| 74 | return 0 |
---|
| 75 | if hasattr( aq_base(members), member_id ): |
---|
| 76 | members.manage_delObjects(member_id) |
---|
| 77 | return 1 |
---|
| 78 | else: |
---|
| 79 | return 0 |
---|
| 80 | |
---|
| 81 | |
---|
| 82 | def deleteMembers(self, member_ids, delete_memberareas=1, |
---|
| 83 | delete_localroles=1): |
---|
| 84 | """ Delete members specified by member_ids. |
---|
| 85 | """ |
---|
| 86 | |
---|
| 87 | # Delete members in acl_users. |
---|
| 88 | acl_users = self.acl_users |
---|
| 89 | if _checkPermission('Manage Users', acl_users): |
---|
| 90 | if isinstance(member_ids, basestring): |
---|
| 91 | member_ids = (member_ids,) |
---|
| 92 | member_ids = list(member_ids) |
---|
| 93 | for member_id in member_ids[:]: |
---|
| 94 | if not acl_users.getUserById(member_id, None): |
---|
| 95 | member_ids.remove(member_id) |
---|
| 96 | try: |
---|
| 97 | acl_users.userFolderDelUsers(member_ids) |
---|
| 98 | except (NotImplementedError, 'NotImplemented'): |
---|
| 99 | raise NotImplementedError('The underlying User Folder ' |
---|
| 100 | 'doesn\'t support deleting members.') |
---|
| 101 | else: |
---|
| 102 | raise AccessControl_Unauthorized('You need the \'Manage users\' ' |
---|
| 103 | 'permission for the underlying User Folder.') |
---|
| 104 | |
---|
| 105 | # Delete member data in portal_memberdata. |
---|
| 106 | mdtool = getToolByName(self, 'portal_memberdata', None) |
---|
| 107 | if mdtool is not None: |
---|
| 108 | for member_id in member_ids: |
---|
| 109 | members = mdtool._members |
---|
| 110 | if members.has_key(member_id): |
---|
| 111 | del members[member_id] |
---|
| 112 | |
---|
| 113 | # Delete members' home folders including all content items. |
---|
| 114 | if delete_memberareas: |
---|
| 115 | for member_id in member_ids: |
---|
| 116 | self.deleteMemberArea(member_id) |
---|
| 117 | |
---|
| 118 | # Delete members' local roles. |
---|
| 119 | if delete_localroles: |
---|
| 120 | utool = getToolByName(self, 'portal_url', None) |
---|
| 121 | self.deleteLocalRoles( utool.getPortalObject(), member_ids, |
---|
| 122 | reindex=1 ) |
---|
| 123 | |
---|
| 124 | return tuple(member_ids) |
---|
| 125 | |
---|
| 126 | |
---|
| 127 | MembershipTool.manage_options = (MembershipTool.manage_options + |
---|
| 128 | ( { 'label' : 'Portraits' |
---|
| 129 | , 'action' : 'manage_portrait_fix' |
---|
| 130 | },)) |
---|
| 131 | MembershipTool.security = ClassSecurityInfo() |
---|
| 132 | MembershipTool.security.declareProtected('Manage Portal', 'manage_portrait_fix') |
---|
| 133 | MembershipTool.manage_portrait_fix = DTMLFile('www/portrait_fix', globals()) |
---|
| 134 | MembershipTool.security.declareProtected('Manage Portal', 'getBadMembers') |
---|
| 135 | MembershipTool.getBadMembers = getBadMembers |
---|
| 136 | MembershipTool.security.declareProtected('Manage Users', 'deleteMemberArea') |
---|
| 137 | MembershipTool.deleteMemberArea = deleteMemberArea |
---|
| 138 | MembershipTool.security.declareProtected('Manage Users', 'deleteMembers') |
---|
| 139 | MembershipTool.deleteMembers = deleteMembers |
---|
| 140 | |
---|
| 141 | InitializeClass(MembershipTool) |
---|