adminsys
/
spamotron
8
0
Fork 0
Code Pull Requests Releases Wiki Activity

lit les datas depuis un csv et permet des champs par défaut

This commit is contained in:
François Poulain 2020-08-24 21:05:22 +02:00
parent 03d2a29311
commit a212c2bdd8
1 changed files with 24 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

33
mailing.py
View File

@ -22,7 +22,7 @@
# This program is used to mail a message to several recipients, thus # This program is used to mail a message to several recipients, thus
# providing customization for each recipient. # providing customization for each recipient.
import argparse, sys, re import argparse, csv, sys, re
import smtplib, time, mimetypes import smtplib, time, mimetypes
from email.header import Header from email.header import Header
from email.message import EmailMessage from email.message import EmailMessage
@ -34,10 +34,9 @@ def parse_args():
ma.add_argument( ma.add_argument(
"-t", "-t",
"--tofile", "--tofile",
metavar="TO.FILE",
type=str, type=str,
required=True, required=True,
help="Sort of CSV file containing addresses of recipients.", help="CSV file containing recipients data.",
) )
ma.add_argument( ma.add_argument(
"-b", "-b",
@ -62,11 +61,25 @@ def parse_args():
return parser.parse_args() return parser.parse_args()
# read the recipients file where values are separated by | characters def read_recipients_datas(args):
def read_recipients(args): with open(args.tofile) as csv_file:
recipientfile = open(args.tofile) csv_soup = csv.reader(csv_file)
lines = recipientfile.readlines() first = [(s.split(":", 1)) for s in csv_soup.__next__()]
return [line[:-1].split("|") for line in lines] header = [
(h[0].strip().lower(), h[1].strip() if len(h) == 2 else None) for h in first
]
datas = [[s.strip() for s in d] for d in csv_soup]
full_datas = []
for data in datas:
try:
full_data = {
h[0]: data[i] if i < len(data) or not h[1] else h[1]
for i, h in enumerate(header)
}
full_datas.append(full_data)
except Exception:
print("Ligne mal renseignée:", data, file=sys.stderr)
return full_datas
def read_body(args): def read_body(args):
@ -204,7 +217,9 @@ if __name__ == "__main__":
args = parse_args() args = parse_args()
# read the recipients file # read the recipients file
sets = read_recipients(args) datas = read_recipients_datas(args)
print(datas)
exit(0)
# read the template of the mail # read the template of the mail
bodytemplate = read_body(args) bodytemplate = read_body(args)