app-MAIL-temp/app/api/views/new_random_alias.py

59 lines
1.6 KiB
Python
Raw Normal View History

2020-01-05 21:15:16 +01:00
from flask import g
from flask import jsonify, request
from flask_cors import cross_origin
from app.api.base import api_bp, verify_api_key
from app.config import MAX_NB_EMAIL_FREE_PLAN
2020-01-05 21:15:16 +01:00
from app.extensions import db
from app.log import LOG
from app.models import Alias, AliasUsedOn, AliasGeneratorEnum
2020-01-05 21:15:16 +01:00
@api_bp.route("/alias/random/new", methods=["POST"])
@cross_origin()
@verify_api_key
def new_random_alias():
"""
Create a new random alias
Input:
(Optional) note
2020-01-05 21:15:16 +01:00
Output:
201 if success
"""
user = g.user
if not user.can_create_new_alias():
LOG.d("user %s cannot create new random alias", user)
return (
jsonify(
error=f"You have reached the limitation of a free account with the maximum of "
f"{MAX_NB_EMAIL_FREE_PLAN} aliases, please upgrade your plan to create more aliases"
),
400,
)
note = None
2020-03-11 14:02:35 +01:00
data = request.get_json(silent=True)
if data:
note = data.get("note")
2020-01-05 21:15:16 +01:00
scheme = user.alias_generator
mode = request.args.get("mode")
if mode:
if mode == "word":
scheme = AliasGeneratorEnum.word.value
elif mode == "uuid":
scheme = AliasGeneratorEnum.uuid.value
else:
return jsonify(error=f"{mode} must be either word or alias"), 400
alias = Alias.create_new_random(user=user, scheme=scheme, note=note)
2020-01-05 21:15:16 +01:00
db.session.commit()
hostname = request.args.get("hostname")
if hostname:
AliasUsedOn.create(gen_email_id=alias.id, hostname=hostname)
2020-01-05 21:15:16 +01:00
db.session.commit()
return jsonify(alias=alias.email), 201