1 # The software in this package is distributed under the GNU General
2 # Public License version 2 (with a special exception described below).
4 # A copy of GNU General Public License (GPL) is included in this distribution,
5 # in the file COPYING.GPL.
7 # As a special exception, if other files instantiate templates or use macros
8 # or inline functions from this file, or you compile this file and link it
9 # with other works to produce a work based on this file, this file
10 # does not by itself cause the resulting work to be covered
11 # by the GNU General Public License.
13 # However the source code for this file must still be made available
14 # in accordance with section (3) of the GNU General Public License.
16 # This exception does not invalidate any other reasons why a work based
17 # on this file might be covered by the GNU General Public License.
19 # Copyright (c) 2016-2018 Intra2net AG <info@intra2net.com>
24 ------------------------------------------------------
25 Utility to customize email and other test data.
27 Copyright: Intra2net AG
31 ------------------------------------------------------
38 log = logging.getLogger('pyi2ncommon.test_data_sync')
41 def append_email_id_header(data_dir):
43 Use to append unique autotest id header to emails.
45 :param str data_dir: directory containing the emails
49 for main_dir, _, files in os.walk(data_dir):
50 for i in range(len(files)):
51 file_path = os.path.join(main_dir, files[i])
53 log.info("%i done\n", i)
55 if not re.match(r"^\d+\.$", files[i]):
57 if os.path.getsize(file_path) == 0:
58 log.warning("Skipping empty file %s", file_path)
61 log.info("Adding header to email %s", file_path)
63 message_file = open(file_path, "r")
64 for line in message_file:
65 if re.match("^Message-Id:", line, re.IGNORECASE):
70 readed = open(file_path, "r").read()
71 autotest_id_line = "Autotest-Message-ID: <" + str(email_id + i) + ".63D49232CC@gwt-intranator.m.i2n>\r\n"
73 final = autotest_id_line + readed
75 final = readed.replace(id_line, autotest_id_line)
76 log.info("%s >> %s", id_line, autotest_id_line)
77 open(file_path, "w").write(final)
78 log.info("%i mails replaced.", len(files))