mirror of
https://github.com/Mayuri-Chan/pyrofork.git
synced 2025-12-29 12:04:51 +00:00
8aee144313
Co-authored-by: Alisson Lauffer <alissonvitortc@gmail.com> Co-authored-by: Yasir Aris M <git@yasirdev.my.id> Co-authored-by: wulan17 <wulan17@nusantararom.org> Signed-off-by: Yasir Aris M <git@yasirdev.my.id> Signed-off-by: wulan17 <wulan17@nusantararom.org>
94 lines
No EOL
3.4 KiB
Python
94 lines
No EOL
3.4 KiB
Python
"""
|
||
pyromod - A monkeypatcher add-on for Pyrogram
|
||
Copyright (C) 2020 Cezar H. <https://github.com/usernein>
|
||
|
||
This file is part of pyromod.
|
||
|
||
pyromod is free software: you can redistribute it and/or modify
|
||
it under the terms of the GNU General Public License as published by
|
||
the Free Software Foundation, either version 3 of the License, or
|
||
(at your option) any later version.
|
||
|
||
pyromod is distributed in the hope that it will be useful,
|
||
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||
GNU General Public License for more details.
|
||
|
||
You should have received a copy of the GNU General Public License
|
||
along with pyromod. If not, see <https://www.gnu.org/licenses/>.
|
||
"""
|
||
import math
|
||
from pyrogram.helpers import array_chunk
|
||
|
||
class Pagination:
|
||
def __init__(self, objects, page_data=None, item_data=None, item_title=None):
|
||
def default_page_callback(x):
|
||
return str(x)
|
||
|
||
def default_item_callback(i, pg):
|
||
return f"[{pg}] {i}"
|
||
|
||
self.objects = objects
|
||
self.page_data = page_data or default_page_callback
|
||
self.item_data = item_data or default_item_callback
|
||
self.item_title = item_title or default_item_callback
|
||
|
||
def create(self, page, lines=5, columns=1):
|
||
quant_per_page = lines * columns
|
||
page = 1 if page <= 0 else page
|
||
offset = (page - 1) * quant_per_page
|
||
stop = offset + quant_per_page
|
||
cutted = self.objects[offset:stop]
|
||
|
||
total = len(self.objects)
|
||
pages_range = [
|
||
*range(1, math.ceil(total / quant_per_page) + 1)
|
||
] # each item is a page
|
||
last_page = len(pages_range)
|
||
|
||
nav = []
|
||
if page <= 3:
|
||
for n in [1, 2, 3]:
|
||
if n not in pages_range:
|
||
continue
|
||
text = f"· {n} ·" if n == page else n
|
||
nav.append((text, self.page_data(n)))
|
||
if last_page >= 4:
|
||
nav.append(("4 ›" if last_page > 5 else 4, self.page_data(4)))
|
||
if last_page > 4:
|
||
nav.append(
|
||
(
|
||
f"{last_page} »" if last_page > 5 else last_page,
|
||
self.page_data(last_page),
|
||
)
|
||
)
|
||
elif page >= last_page - 2:
|
||
nav.extend(
|
||
[
|
||
("« 1" if last_page - 4 > 1 else 1, self.page_data(1)),
|
||
(
|
||
f"‹ {last_page - 3}" if last_page - 4 > 1 else last_page - 3,
|
||
self.page_data(last_page - 3),
|
||
),
|
||
]
|
||
)
|
||
for n in range(last_page - 2, last_page + 1):
|
||
text = f"· {n} ·" if n == page else n
|
||
nav.append((text, self.page_data(n)))
|
||
else:
|
||
nav = [
|
||
("« 1", self.page_data(1)),
|
||
(f"‹ {page - 1}", self.page_data(page - 1)),
|
||
(f"· {page} ·", "noop"),
|
||
(f"{page + 1} ›", self.page_data(page + 1)),
|
||
(f"{last_page} »", self.page_data(last_page)),
|
||
]
|
||
|
||
buttons = []
|
||
for item in cutted:
|
||
buttons.append((self.item_title(item, page), self.item_data(item, page)))
|
||
kb_lines = array_chunk(buttons, columns)
|
||
if last_page > 1:
|
||
kb_lines.append(nav)
|
||
|
||
return kb_lines |