nsfw.py

"""
MIT License

Copyright (c) 2017 Grok's naughty dev XAOS

Permission is hereby granted, free of charge, to any person obtaining a copy
of this software and associated documentation files (the "Software"), to deal
in the Software without restriction, including without limitation the rights
to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
copies of the Software, and to permit persons to whom the Software is
furnished to do so, subject to the following conditions:

The above copyright notice and this permission notice shall be included in all
copies or substantial portions of the Software.

THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
SOFTWARE.
"""

import discord
from discord.ext import commands
import bs4 as bs
import urllib.request
from bs4 import BeautifulSoup
from urllib.request import Request, urlopen
import json
import io
import safygiphy
from ext import embedtobox

class Nsfw:
    """ Nsfw commands """
    def __init__(self, bot):
        self.bot = bot

    async def __local_check(self, ctx):
        if not ctx.channel.is_nsfw():
            return False
        git = self.bot.get_cog('Git')
        if not await git.starred('verixx/selfbot.py'):
            return False
        return True

class Dick:

    def init(self, bot):
        self.bot = bot

    def dick_size(self, id):
        s = 0
        while id:
            s += id % 10
            id //= 10
        return (s%10) * 2
    
    @commands.group(invoke_without_command=True)
    async def nsfw(self, ctx):
        """ Get random lewds from the web """
        pass

    @nsfw.command()
    async def xbooru(self, ctx):
        """ Random image from Xbooru """
        try:
            try:
                await ctx.message.delete()
            except discord.Forbidden:
                pass
            await ctx.channel.trigger_typing()
            query = urllib.request.urlopen("http://xbooru.com/index.php?page=post&s=random").read()
            soup = bs.BeautifulSoup(query, 'html.parser')
            image = soup.find(id="image").get("src")
            last = str(image.split('?')[-2]).replace('//', '/').replace(':/', '://')
            em = discord.Embed(colour=discord.Colour(0xed791d))
            em.description = f'[Full Size Link*]({last})'
            em.set_image(url=last)
            em.set_footer(text='* click link at your own risk!')
            try:
                await ctx.send(embed=em)
            except discord.HTTPException:
                await ctx.send('Unable to send embeds here!')
                try:
                    async with ctx.session.get(image) as resp:
                        image = await resp.read()
                    with io.BytesIO(image) as file:
                        await ctx.send(file=discord.File(file, 'xbooru.png'))
                except discord.HTTPException:
                    await ctx.send(image)

        except Exception as e:
            await ctx.send(f'```{e}```')

    @commands.command(aliases=['gelbooru'])
    async def gel(self, ctx):
        """ Random image from Gelbooru """
        try:
            try:
                await ctx.message.delete()
            except discord.Forbidden:
                pass

            await ctx.channel.trigger_typing()
            query = urllib.request.urlopen("http://www.gelbooru.com/index.php?page=post&s=random").read()
            soup = bs.BeautifulSoup(query, 'html.parser')
            sans = soup.find_all('div', {'class': 'highres-show'})
            partial = soup.find(id="image").get("src")
            image = partial.replace('//', '/').replace(':/', '://')

            em = discord.Embed(colour=discord.Colour(0xed791d))
            em.description = f'[Full Size Link*]({image})'
            em.set_image(url=image)
            em.set_footer(text='* click link at your own risk!')
            try:
                await ctx.send(embed=em)
            except discord.HTTPException:
                # em_list = await embedtobox.etb(em)
                # for page in em_list:
                #    await ctx.send(page)
                await ctx.send('Unable to send embeds here!')
                try:
                    async with ctx.session.get(image) as resp:
                        image = await resp.read()
                    with io.BytesIO(image) as file:
                        await ctx.send(file=discord.File(file, 'gelbooru.png'))
                except discord.HTTPException:
                    await ctx.send(image)

        except Exception as e:
            await ctx.send(f'```{e}```')

    @commands.command(pass_context=True)
    async def dick(self, ctx, mention=""):
        """How big are you?"""
        id_regex = re.compile('<@!?(\d+)>')
        ids = id_regex.findall(ctx.message.content)
        if len(ids) > 0:
            for member in ids:
                await ctx.send(' <@{}> Size: 8{}D'.format(member, self.dick_size(int(member)) * '='))
        else:
            await ctx.send(' <@{}> Size: 8{}D'.format(str(ctx.message.author.id), self.dick_size(ctx.message.author.id) * '='))
    
    @nsfw.command()
    async def gif(self, ctx, *, tag):
        """ Get a random lewd gif
        Usage: gif <tag>
        Available tags: rule34, nsfw, hentai, tits... """
        try:
            await ctx.message.delete()
        except discord.Forbidden:
            pass
        g = safygiphy.Giphy()
        gif = g.random(tag=tag)
        color = await ctx.get_dominant_color(ctx.author.avatar_url)
        em = discord.Embed(color=color)
        em.set_image(url=str(gif.get('data', {}).get('image_original_url')))
        try:
            await ctx.send(embed=em)
        except discord.HTTPException:
            em_list = await embedtobox.etb(em)
            for page in em_list:
                await ctx.send(page)


def setup(bot):
    bot.add_cog(Nsfw(bot))