예제 #1
0
from discord.ext import commands
from config import config
from features import verification
from repository import user_repo

user_r = user_repo.UserRepository()

config = config.Config


class Verify(commands.Cog):
    def __init__(self, bot):
        self.bot = bot
        self.verification = verification.Verification(bot, user_r)

    @commands.cooldown(rate=5, per=30.0, type=commands.BucketType.user)
    @commands.command()
    async def verify(self, ctx):
        await self.verification.verify(ctx.message)

    @commands.cooldown(rate=5, per=30.0, type=commands.BucketType.user)
    @commands.command()
    async def getcode(self, ctx):
        await self.verification.send_code(ctx.message)


def setup(bot):
    bot.add_cog(Verify(bot))
예제 #2
0
import re
import smtplib
import string

from email.mime.text import MIMEText
from email.mime.multipart import MIMEMultipart

import discord
from discord.ext import commands

from cogs.resource import CogConfig, CogText
from core import acl, rubbercog, utils
from core.config import config
from repository import user_repo

repo_u = user_repo.UserRepository()


class Verify(rubbercog.Rubbercog):
    """Verify your account"""
    def __init__(self, bot):
        super().__init__(bot)

        self.text = CogText("verify")
        self.config = CogConfig("verify")

    ##
    ## Commands
    ##

    @commands.check(acl.check)