From 1f20b6c58358eb893f4e9be446bb7233674213ab Mon Sep 17 00:00:00 2001
From: Rafael Vargas <rafael.s.vargas18@gmail.com>
Date: Tue, 28 Dec 2021 20:00:45 -0400
Subject: [PATCH] Upgrading the stability of the warframe api

---
 vulkanbot/commands/Warframe.py | 45 +++++++++++++++++-----------------
 1 file changed, 23 insertions(+), 22 deletions(-)

diff --git a/vulkanbot/commands/Warframe.py b/vulkanbot/commands/Warframe.py
index dca4bb7..129f0ae 100644
--- a/vulkanbot/commands/Warframe.py
+++ b/vulkanbot/commands/Warframe.py
@@ -1,7 +1,6 @@
 import requests
 import json
 import discord
-from dotenv import dotenv_values
 from discord.ext import commands
 from config import config
 
@@ -21,30 +20,32 @@ class Warframe(commands.Cog):
         self.__bot = newBot
 
     @commands.command(name='cetus', help='Informa o tempo atual de Cetus - Warframe')
-    async def get_cetus(self, ctx):
-        try:
-            response = requests.get(config.CETUS_API)
-            data = json.loads(response.content)
-            short = data['shortString']
+    async def cetus(self, ctx):
+        description = await self.__get_api()
+        embed = discord.Embed(
+            title='Warframe Cetus Timing',
+            description=description,
+            colour=config.COLOURS['blue']
+        )
+        await ctx.send(embed=embed)
 
-            responseText = f'{short}'
+    async def __get_api(self):
+        """Return the information of the Warframe API"""
+        tries = 0
+        while True:
+            tries += 1
+            if tries > config.MAX_API_CETUS_TRIES:
+                return 'Os DE baiano não tão com o banco de dados ligado'
 
-            embed = discord.Embed(
-                title='Warframe Cetus Timing',
-                description=responseText,
-                colour=0xFF0000
-            )
-            await ctx.send(embed=embed)
+            try:
+                response = requests.get(config.CETUS_API)
+                data = json.loads(response.content)
+                short = data['shortString']
 
-        except Exception as e:
-            print(e)
-            responseText = f'Houve um erro inesperado :/'
-            embed = discord.Embed(
-                title='Warframe Cetus Timing',
-                description=responseText,
-                colour=0xFF0000
-            )
-            await ctx.send(embed=embed)
+                return short
+
+            except Exception as e:
+                continue
 
 
 def setup(bot):