gitbot/cogs/python/pypi.py at e81fb4f37f22900ec4a6c8888a8d503b0eeef50c · statch/gitbot · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
import io
import discord
import plotly.express as px
import plotly.io
import plotly.graph_objects as go
import pandas as pd
from discord.ext import commands
from lib.utils.decorators import gitbot_group
from typing import Optional
from pkg_resources import parse_version
from lib.typehints import PyPIProject
from lib.structs import GitBotEmbed, GitBot
from lib.structs.discord.context import GitBotContext


class PyPI(commands.Cog):
    def __init__(self, bot: GitBot):
        self.bot: GitBot = bot

    @gitbot_group('pypi', invoke_without_command=True)
    @commands.cooldown(5, 30, commands.BucketType.user)
    async def pypi_command_group(self, ctx: GitBotContext, project: Optional[PyPIProject] = None) -> None:
        if project is not None:
            await ctx.invoke(self.project_info_command, project=project)
        else:
            commands_: list = [
                f'`git pypi {{{ctx.l.help.argument_explainers.python_package_name.name}}}` - {ctx.l.pypi.default.commands.info}',
                f'`git pypi downloads {{{ctx.l.help.argument_explainers.python_package_name.name}}}` - {ctx.l.pypi.default.commands.downloads}'
            ]
            embed: GitBotEmbed = GitBotEmbed(
                    color=self.bot.mgr.c.languages.python,
                    title=ctx.l.pypi.default.title,
                    description=ctx.l.pypi.default.description
                                + '\n\n'
                                + '\n'.join(commands_),
                    thumbnail=self.bot.mgr.i.pip_logo,
                    url='https://pypi.org'
            )
            await ctx.send(embed=embed)

    @pypi_command_group.command('info', aliases=['i'])
    @commands.cooldown(5, 30, commands.BucketType.user)
    async def project_info_command(self, ctx: GitBotContext, project: PyPIProject) -> None:
        ctx.fmt.set_prefix('pypi info')
        data: Optional[dict] = await self.bot.pypi.get_project_data(project.lower())
        if not data:
            return await ctx.error(ctx.l.generic.nonexistent.python_package)
        gravatar: str = (self.bot.mgr.construct_gravatar_url(data['info']['author_email'],
                                                            default=self.bot.mgr.i.pip_logo)
                         if data['info']['author_email'] else self.bot.mgr.i.pip_logo)
        embed: GitBotEmbed = GitBotEmbed(
                color=0x3572a5,
                title=f'{data["info"]["name"]} `{data["info"]["version"]}`',
                url=data['info']['project_url'],
                thumbnail=gravatar
        )

        if data['info']['summary'] is not None and len(data['info']['summary']) != 0:
            summary: str = self.bot.mgr.sanitize_codeblock_content(data['info']['summary'].strip())
            embed.add_field(name=f":notepad_spiral: {ctx.l.pypi.info.glossary[0]}:",
                            value=f"```{summary}```")
        author: str = ctx.fmt('author', f'[{(author := data["info"]["author"])}]'
                                        f'({await self.bot.mgr.ensure_http_status(f"https://pypi.org/user/{author}", alt="")})') + '\n'

        first_release = (None, None)
        for tag_name, release in data['releases'].items():
            if (v := parse_version(tag_name)) and first_release[0] is None or first_release[0] > v:
                first_release = v, release
        first_uploaded_at: str = f''
        first_release: tuple[..., list | ...] | list[..., list | ...]
        if first_release[1]:
            first_uploaded_at: str = ctx.fmt('first_upload',
                                             self.bot.mgr.external_to_discord_timestamp(
                                                     first_release[1][0]["upload_time"],
                                                     "%Y-%m-%dT%H:%M:%S")) + '\n'

        requires_python: str = ctx.fmt('requires_python', f'`{data["info"]["requires_python"]}`') + '\n'
        info: str = f'{author}{first_uploaded_at}{requires_python}'
        embed.add_field(name=f":mag_right: {ctx.l.pypi.info.glossary[1]}:", value=info)

        homepage: tuple = (data['info']['home_page'] if 'home_page' in data['info'] and data['info']['home_page'] else None, ctx.l.pypi.info.glossary[3])
        docs: tuple = (data['info']['docs_url'] if 'docs_url' in data['info'] and data['info']['docs_url'] else None, ctx.l.pypi.info.glossary[4])
        bugs: tuple = (data['info']['bugtrack_url'] if 'bugtrack_url' in data['info'] and data['info']['bugtrack_url'] else None, ctx.l.pypi.info.glossary[4])
        links: list = [homepage, docs, bugs]
        link_strings: list = []
        for lnk in links:
            if lnk[0] is not None and len(lnk[0]) != 0:
                link_strings.append(f"- [{lnk[1]}]({lnk[0]})")
        if len(link_strings) != 0:
            embed.add_field(name=f":link: {ctx.l.pypi.info.glossary[2]}:", value='\n'.join(link_strings))

        if 'license' in data['info'] and data['info']['license']:
            embed.set_footer(text=self.bot.mgr.truncate(ctx.fmt('license', data['info']['license']), length=90, full_word=True))

        await ctx.send(embed=embed, view_on_url=data['info']['project_url'])


    @pypi_command_group.command('downloads', aliases=['dl', 'stats', 'statistics'])
    @commands.cooldown(3, 30, commands.BucketType.user)
    @commands.max_concurrency(7)
    async def project_downloads_command(self, ctx: GitBotContext, project: PyPIProject) -> None:
        ctx.fmt.set_prefix('pypi downloads')
        downloads_overall: Optional[dict] = await self.bot.pypi.get_project_overall_downloads(project)
        if downloads_overall and (data := downloads_overall['data']):
            downloads_recent: dict = (await self.bot.pypi.get_project_recent_downloads(project))['data']
            df: pd.DataFrame = pd.DataFrame({'date': [item['date'] for item in data],
                                             'downloads': [item['downloads'] for item in data]})
            fig: go.Figure = px.line(df,
                                     x='date',
                                     y='downloads',
                                     labels={'date': ctx.l.pypi.downloads.glossary[0],
                                             'downloads': ctx.l.pypi.downloads.glossary[1]},
                                     template='plotly_dark')
            embed: GitBotEmbed = GitBotEmbed(
                    color=self.bot.mgr.c.rounded,
                    title=ctx.fmt('title', project, len(data) - 1),
                    url=f'https://pypistats.org/packages/{project.replace(".", "-").lower()}',
                    description=f'{ctx.fmt("stats yesterday", downloads_recent["last_day"])}\n'
                                f'{ctx.fmt("stats last_week", downloads_recent["last_week"])}\n'
                                f'{ctx.fmt("stats last_month", downloads_recent["last_month"])}',
                    thumbnail=self.bot.mgr.i.pip_logo,
                    footer=ctx.l.pypi.downloads.footer
            )
            await ctx.reply(embed=embed, file=discord.File(fp=io.BytesIO(plotly.io.to_image(fig,
                                                                                            format='png',
                                                                                            engine='kaleido')),
                                                           filename=f'{project}-downloads-overall.png'),
                            mention_author=False, view_on_url=f'https://pypistats.org/packages/{project.replace(".", "-").lower()}')
        else:
            await ctx.error(ctx.l.generic.nonexistent.python_package)


async def setup(bot: GitBot) -> None:
    await bot.add_cog(PyPI(bot))