From b6cd16e6630e0f0ce908f594b8749b5f4eec0509 Mon Sep 17 00:00:00 2001
From: Terrtia <or1994@hotmail.fr>
Date: Thu, 12 Dec 2019 14:29:44 +0100
Subject: [PATCH] chg: [Domain] get all/by month domains up

---
 bin/lib/Domain.py | 48 +++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 48 insertions(+)

diff --git a/bin/lib/Domain.py b/bin/lib/Domain.py
index 17ffcbef..2ae6e9e3 100755
--- a/bin/lib/Domain.py
+++ b/bin/lib/Domain.py
@@ -30,6 +30,54 @@ config_loader = ConfigLoader.ConfigLoader()
 r_serv_onion = config_loader.get_redis_conn("ARDB_Onion")
 config_loader = None
 
+######## DOMAINS ########
+def get_all_domains_up(domain_type):
+    '''
+    Get all domain up (at least one time)
+
+    :param domain_type: domain type
+    :type domain_type: str
+
+    :return: list of domain
+    :rtype: list
+    '''
+    return list(r_serv_onion.smembers("full_onion_up"))
+
+def get_domains_up_by_month(date_year_month, domain_type, rlist=False):
+    '''
+    Get all domain up (at least one time)
+
+    :param domain_type: date_year_month YYYYMM
+    :type domain_type: str
+
+    :return: list of domain
+    :rtype: list
+    '''
+    res = r_serv_onion.smembers("month_onion_up:{}".format(date_year_month))
+    if rlist:
+        return list(res)
+    else:
+        return res
+
+def get_domain_up_by_day(date_year_month, domain_type, rlist=False):
+    '''
+    Get all domain up (at least one time)
+
+    :param domain_type: date YYYYMMDD
+    :type domain_type: str
+
+    :return: list of domain
+    :rtype: list
+    '''
+    res = r_serv_onion.smembers("onion_up:{}".format(date_year_month))
+    if rlist:
+        return list(res)
+    else:
+        return res
+
+
+######## DOMAIN ########
+
 def get_domain_type(domain):
     if str(domain).endswith('.onion'):
         return 'onion'