def get_email(user):
server = IMAPClient(HOST, use_uid=True, ssl=True)
username = db.get("user:%s:login" % user)
password = db.get("user:%s:password" % user)
server.login(username, password)
server.select_folder('INBOX', readonly=True)
messages = server.search(['NOT DELETED','SINCE 15-May-2014'])
response = server.fetch(messages, ['RFC822', 'FLAGS'])
for msgid, data in response.iteritems():
# check for duplicates
duplicate = db.zrangebyscore("mail:%s:inbox" % user, msgid, msgid)
if duplicate:
continue
emailUTF8 = data['RFC822'].encode('utf-8')
msg = parser.parsestr(emailUTF8)
body = extract_body(msg).encode('utf-8')
msg['message'] = body
msg['subject'] = ('NoSubj' if (msg['Subject'] == None or msg['Subject'].encode('utf-8') == "".encode('utf-8')) else msg['Subject'])
msg['to'] = ('NoTo' if (msg['To'] == None) else msg['To'])
plain = {'plain_body': extract_body_text(msg).encode('utf-8'), 'subject': msg['subject']}
# TODO set unread
email = {'id': msgid, 'from': msg['From'], 'to': msg['To'], 'subject': msg['Subject'],
'date': msg['Date'], 'cc': msg['CC'], 'read': False,
'message': body, 'categorized': False, 'summary': shorten(plain), 'archived': False}
trained = db.get("user:%s:trained" % user)
if trained == "true":
email['category'] = int(classify(msg, user))
else:
email['category'] = 1
emailJSON = json.dumps(email, sort_keys=True, indent=4, separators=(',', ': '))
db.zadd("mail:%s:inbox" % user, emailJSON, msgid)
db.sadd("mail:%s:%s" % (user, email['category']), msgid)
server.logout()
def signup():
if current_user is not None and current_user.is_authenticated():
return redirect(url_for("index"))
form = LoginForm()
if form.validate_on_submit():
user = User(username=form.email.data, password=form.password.data)
if user.is_valid_login():
login_user(user)
return redirect(url_for("index"))
elif form.new_account.data:
login_user(user)
# add login, add password
db.getset("user:%s:login" % user.user, user.username)
db.getset("user:%s:password" % user.user, user.password)
# add to email list
db.sadd("user:users", user.user)
# add "untrained" to models
db.getset("user:%s:trained" % user.user, "false")
# add category
category = [{"id": 1, "emails": 3, "split": 0, "name": "inbox", "class": "category-inbox"}]
categoryJSON = json.dumps(category, sort_keys=True, indent=4, separators=(",", ": "))
db.getset("user:%s:categories" % user.user, categoryJSON)
return redirect(url_for("index"))
return render_template("signup.html", form=form, title="Sign In")
