Пример #1
0
Файл: main.py Проект: sysr-q/chr
def index():
    form = get_shrink_form()
    surl = None
    render = lambda: render_template("index.html", form=form, url=surl)
    if form.validate_on_submit():
        if form.custom.data and url.exists(form.custom.data):
            _flash("Sorry, that custom URL already exists.", "error")
            return render()
        url_ = url.add(
            form.url.data,
            statistics=form.statistics.data,
            burn=form.burn.data,
            short=form.custom.data if form.custom.data else None,
            ua=request.user_agent.string,
            ip=request.remote_addr
        )
        if url_:
            _flash("Successfully shrunk your URL!", "success")
            surl = {
                "long": form.url.data,
                "short": url_for("reroute", short=url_[0], _external=True),
                "stats": url_for("stats", short=url_[0], _external=True),
                "delete": url_for("delete", short=url_[0], key=url_[1], _external=True),
                "burn": form.burn.data,
            }
            form.url.data = form.custom.data = ""
            form.statistics.data = True
            form.burn.data = False
    return render()
Пример #2
0
Файл: main.py Проект: sysr-q/chr
def after_request(res):
    if not g.is_api_req:
        # Flash all our flashes
        map(lambda f: _flash(*f), g.flashes)
    if not hasattr(res, "_ret_dict") or res._ret_dict is None:
        # Too specific for us, let's not screw around.
        return res
    res = res._ret_dict
    errors = filter(lambda f: len(f) >= 2 and f[1] == "error", g.flashes)
    if g.is_api_req:
        res["error"] = bool(errors)
        # Backwards compat.. Sort of?
        res.setdefault("message", "Check `messages` instead, buddy.")
        res["messages"] = map(lambda f: f[0], g.flashes)
        # Not interesting to API consumers.
        res.pop("_template", None)
        return jsonify(res)
    return make_response(render_template(res["_template"], res=res))
Пример #3
0
 def __call__(self, message: str, level: str = 'info') -> None:
     _flash(message, level)
Пример #4
0
 def __call__(self, errors, category):
     if not isinstance(errors, list):
         errors = [errors]
     for msg in errors:
         _flash(msg, category)
Пример #5
0
def flash(message, category='info'):
    _flash(message, category)