mirror of https://github.com/stijndcl/didier
29 lines
925 B
Python
29 lines
925 B
Python
import discord
|
|
from overrides import overrides
|
|
|
|
from database.schemas import Bookmark
|
|
from didier.menus.common import PageSource
|
|
|
|
__all__ = ["BookmarkSource"]
|
|
|
|
from didier.utils.discord.assets import get_author_avatar
|
|
|
|
|
|
class BookmarkSource(PageSource[Bookmark]):
|
|
"""PageSource for the Bookmark commands"""
|
|
|
|
@overrides
|
|
def create_embeds(self):
|
|
for page in range(self.page_count):
|
|
embed = discord.Embed(title="Bookmarks", colour=discord.Colour.blue())
|
|
avatar_url = get_author_avatar(self.ctx).url
|
|
embed.set_author(name=self.ctx.author.display_name, icon_url=avatar_url)
|
|
|
|
description_data = []
|
|
|
|
for bookmark in self.get_page_data(page):
|
|
description_data.append(f"`#{bookmark.bookmark_id}`: [{bookmark.label}]({bookmark.jump_url})")
|
|
|
|
embed.description = "\n".join(description_data)
|
|
self.embeds.append(embed)
|