Ver código fonte

Added configuration to disable services

Penta 5 anos atrás
pai
commit
dee5f76ff8
3 arquivos alterados com 15 adições e 4 exclusões
  1. 7 2
      myanimebot.py
  2. 6 2
      myanimebot/discord.py
  3. 2 0
      myanimebot/globals.py

+ 7 - 2
myanimebot.py

@@ -36,8 +36,13 @@ if not sys.version_info[:2] >= (3, 7):
 
 def exit_app():
 	logging.info("Closing all tasks...")
-	globals.task_feed.cancel()
-	globals.task_feed_anilist.cancel()
+	
+	if globals.MAL_ENABLED:
+		globals.task_feed.cancel()
+
+	if globals.ANI_ENABLED:
+		globals.task_feed_anilist.cancel()
+
 	globals.task_thumbnail.cancel()
 	globals.task_gameplayed.cancel()
 

+ 6 - 2
myanimebot/discord.py

@@ -26,8 +26,12 @@ class MyAnimeBot(discord.Client):
 
 		globals.logger.info("Starting all tasks...")
 
-		globals.task_feed = globals.client.loop.create_task(background_check_feed(globals.client.loop))
-		globals.task_feed_anilist = globals.client.loop.create_task(anilist.background_check_feed(globals.client.loop))
+		if globals.MAL_ENABLED:
+			globals.task_feed = globals.client.loop.create_task(background_check_feed(globals.client.loop))
+
+		if globals.ANI_ENABLED:
+			globals.task_feed_anilist = globals.client.loop.create_task(anilist.background_check_feed(globals.client.loop))
+
 		globals.task_thumbnail = globals.client.loop.create_task(update_thumbnail_catalog(globals.client.loop))
 		globals.task_gameplayed = globals.client.loop.create_task(change_gameplayed(globals.client.loop))
 

+ 2 - 0
myanimebot/globals.py

@@ -65,6 +65,8 @@ MAL_URL="https://myanimelist.net/"
 MAL_PROFILE_URL="https://myanimelist.net/profile/"
 ANILIST_PROFILE_URL="https://anilist.co/user/"
 DB_USER_NAME="mal_user" # Column's name for usernames in the t_users table
+MAL_ENABLED=CONFIG.getboolean("mal_enabled", True)
+ANI_ENABLED=CONFIG.getboolean("ani_enabled", True)
 
 # class that send logs to DB
 class LogDBHandler(logging.Handler):