tasks_producer.py

import random
from faker.providers import BaseProvider
from faker import Faker
import config
import time
import requests
import json
import uuid

#Define a TaskProvider
class TaskProvider(BaseProvider):
    def task_priority(self):
        severity_levels = [
            'Low'
           ,'Moderate'
           ,'Major'
           ,'Critical'
        ]
        return severity_levels[random.randint(0,len(severity_levels)-1)]

#Create a Faker instance and seeding to have the same results every time we execute the script
#Return data in English
fakeTasks = Faker('en_US')
#Seed the Faker instance to have the same results every time we run the program
fakeTasks.seed_instance(0)

#Assign the TaskProvider to the Faker instance
fakeTasks.add_provider(TaskProvider)

#Generate A Fake Task
def produce_task(batchid,taskid):
    #Message composition
    message = {
        'batchid': batchid
       ,'id': taskid
       ,'owner': fakeTasks.unique.name()
       ,'priority': fakeTasks.task_priority()
      #,'raised_date':fakeTasks.date_time_this_year()
      #,'description':fakeTasks.text()
    }
    return message

def send_webhook(msg):
    """
    Send a webhook to a specified URL
    :param msg: task details
    :return:
    """
    try:
      # Post a webhook message
      # default is a function applied to objects that are not serializable = it converts them to str
      resp = requests.post(config.WEBHOOK_RECEIVER_URL, data=json.dumps(msg, sort_keys=True, default=str), headers={'Content-Type': 'application/json'}, timeout = 1.0)
      # Returns an HTTPError if an error has occurred during the process (used for debugging).
      resp.raise_for_status()
    except requests.exceptions.HTTPError as err:
           #print("An HTTP Error occurred",repr(err))
           pass
    except requests.exceptions.ConnectionError as err:
           #print("An Error Connecting to the API occurred", repr(err))
           pass
    except requests.exceptions.Timeout as err:
           #print("A Timeout Error occurred", repr(err))
           pass
    except requests.exceptions.RequestException as err:
           #print("An Unknown Error occurred", repr(err))
           pass
    except:
        pass
    else:
        return resp.status_code

#Generate A Bunch Of Fake Tasks
def produce_bunch_tasks():
    """
    Generate a Bunch of Fake Tasks
    """
    n = random.randint(config.MIN_NBR_TASKS, config.MAX_NBR_TASKS)
    batchid = str(uuid.uuid4())
    for i in range(n):
       msg = produce_task(batchid,i)
       resp = send_webhook(msg)
       time.sleep(config.WAIT_TIME)
       print(i, "out of ", n , " -- Status", resp, " -- Message = ", msg)
       yield resp, n , msg

if __name__ == "__main__":
   for resp, total , msg in produce_bunch_tasks():
       pass