wayfinderak/Geeksbot
1
0
Fork 0
Code Issues Pull Requests Actions 118 Packages Projects Releases Wiki Activity

Process pages for book

Browse Source
This commit is contained in:
Dusty.P 2018-06-06 21:21:35 -08:00
parent 3700ab2344
commit 0600b883fe
1 changed files with 14 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
exts/imports/utils.py
View File

@ -93,20 +93,21 @@ async def run_command(args):
class Paginator:
def __init__(self, *,
max_chars: int=1990,
max_chars: int=1970,
max_lines: int=20,
prefix: str='```md',
suffix: str='```',
page_break: str='\uFFF8',
max_line_length: int=100):
_max_len = 1980
assert 0 < max_lines <= max_chars
assert 0 < max_line_length < 120
self._parts = list()
self._prefix = prefix
self._suffix = suffix
self._max_chars = max_chars if max_chars + len(prefix) + len(suffix) + 2 <= 2000 \
else 2000 - len(prefix) - len(suffix) - 2
self._max_chars = max_chars if max_chars + len(prefix) + len(suffix) + 2 <= _max_len \
else _max_len - len(prefix) - len(suffix) - 2
self._max_lines = max_lines - (prefix + suffix).count('\n') + 1
self._page_break = page_break
self._max_line_length = max_line_length
@ -116,7 +117,6 @@ class Paginator:
pages = list()
page = ''
lines = 0
parts = list()
def open_page():
nonlocal page, lines
@ -149,6 +149,15 @@ class Paginator:
self._pages = pages
return pages
def process_pages(self) -> typing.List[str]:
_pages = self._pages or self.pages()
_len_pages = len(_pages)
_len_page_str = len(f'{_len_pages}/{_len_pages}')
for i, page in enumerate(_pages):
if len(page) + _len_page_str <= 2000:
_pages[i] = f'{i + 1}/{_len_pages}\n{page}'
return _pages
def __len__(self):
return sum(len(p) for p in self._parts)
@ -215,7 +224,7 @@ class Book:
if pag == Paginator():
raise RuntimeError('Cannot create a book out of an empty Paginator.')
self._pages = pag.pages()
self._pages = pag.process_pages()
self._len_pages = len(self._pages)
self._current_page = 0
self._message, self._channel, self._bot = ctx