diff options
| author | Aurélien Bompard | 2015-03-16 11:22:04 +0100 |
|---|---|---|
| committer | Aurélien Bompard | 2015-03-16 11:22:04 +0100 |
| commit | a75be4230498933792e6ed02504ab4ab3e600f90 (patch) | |
| tree | d51697f2d0009d8d159bcebce868e7864753b6fc /src | |
| parent | 6280c5ffcd2fdebf80f170f7c9a4e47adf0c6c4a (diff) | |
| parent | 16ab6ac0cc2653f6123ba0781cd3270b4eedc776 (diff) | |
| download | mailman-a75be4230498933792e6ed02504ab4ab3e600f90.tar.gz mailman-a75be4230498933792e6ed02504ab4ab3e600f90.tar.zst mailman-a75be4230498933792e6ed02504ab4ab3e600f90.zip | |
Diffstat (limited to 'src')
| -rw-r--r-- | src/mailman/commands/cli_import.py | 12 | ||||
| -rw-r--r-- | src/mailman/utilities/importer.py | 21 | ||||
| -rw-r--r-- | src/mailman/utilities/tests/test_import.py | 35 |
3 files changed, 62 insertions, 6 deletions
diff --git a/src/mailman/commands/cli_import.py b/src/mailman/commands/cli_import.py index 30aeb7894..50d4f1864 100644 --- a/src/mailman/commands/cli_import.py +++ b/src/mailman/commands/cli_import.py @@ -35,6 +35,14 @@ from zope.interface import implementer +# Mock the Bouncer class from Mailman 2.1, we don't use it but there are +# instances in the pickled config files +class Bouncer: + class _BounceInfo: + pass + + + @implementer(ICLISubCommand) class Import21: """Import Mailman 2.1 list data.""" @@ -74,10 +82,11 @@ class Import21: assert len(args.pickle_file) == 1, ( 'Unexpected positional arguments: %s' % args.pickle_file) filename = args.pickle_file[0] + sys.modules["Mailman.Bouncer"] = Bouncer with open(filename, 'rb') as fp: while True: try: - config_dict = pickle.load(fp) + config_dict = pickle.load(fp, encoding="utf-8", errors="ignore") except EOFError: break except pickle.UnpicklingError: @@ -94,3 +103,4 @@ class Import21: except Import21Error as error: print(error, file=sys.stderr) sys.exit(1) + del sys.modules["Mailman.Bouncer"] diff --git a/src/mailman/utilities/importer.py b/src/mailman/utilities/importer.py index 8590d9b1b..f4373be79 100644 --- a/src/mailman/utilities/importer.py +++ b/src/mailman/utilities/importer.py @@ -32,6 +32,7 @@ from mailman.config import config from mailman.core.errors import MailmanError from mailman.handlers.decorate import decorate, decorate_template from mailman.interfaces.action import Action, FilterAction +from mailman.interfaces.address import IEmailValidator from mailman.interfaces.archiver import ArchivePolicy from mailman.interfaces.autorespond import ResponseAction from mailman.interfaces.bans import IBanManager @@ -387,11 +388,17 @@ def import_config_pck(mlist, config_dict): regulars_set = set(config_dict.get('members', {})) digesters_set = set(config_dict.get('digest_members', {})) members = regulars_set.union(digesters_set) - import_roster(mlist, config_dict, members, MemberRole.member) - import_roster(mlist, config_dict, config_dict.get('owner', []), - MemberRole.owner) - import_roster(mlist, config_dict, config_dict.get('moderator', []), - MemberRole.moderator) + # don't send welcome messages... + send_welcome_message = mlist.send_welcome_message + mlist.send_welcome_message = False + try: + import_roster(mlist, config_dict, members, MemberRole.member) + import_roster(mlist, config_dict, config_dict.get('owner', []), + MemberRole.owner) + import_roster(mlist, config_dict, config_dict.get('moderator', []), + MemberRole.moderator) + finally: + mlist.send_welcome_message = send_welcome_message @@ -427,8 +434,12 @@ def import_roster(mlist, config_dict, members, role): merged_members.update(config_dict.get('digest_members', {})) if merged_members.get(email, 0) != 0: original_email = bytes_to_str(merged_members[email]) + if not getUtility(IEmailValidator).is_valid(original_email): + original_email = email else: original_email = email + if not getUtility(IEmailValidator).is_valid(original_email): + continue # skip this one entirely address = usermanager.create_address(original_email) address.verified_on = datetime.datetime.now() user.link(address) diff --git a/src/mailman/utilities/tests/test_import.py b/src/mailman/utilities/tests/test_import.py index b0ab9938d..83ad06108 100644 --- a/src/mailman/utilities/tests/test_import.py +++ b/src/mailman/utilities/tests/test_import.py @@ -38,6 +38,7 @@ from mailman.app.lifecycle import create_list from mailman.config import config from mailman.handlers.decorate import decorate from mailman.interfaces.action import Action, FilterAction +from mailman.interfaces.address import InvalidEmailAddressError from mailman.interfaces.archiver import ArchivePolicy from mailman.interfaces.autorespond import ResponseAction from mailman.interfaces.bans import IBanManager @@ -747,6 +748,40 @@ class TestRosterImport(unittest.TestCase): anne = self._usermanager.get_user('anne@example.com') self.assertTrue(anne.controls('anne@example.com')) + def test_invalid_original_email(self): + self._pckdict["members"]["anne@example.com"] = b'invalid email address' + try: + import_config_pck(self._mlist, self._pckdict) + except InvalidEmailAddressError as e: + self.fail(e) + self.assertIn('anne@example.com', + [a.email for a in self._mlist.members.addresses]) + anne = self._usermanager.get_address('anne@example.com') + self.assertEqual(anne.original_email, 'anne@example.com') + + def test_invalid_email(self): + self._pckdict["members"] = { + 'anne@example.com': 0, + 'invalid email address': b'invalid email address' + } + self._pckdict["digest_members"] = {} + try: + import_config_pck(self._mlist, self._pckdict) + except InvalidEmailAddressError as e: + self.fail(e) + self.assertEqual(['anne@example.com'], + [a.email for a in self._mlist.members.addresses]) + + def test_no_email_sent(self): + self._pckdict + import_config_pck(self._mlist, self._pckdict) + self.assertIn("anne@example.com", + [a.email for a in self._mlist.members.addresses]) + # no email in any queue + for qname, sb in config.switchboards.items(): + self.assertEqual(len(sb.files), 0, + "Queue '{}' has {} emails".format(qname, len(sb.files))) + self.assertTrue(self._mlist.send_welcome_message) |
