From 2ebd8c5235b2aff7e3ff90e838165d3e9ddb6380 Mon Sep 17 00:00:00 2001 From: khuxkm fbexl Date: Sat, 17 Aug 2019 02:46:14 +0000 Subject: [PATCH] Write new QEC log for each email --- parse_email.py | 4 ++++ utils.py | 24 +++++++++++++++++++++++- 2 files changed, 27 insertions(+), 1 deletion(-) diff --git a/parse_email.py b/parse_email.py index e5ac5e6..4dd668a 100755 --- a/parse_email.py +++ b/parse_email.py @@ -25,3 +25,7 @@ if payload==payload.rstrip(): log("title = {!r}".format(title)) log("payload = {!r}".format(payload)) + +filename = utils.get_name_from_title(title) + +utils.write_file(filename,payload) diff --git a/utils.py b/utils.py index f5dee57..c5cd5df 100644 --- a/utils.py +++ b/utils.py @@ -1,7 +1,10 @@ -import sys,os,contextlib,builtins,email +import sys,os,contextlib,builtins,email,re + +NOT_ALPHANUM = re.compile("[^A-Za-z0-9]") inp = email.message_from_file(sys.stdin) homedir = os.path.dirname(__file__) +ship = "anon.hmm.st" def input(): return inp @@ -18,3 +21,22 @@ def open(*args,**kwargs): yield f finally: f.close() + +def getshipdir(): + return os.path.join(homedir,"ships",ship) + +def get_name_from_title(title): + shipdir = getshipdir() + t = NOT_ALPHANUM.sub("-",title) + t = t[:10] + if os.path.exists(os.path.join(shipdir,t+".txt")): + n = 1 + while os.path.exists(os.path.join(shipdir,t+str(n)+".txt")): + n+=1 + return os.path.join(shipdir,t+str(num)+".txt") + else: return os.path.join(shipdir,t+".txt") + +def write_file(filename,text): + with open(filename,"w") as f: + f.write(text) + os.chmod(filename,0o666)