.
This commit is contained in:
		
							parent
							
								
									7222239774
								
							
						
					
					
						commit
						c540f624af
					
				@ -897,13 +897,36 @@ class SpicePayCog(commands.Cog):
 | 
			
		||||
 | 
			
		||||
# ------------------------ setup ------------------------
 | 
			
		||||
 | 
			
		||||
async def setup(bot):
 | 
			
		||||
async def setup(bot: commands.Bot):
 | 
			
		||||
    cog = SpicePayCog(bot)
 | 
			
		||||
    await bot.add_cog(cog)
 | 
			
		||||
 | 
			
		||||
    home_gid = cfg(bot).int('home_guild_id', 0)
 | 
			
		||||
    # If you use cfg(bot), great; otherwise fall back to DEFAULT.
 | 
			
		||||
    try:
 | 
			
		||||
        from modules.common.settings import cfg as _cfg
 | 
			
		||||
        home_gid = _cfg(bot).int('home_guild_id', 0)
 | 
			
		||||
    except Exception:
 | 
			
		||||
        try:
 | 
			
		||||
            home_gid = int(bot.config['DEFAULT'].get('home_guild_id', '0'))
 | 
			
		||||
        except Exception:
 | 
			
		||||
            home_gid = 0
 | 
			
		||||
 | 
			
		||||
    guild_obj = discord.Object(id=home_gid) if home_gid else None
 | 
			
		||||
 | 
			
		||||
    # Make reloads safe: remove if present, then add.
 | 
			
		||||
    def _rm(name: str):
 | 
			
		||||
        try:
 | 
			
		||||
            bot.tree.remove_command(name, guild=guild_obj)
 | 
			
		||||
        except Exception:
 | 
			
		||||
            try:
 | 
			
		||||
                bot.tree.remove_command(name, guild=None)
 | 
			
		||||
            except Exception:
 | 
			
		||||
                pass
 | 
			
		||||
 | 
			
		||||
    for name in ("spicepay", "spicepay_resume", "spicepay_cancel", "spicepay_config"):
 | 
			
		||||
        _rm(name)
 | 
			
		||||
 | 
			
		||||
    if home_gid:
 | 
			
		||||
        guild_obj = discord.Object(id=home_gid)
 | 
			
		||||
        bot.tree.add_command(cog.spicepay, guild=guild_obj)
 | 
			
		||||
        bot.tree.add_command(cog.spicepay_resume, guild=guild_obj)
 | 
			
		||||
        bot.tree.add_command(cog.spicepay_cancel, guild=guild_obj)
 | 
			
		||||
 | 
			
		||||
		Loading…
	
		Reference in New Issue
	
	Block a user