from datetime import datetime import discord from discord.ext import commands from discord.utils import get from markdownify import markdownify import re from bs4 import BeautifulSoup from hrochobot.utils.ksp_utils import ksp_feed import hrochobot.utils.data as data NEWS_JSON = "news" def guess_color(title): """ Automagically guess color of given post. Not always reliable as all things automagic. """ def contains(*regexes): return any(re.search(regex, title) for regex in regexes) if contains(r"(\d+)-Z(\d+)", "začátečnic", "KSP-Z"): return discord.Color.green() elif contains(r"(\d+)-(\d+)", "seriál", "série", "KSP-H"): return discord.Color.blue() else: return discord.Color.dark_purple() def format_entry(entry, author=None): content = "\n\n".join(map(lambda x: x.replace('\n', ' '), entry.summary.split("\n\n"))) embed = discord.Embed( title=entry.title, url=entry.link, description=markdownify(content, strip=["img"]), color=guess_color(entry.title), ) soup = BeautifulSoup(content, 'html.parser') img = soup.find('img') if img: embed.set_image(url=img['src']) if author: embed.set_author(name=author) embed.set_thumbnail(url="https://ksp.mff.cuni.cz/img/hippo_head.png") date = datetime.fromisoformat(entry.published) embed.set_footer(text=date.strftime("%-d. %-m. %Y")) return embed async def post_news(bot, guild, index): news_json = data.load_guild_data(guild.id, NEWS_JSON) if "news_channel" not in news_json: return channel = get(guild.channels, id=news_json["news_channel"]) feed = await ksp_feed() return await channel.send(embed=format_entry(feed.entries[index], author=feed.feed.author)) class News(commands.Cog): def __init__(self, bot): self.bot = bot news = discord.SlashCommandGroup( "news", "Commands for management of ksp news.", guild_only=True, checks=[commands.has_permissions(manage_guild=True)] ) @news.command(description="Adds a new secret role.") @discord.option("channel", discord.TextChannel, description="Channel for sending news.") async def set_channel(self, ctx, channel: discord.TextChannel): news_json = data.load_guild_data(ctx.guild.id, NEWS_JSON) news_json["news_channel"] = channel.id data.dump_guild_data(ctx.guild.id, NEWS_JSON, news_json) return await ctx.respond(f"News channel set to {channel.mention}.", ephemeral=True) @news.command(description="Synchronize news feed.") @discord.option("index", int, description="Send index-th most recent entry. (indexed from 0)") async def post_news(self, ctx, index: int): await post_news(self.bot, ctx.guild, index) return await ctx.respond(f"News posted.", ephemeral=True) def setup(bot): bot.add_cog(News(bot))