Пример #1
0
def config_fr(json_driver):
    """
    Mocked config object with force_register enabled.
    """
    conf = Config(
        cog_name="PyTest",
        unique_identifier=json_driver.unique_cog_identifier,
        driver=json_driver,
        force_registration=True,
    )
    yield conf
    conf._defaults = {}
Пример #2
0
    def __init__(self, bot: Red):
        self.bot = bot

        self.conf = Config.get_conf(self,
                                    identifier=998240343,
                                    force_registration=True)

        self.conf.register_global(installed=[])

        self.already_agreed = False

        self.LIB_PATH = cog_data_path(self) / "lib"
        self.SHAREDLIB_PATH = self.LIB_PATH / "cog_shared"
        self.SHAREDLIB_INIT = self.SHAREDLIB_PATH / "__init__.py"

        self.LIB_PATH.mkdir(parents=True, exist_ok=True)
        self.SHAREDLIB_PATH.mkdir(parents=True, exist_ok=True)
        if not self.SHAREDLIB_INIT.exists():
            with self.SHAREDLIB_INIT.open(mode="w", encoding="utf-8") as _:
                pass

        if str(self.LIB_PATH) not in syspath:
            syspath.insert(1, str(self.LIB_PATH))

        self._repo_manager = RepoManager()
Пример #3
0
 def __init__(self, bot: Red):
     self.config = Config.get_conf(self, identifier=5757575755)
     self.config.register_guild(**self.default_guild)
     self.config.register_member(**self.default_member)
     self.bot = bot
     loop = asyncio.get_event_loop()
     loop.create_task(self.register_warningtype())
Пример #4
0
 def __init__(self, bot: Red):
     self.bot = bot
     self.config = Config.get_conf(self, 78631113035100160, force_registration=True)
     self.config.register_guild(**self.default_guild_settings)
     self.config.register_custom("REPORT", **self.default_report)
     self.antispam = {}
     self.user_cache = []
     self.tunnel_store = {}
Пример #5
0
 def __init__(self, bot):
     self.bot = bot
     self.settings = Config.get_conf(self,
                                     identifier=2652104208,
                                     force_registration=True)
     self.settings.register_global(**self.default_global)
     self.session = aiohttp.ClientSession()
     self.imgur_base_url = "https://api.imgur.com/3/"
Пример #6
0
 def __init__(self, bot: Red):
     self.bot = bot
     self.file_path = "data/economy/settings.json"
     self.config = Config.get_conf(self, 1256844281)
     self.config.register_guild(**self.default_guild_settings)
     self.config.register_global(**self.default_global_settings)
     self.config.register_member(**self.default_member_settings)
     self.config.register_user(**self.default_user_settings)
     self.config.register_role(**self.default_role_settings)
     self.slot_register = defaultdict(dict)
Пример #7
0
 def __init__(self, bot: Red):
     self.bot = bot
     self.settings = Config.get_conf(self, 4766951341)
     default_guild_settings = {
         "filter": [],
         "filterban_count": 0,
         "filterban_time": 0,
         "filter_names": False,
         "filter_default_name": "John Doe",
     }
     default_member_settings = {"filter_count": 0, "next_reset_time": 0}
     self.settings.register_guild(**default_guild_settings)
     self.settings.register_member(**default_member_settings)
     self.register_task = self.bot.loop.create_task(self.register_filterban())
Пример #8
0
async def warning_points_remove_check(
    config: Config, ctx: commands.Context, user: discord.Member, points: int
):
    guild = ctx.guild
    guild_settings = config.guild(guild)
    act = {}
    async with guild_settings.actions() as registered_actions:
        for a in registered_actions:
            if points >= a["points"]:
                act = a
            else:
                break
    if act and act["drop_command"] is not None:  # some action needs to be taken
        await create_and_invoke_context(ctx, act["drop_command"], user)
Пример #9
0
    def __init__(self):
        self.trivia_sessions = []
        self.conf = Config.get_conf(self, identifier=UNIQUE_ID, force_registration=True)

        self.conf.register_guild(
            max_score=10,
            timeout=120.0,
            delay=15.0,
            bot_plays=False,
            reveal_answer=True,
            payout_multiplier=0.0,
            allow_override=True,
        )

        self.conf.register_member(wins=0, games=0, total_score=0)
Пример #10
0
    def __init__(self, bot: Red):
        self.db = Config.get_conf(self, 26262626)

        self.db.register_global(**self.global_defaults)

        self.db.register_guild(**self.guild_defaults)

        self.db.register_role(**self.role_defaults)

        self.bot: Red = bot

        self.streams: List[Stream] = []
        self.communities: List[TwitchCommunity] = []
        self.task: Optional[asyncio.Task] = None

        self.yt_cid_pattern = re.compile("^UC[-_A-Za-z0-9]{21}[AQgw]$")
Пример #11
0
def config(json_driver):
    conf = Config(cog_name="PyTest",
                  unique_identifier=json_driver.unique_cog_identifier,
                  driver=json_driver)
    yield conf
    conf._defaults = {}
Пример #12
0
    def __init__(self, bot: Red):
        self.bot = bot
        self._aliases = Config.get_conf(self, 8927348724)

        self._aliases.register_global(**self.default_global_settings)
        self._aliases.register_guild(**self.default_guild_settings)
Пример #13
0
 def __init__(self, bot):
     self.bot = bot
     self.key = 414589031223512
     self.config = Config.get_conf(self, self.key)
     self.config.register_guild(commands={})
     self.commandobj = CommandObj(config=self.config, bot=self.bot)
Пример #14
0
    def __init__(self, name: str, balance: int, created_at: datetime.datetime):
        self.name = name
        self.balance = balance
        self.created_at = created_at


def _register_defaults():
    _conf.register_global(**_DEFAULT_GLOBAL)
    _conf.register_guild(**_DEFAULT_GUILD)
    _conf.register_member(**_DEFAULT_MEMBER)
    _conf.register_user(**_DEFAULT_USER)


if not os.environ.get("BUILDING_DOCS"):
    _conf = Config.get_conf(None, 384734293238749, cog_name="Bank", force_registration=True)
    _register_defaults()


def _encoded_current_time() -> int:
    """Get the current UTC time as a timestamp.
    
    Returns
    -------
    int
        The current UTC timestamp.

    """
    now = datetime.datetime.utcnow()
    return _encode_time(now)