chg: [ail_stats] get all subtypes objects nb

Author: Terrtia

bin/lib/ail_stats.py

@@ -15,22 +15,10 @@
 ##################################
 from lib.ConfigLoader import ConfigLoader
 from lib.objects import ail_objects
-from lib.objects import BarCodes
-from lib.objects import CookiesNames
-from lib.objects import CryptoCurrencies
-from lib.objects import Domains
-from lib.objects import Favicons
-from lib.objects import FilesNames
-from lib.objects import GTrackers
-from lib.objects import Mails
-from lib.objects import Pgps
-from lib.objects import QrCodes
-from lib.objects import Titles
-from lib.objects import Usernames
 from lib.crawlers import get_crawlers_stats
+from lib import ail_core
 from lib import ail_orgs
 from lib import ail_users
-from lib import chats_viewer
 from lib import Tag
 from lib import Tracker
 
@@ -133,25 +121,13 @@ def get_tracked_objs_dashboard(user_org, user_id):
     return trackers
 
 
-def get_global_stats():  # decoded ??  domhash, hhhash  etag ???
+def get_global_stats():
     stats = {'orgs': ail_orgs.get_nb_orgs(),
              'users': ail_users.get_nb_users(),
-             'objs':
-                 {'barcode': BarCodes.Barcodes().get_nb(),
-                  'chat': chats_viewer.get_nb_chats_stats(),
-                  'cookie-name': CookiesNames.CookiesNames().get_nb(),
-                  'cryptocurrency': CryptoCurrencies.CryptoCurrencies().get_nb(),
-                  'domain': {'onion': Domains.get_nb_domains_up_by_type('onion'),
-                             'web': Domains.get_nb_domains_up_by_type('web')
-                             },
-                  'favicon': Favicons.Favicons().get_nb(),
-                  'file-name': FilesNames.FilesNames().get_nb(),
-                  'gtracker': GTrackers.GTrackers().get_nb(),
-                  'mail': Mails.Mails().get_nb(),
-                  'pgp': Pgps.Pgps().get_nb(),
-                  'qrcode': QrCodes.Qrcodes().get_nb(),
-                  'title': Titles.Titles().get_nb(),
-                  'username': Usernames.Usernames().get_nb(),
-                  },
+             'objs': {},
              }
+    for obj_type in ail_core.get_all_objects():
+        nb = ail_objects.get_nb_objects(obj_type)
+        if nb:
+            stats['objs'][obj_type] = nb
     return stats

bin/lib/objects/Domains.py

@@ -844,6 +844,12 @@ def get_nb_by_date(self, date):
             nb += r_crawler.scard(f'{domain_type}_up:{date}')
         return nb
 
+    def get_nb(self):
+        nb = {}
+        for domain_type in get_all_domains_types():
+            nb[domain_type] = get_nb_domains_up_by_type(domain_type)
+        return nb
+
 #### API ####
 
 def api_get_onions_by_month(date_year_month):

bin/lib/objects/ail_objects.py

@@ -201,6 +201,15 @@ def get_nb_objects_dashboard(date, flask_context=True):
             objs[obj_type]['link'] = objs_class.get_link(flask_context=flask_context)
     return objs
 
+def get_nb_objects(obj_type):
+    if OBJECTS_CLASS[obj_type].get('objs'):
+        return OBJECTS_CLASS[obj_type]['objs']().get_nb()
+    elif obj_type == 'chat':
+        return chats_viewer.get_nb_chats_stats()
+    # 'item'
+    # 'screenshot'
+    # 'message'
+    return None
 
 def get_objects(objects):
     objs = set()