Python script to import tasks in bulk

task_import.py

#!/usr/bin/env python3
# -*- coding: utf-8 -*-

"""
task_import.py - A script to ingest a mass data set and create tasks out of it.

Author: Matt Bromiley (@mbromileyDFIR)

To Do:
"""
# Future Imports
from __future__ import print_function
from __future__ import unicode_literals

# TheHive4Py Imports
from thehive4py.api import TheHiveApi
from thehive4py.models import CaseTask

# Server variables
hiveServer = '<thehive server>'
hiveKey = '<api_key>'

# Establish API
api = TheHiveApi(hiveServer, hiveKey)

with open(sys.argv[1]) as f:
    for line in f:
        j = json.loads(line)
        task = CaseTask(
          title = j['title_data'],
          description = j['description_data'],
          group = j['group_data']
        )        
    
    api.create_case_task(<case_id>, task)