Adding new rangue-frequency based in CRON Dates

joseramon.afonso · joseramon.afonso · commit c3b7e93168ce · 2020-04-15T15:03:07.000+02:00
diff --git a/devoutils/faker/senders/base_fake_generator.py b/devoutils/faker/senders/base_fake_generator.py
@@ -3,6 +3,8 @@
 import random
 import threading
 import time
+from datetime import datetime, timedelta
+from pycron import is_now
 from .template_parser import TemplateParser
 
 
@@ -17,8 +19,20 @@ def __init__(self, engine, template, **kwargs):
         threading.Thread.__init__(self)
         self.engine = engine
         self.template = str(template)
-        self.prob = kwargs.get('prob', 100)
-        self.freq = kwargs.get('freq', (1, 1))
+
+        if kwargs.get('cron_prob', None) is not None:
+            self.cron_prob = kwargs.get("cron_prob")
+            self.set_last_cron_probability()
+
+            self.cron_child = threading.Thread(
+                target=self.update_cron_probability
+            )
+            self.cron_child.setDaemon(True)
+            self.cron_child.start()
+        else:
+            self.prob = kwargs.get('prob', 100)
+            self.freq = kwargs.get('freq', (1, 1))
+
         self.date_format = kwargs.get('date_format', "%Y-%m-%d %H:%M:%S.%f")
         self.interactive = kwargs.get('interactive', False)
         self.simulation = kwargs.get('simulation', False)
@@ -44,6 +58,29 @@ def wait(self):
             secs = random.uniform(self.freq[0], self.freq[1])
         time.sleep(secs)
 
+    def set_last_cron_probability(self):
+        now = datetime.now()
+        try:
+            for _ in range(0, 525600):
+                for item in self.cron_prob:
+                    if is_now(item["cron"], now):
+                        self.prob = item['prob']
+                        self.freq = item['freq']
+                        raise Exception('found')
+                now = now - timedelta(minutes=1)
+        except Exception as inst:
+            if inst.args[0] != "found":
+                raise Exception(inst)
+
+    def update_cron_probability(self):
+        while True:
+            for item in self.cron_prob:
+                if is_now(item["cron"]):
+                    self.prob = item['prob']
+                    self.freq = item['freq']
+                    break
+            time.sleep(60)
+
     def probability(self):
         """Calculate probability"""
         k = random.randint(0, 100)
diff --git a/requirements.txt b/requirements.txt
@@ -6,3 +6,5 @@ Jinja2==2.11.1
 psutil==5.7.0
 python-dateutil==2.8.1
 Faker==4.0.2
+pycron==3.0.0
+
diff --git a/setup.py b/setup.py
@@ -31,7 +31,8 @@
                     'Jinja2==2.11.1',
                     'psutil==5.7.0',
                     'python-dateutil==2.8.1',
-                    'Faker==4.0.2']
+                    'Faker==4.0.2',
+                    'pycron==3.0.0']
 CLI = ['devo-faker=devoutils.faker.scripts.faker_cli:cli']
 
 
