Update init

jarvis/__init__.py

@@ -1,13 +1,13 @@
 """Main J.A.R.V.I.S. package."""
 import logging
 
-from dis_snek import Intents, Snake
+from dis_snek import Intents, Snake, listen
 from mongoengine import connect
 
 # from jarvis import logo  # noqa: F401
 from jarvis import tasks, utils
 from jarvis.config import get_config
-from jarvis.events import guild, member, message
+from jarvis.events import member, message
 
 jconfig = get_config()
 
@@ -17,17 +17,17 @@ file_handler = logging.FileHandler(filename="jarvis.log", encoding="UTF-8", mode
 file_handler.setFormatter(logging.Formatter("[%(asctime)s][%(levelname)s][%(name)s] %(message)s"))
 logger.addHandler(file_handler)
 
-intents = Intents.default()
+intents = Intents.DEFAULT
 intents.members = True
 restart_ctx = None
 
 
-jarvis = Snake(intents=intents, default_prefix=utils.get_prefix, sync_interactions=jconfig.sync)
+jarvis = Snake(intents=intents, default_prefix="!", sync_interactions=jconfig.sync)
 
 __version__ = "2.0.0a0"
 
 
-@jarvis.add_listener
+@listen()
 async def on_ready() -> None:
     """Lepton on_ready override."""
     global restart_ctx
@@ -35,6 +35,12 @@ async def on_ready() -> None:
     print("    Connected to {} guild(s)".format(len(jarvis.guilds)))
 
 
+@listen()
+async def on_startup() -> None:
+    """Lepton on_startup override."""
+    tasks.init()
+
+
 def run() -> None:
     """Run J.A.R.V.I.S."""
     connect(
@@ -60,19 +66,11 @@ def run() -> None:
     )
 
     jarvis.max_messages = jconfig.max_messages
-    tasks.init()
 
     # Add event listeners
     if jconfig.events:
         _ = [
-            guild.GuildEventHandler(jarvis),
             member.MemberEventHandler(jarvis),
             message.MessageEventHandler(jarvis),
         ]
-    jarvis.run(jconfig.token, bot=True, reconnect=True)
-    for cog in jarvis.cogs:
-        session = getattr(cog, "_session", None)
-        if session:
-            session.close()
-    if restart_ctx:
-        return restart_ctx
+    jarvis.start(jconfig.token)