paulbsd/paulbsd-salt
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

updated maildb state

Browse Source
This commit is contained in:
Paul 2024-01-27 10:42:03 +01:00
parent 03c24893d0
commit 914a774113
1 changed files with 17 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

32
states/maildb/templates/maildb.py
View File

@ -13,10 +13,10 @@ def DBConfig(configfile):
config.read(configfile) config.read(configfile)
d_dbconfig = { d_dbconfig = {
'host' : config.get("maildb", "hostname"), "host" : config.get("maildb", "hostname"),
'user' : config.get("maildb", "username"), "user" : config.get("maildb", "username"),
'passwd' : config.get("maildb", "password"), "passwd" : config.get("maildb", "password"),
'db' : config.get("maildb", "database") "db" : config.get("maildb", "database")
} }
return d_dbconfig return d_dbconfig
@ -35,14 +35,16 @@ def CloseDB(conn):
def CreateDB(conn, cursor): def CreateDB(conn, cursor):
try: try:
cursor.execute('''CREATE TABLE IF NOT EXISTS mail ( cursor.execute('''
id serial NOT NULL, CREATE TABLE IF NOT EXISTS mail (
sender text NULL, id serial NOT NULL,
recipient text NULL, sender text NULL,
"date" timestamp NOT NULL DEFAULT NOW(), recipient text NULL,
"content" text NULL, "date" timestamp NOT NULL DEFAULT NOW(),
CONSTRAINT mail_pkey PRIMARY KEY (id) "content" text NULL,
)''') CONSTRAINT mail_pkey PRIMARY KEY (id)
)
''')
conn.commit() conn.commit()
except Exception as e: except Exception as e:
print(e) print(e)
@ -64,8 +66,8 @@ def ParseMail(input_data):
content += line content += line
msgobj = email.message_from_string(content) msgobj = email.message_from_string(content)
sender = parseaddr(msgobj['From'])[1] sender = parseaddr(msgobj["From"])[1]
recipient = parseaddr(msgobj['To'])[1] recipient = parseaddr(msgobj["To"])[1]
return sender, recipient, content return sender, recipient, content
@ -92,5 +94,5 @@ def main():
CloseDB(conn) CloseDB(conn)
if __name__ == '__main__': if __name__ == "__main__":
main() main()