Redirect url to correct one when id is correct. Consider blog with /id/slug as url for a specific post if user enters /id/foo it must redirect to /id/slug

Flask.py

@app.route('/post/<id>')
@app.route('/post/<id>/<slug>')
def view_post(id, slug=None):
    post = models.Post.query.get_or_404(id)
    if post.slug != slug:
        return redirect(url_for('post', id=id,slug = post.slug))
    return render_template("post.html", post=post)