easyctf-2017/server/api/utils.py

import datetime
import json
import random
import re
import requests
import string
import traceback
import unicodedata

from flask import current_app as app
from functools import wraps
from werkzeug.security import generate_password_hash, check_password_hash

__check_email_format = lambda email: re.match(".+@.+\..{2,}", email) is not None
__check_ascii = lambda s: all(c in string.printable for c in s)
__check_alphanumeric = lambda s: all(c in string.digits + string.ascii_uppercase + string.ascii_lowercase for c in s)

def hash_password(s):
	return generate_password_hash(s)

def check_password(hashed_password, try_password):
	return check_password_hash(hashed_password, try_password)

def generate_string(length=32, alpha=string.hexdigits):
	return "".join([random.choice(alpha) for x in range(length)])

def unix_time_millis(dt):
	epoch = datetime.datetime.utcfromtimestamp(0)
	return (dt - epoch).total_seconds() * 1000.0

def get_time_since_epoch():
	return unix_time_millis(datetime.datetime.now())

def flat_multi(multidict):
	flat = {}
	for key, values in multidict.items():
		value = values[0] if type(values) == list and len(values) == 1 else values
		flat[key] = value.encode("utf-8")
	return flat

def send_email(recipient, subject, body):
    api_key = app.config["MG_API_KEY"]
    data = {"from": "EasyCTF Administrator <%s>" % (app.config["ADMIN_EMAIL"]),
            "to": recipient,
            "subject": subject,
            "text": body
    }
    auth = ("api", api_key)
    return requests.post("https://api.mailgun.net/v3/%s/messages" % (app.config["MG_HOST"]), auth=auth, data=data)