corbz/PYRSS-Bot
corbz/PYRSS-Bot
1
0
Fork 0
Code Issues 14 Pull Requests Actions Releases 2 Activity

check if formatdata func is coroutine func

Browse Source
This commit is contained in:
Corban-Lee Jones 2024-02-13 00:07:38 +00:00
parent 08ba995520
commit 45f519535a
1 changed files with 28 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

29
src/utils.py
View File

@ -1,5 +1,6 @@
"""A collection of utility functions that can be used in various places.""" """A collection of utility functions that can be used in various places."""
import asyncio
import aiohttp import aiohttp
import logging import logging
import async_timeout import async_timeout
@ -137,6 +138,10 @@ class PaginationView(View):
@button(emoji="◀️", style=ButtonStyle.blurple) @button(emoji="◀️", style=ButtonStyle.blurple)
async def backward(self, inter: Interaction, button: Button): async def backward(self, inter: Interaction, button: Button):
"""
Action the backwards button.
"""
self.index -= 1 self.index -= 1
await inter.response.defer() await inter.response.defer()
self.inter = inter self.inter = inter
@ -144,6 +149,10 @@ class PaginationView(View):
@button(emoji="▶️", style=ButtonStyle.blurple) @button(emoji="▶️", style=ButtonStyle.blurple)
async def forward(self, inter: Interaction, button: Button): async def forward(self, inter: Interaction, button: Button):
"""
Action the forwards button.
"""
self.index += 1 self.index += 1
await inter.response.defer() await inter.response.defer()
self.inter = inter self.inter = inter
@ -151,6 +160,12 @@ class PaginationView(View):
@button(emoji="⏭️", style=ButtonStyle.blurple) @button(emoji="⏭️", style=ButtonStyle.blurple)
async def start_or_end(self, inter: Interaction, button: Button): async def start_or_end(self, inter: Interaction, button: Button):
"""
Action the start and end button.
This button becomes return to start if at end, otherwise skip to end.
"""
# Determine if should skip to start or end
if self.index <= self.maxpage // 2: if self.index <= self.maxpage // 2:
self.index = self.maxpage self.index = self.maxpage
else: else:
@ -161,6 +176,10 @@ class PaginationView(View):
await self.navigate() await self.navigate()
async def navigate(self): async def navigate(self):
"""
Acts as an update method for the entire instance.
"""
log.debug("navigating to page: %s", self.index) log.debug("navigating to page: %s", self.index)
self.update_buttons() self.update_buttons()
@ -168,13 +187,21 @@ class PaginationView(View):
await self.inter.edit_original_response(embed=paged_embed, view=self) await self.inter.edit_original_response(embed=paged_embed, view=self)
async def create_paged_embed(self) -> Embed: async def create_paged_embed(self) -> Embed:
"""
Returns a copy of the known embed, but with data from the current page.
"""
embed = self.embed.copy() embed = self.embed.copy()
data, total_results = await self.getdata(self.index) data, total_results = await self.getdata(self.index)
self.maxpage = self.calc_total_pages(total_results, self.pagesize) self.maxpage = self.calc_total_pages(total_results, self.pagesize)
for i, item in enumerate(data): for i, item in enumerate(data):
i = self.calc_dataitem_index(i) i = self.calc_dataitem_index(i)
key, value = self.formatdata(i, item) if asyncio.iscoroutinefunction(self.formatdata):
key, value = await self.formatdata(i, item)
else:
key, value = self.formatdata(i, item)
embed.add_field(name=key, value=value, inline=False) embed.add_field(name=key, value=value, inline=False)
if not total_results: if not total_results: