Added checkers when there is no rows in the DB

discord/unit/manager.py

@@ -117,7 +117,7 @@ class UnitManager(commands.Cog):
         """Gets all units"""
 
         data = await Api.get('/unit/all')
-        if data == None:
+        if(data==None or len(data['units'])==0):
             return await ctx.send('No units found')
         
         units_data = []

server/user/route.js

@@ -38,6 +38,9 @@ router.get('/units', async (context, next) => {
     checkUser(context);
     try{
         const data = await userModel.getUserUnits(context.session.user_id);
+        if(data.length===1  && data[0].id===null){
+            throw Error('No Units Found')
+        }
         context.response.body = {units: data};
         context.status = 200;
     }catch(error){