Added some prechecks

Author: karel26

extensions/realweather/extension.py

@@ -195,17 +195,16 @@ async def generate_config(self, filename: str, tmpname: str, config: dict | None
     async def run_realweather(self, filename: str, tmpname: str) -> tuple[str, bool]:
         try:
             cwd = await self.server.get_missions_dir()
-            rw_home = os.path.expandvars(self.config['installation'])
 
-            def cleanup(cwd: str):
+            def cleanup():
                 # delete the mission_unpacked directory which might still be there from former RW runs
                 mission_unpacked_dir = os.path.join(cwd, 'mission_unpacked')
                 if os.path.exists(mission_unpacked_dir):
                     utils.safe_rmtree(mission_unpacked_dir)
 
             def run_subprocess():
                 # double-check that no mission_unpacked dir is there
-                cleanup(cwd)
+                cleanup()
                 # run RW
                 process = subprocess.Popen(
                     [self.get_rw_exe()],
@@ -240,7 +239,7 @@ def run_subprocess():
                 try:
                     await asyncio.to_thread(run_subprocess)
                 finally:
-                    cleanup(cwd)
+                    cleanup()
 
             # check if DCS Real Weather corrupted the miz file
             await asyncio.to_thread(MizFile, tmpname)

plugins/admin/commands.py

@@ -1415,7 +1415,7 @@ async def cleanup(self):
                 await conn.execute("DELETE FROM nodestats WHERE time < (CURRENT_TIMESTAMP - interval '1 month')")
 
     @cleanup.before_loop
-    async def before_check_unban(self):
+    async def before_cleanup(self):
         await self.bot.wait_until_ready()
 
 

plugins/mission/commands.py

@@ -2001,7 +2001,7 @@ async def expire_token(self):
                 """)
 
     @expire_token.before_loop
-    async def before_check_unban(self):
+    async def before_expire(self):
         await self.bot.wait_until_ready()
 
     @tasks.loop(minutes=1.0)

plugins/restapi/commands.py

@@ -937,6 +937,10 @@ async def refresh_views(self):
                     REFRESH MATERIALIZED VIEW mv_serverstats;
                 """)
 
+    @refresh_views.before_loop
+    async def before_refresh_views(self):
+        await self.bot.wait_until_ready()
+
 
 async def setup(bot: DCSServerBot):
     await bot.add_cog(RestAPI(bot))

plugins/userstats/commands.py

@@ -838,6 +838,10 @@ async def refresh_views(self):
                     REFRESH MATERIALIZED VIEW mv_statistics;
                 """)
 
+    @refresh_views.before_loop
+    async def before_refresh_views(self):
+        await self.bot.wait_until_ready()
+
     @commands.Cog.listener()
     async def on_member_remove(self, member):
         if self.get_config().get('wipe_stats_on_leave', True):