flask-paginate__example/app.py - master - ilya.petrash/SimplePyScripts

flask-paginate__example/app.py
143 строки3 KB
gil9red
Refactoring. Using f-strings
26 июн 2023, 13:38
26 июн 2023, 13:38a1e58ba
#!/usr/bin/env python
# -*- coding: utf-8 -*-


import sqlite3

import click

from flask import Flask, render_template, g, current_app
from flask_paginate import Pagination, get_page_args


click.disable_unicode_literals_warning = True

app = Flask(__name__)
app.config.from_pyfile("app.cfg")


@app.before_request
def before_request():
    g.conn = sqlite3.connect("test.db")
    g.conn.row_factory = sqlite3.Row
    g.cur = g.conn.cursor()


@app.teardown_request
def teardown(error):
    if hasattr(g, "conn"):
        g.conn.close()


@app.route("/")
def index():
    g.cur.execute("select count(*) from users")
    total = g.cur.fetchone()[0]
    page, per_page, offset = get_page_args()
    sql = f"select name from users order by name limit {offset}, {per_page}"
    g.cur.execute(sql)
    users = g.cur.fetchall()
    pagination = get_pagination(
        page=page,
        per_page=per_page,
        total=total,
        record_name="users",
        format_total=True,
        format_number=True,
    )
    return render_template(
        "index.html",
        users=users,
        page=page,
        per_page=per_page,
        pagination=pagination,
    )


@app.route("/users/", defaults={"page": 1})
@app.route("/users", defaults={"page": 1})
@app.route("/users/page/<int:page>/")
@app.route("/users/page/<int:page>")
def users(page):
    g.cur.execute("select count(*) from users")
    total = g.cur.fetchone()[0]
    page, per_page, offset = get_page_args()
    sql = f"select name from users order by name limit {offset}, {per_page}"
    g.cur.execute(sql)
    users = g.cur.fetchall()
    pagination = get_pagination(
        page=page,
        per_page=per_page,
        total=total,
        record_name="users",
        format_total=True,
        format_number=True,
    )
    return render_template(
        "index.html",
        users=users,
        page=page,
        per_page=per_page,
        pagination=pagination,
        active_url="users-page-url",
    )


@app.route("/search/<name>/")
@app.route("/search/<name>")
def search(name):
    """The function is used to test multi values url."""
    sql = "select count(*) from users where name like ?"
    args = (f"%{name}%",)
    g.cur.execute(sql, args)
    total = g.cur.fetchone()[0]

    page, per_page, offset = get_page_args()
    sql = "select * from users where name like ? limit {}, {}"
    g.cur.execute(sql.format(offset, per_page), args)
    users = g.cur.fetchall()
    pagination = get_pagination(
        page=page,
        per_page=per_page,
        total=total,
        record_name="users",
    )
    return render_template(
        "index.html",
        users=users,
        page=page,
        per_page=per_page,
        pagination=pagination,
    )


def get_css_framework():
    return current_app.config.get("CSS_FRAMEWORK", "bootstrap3")


def get_link_size():
    return current_app.config.get("LINK_SIZE", "sm")


def show_single_page_or_not():
    return current_app.config.get("SHOW_SINGLE_PAGE", False)


def get_pagination(**kwargs):
    kwargs.setdefault("record_name", "records")
    return Pagination(
        css_framework=get_css_framework(),
        link_size=get_link_size(),
        show_single_page=show_single_page_or_not(),
        **kwargs
    )


@click.command()
@click.option("--port", "-p", default=5000, help="listening port")
def run(port):
    app.run(debug=True, port=port)


if __name__ == "__main__":
    run()