mirror of https://github.com/MISP/PyMISP
54 lines
2.0 KiB
Python
Executable File
54 lines
2.0 KiB
Python
Executable File
#!/usr/bin/env python
|
|
# -*- coding: utf-8 -*-
|
|
|
|
from pymisp import PyMISP
|
|
from pymisp import MISPObject
|
|
from pymisp.tools import update_objects
|
|
from keys import misp_url, misp_key, misp_verifycert
|
|
import argparse
|
|
import requests
|
|
import sys
|
|
|
|
|
|
"""
|
|
|
|
usage: add_github_user.py [-h] -e EVENT [-f] -u USERNAME
|
|
|
|
Fetch GitHub user details and add it in object in MISP
|
|
|
|
optional arguments:
|
|
-h, --help show this help message and exit
|
|
-e EVENT, --event EVENT
|
|
Event ID to update
|
|
-f, --force-template-update
|
|
-u USERNAME, --username USERNAME
|
|
GitHub username to add
|
|
"""
|
|
|
|
if __name__ == '__main__':
|
|
parser = argparse.ArgumentParser(description='Fetch GitHub user details and add it in object in MISP')
|
|
parser.add_argument("-e", "--event", required=True, help="Event ID to update")
|
|
parser.add_argument("-f", "--force-template-update", required=False, action="store_true")
|
|
parser.add_argument("-u", "--username", required=True, help="GitHub username to add")
|
|
args = parser.parse_args()
|
|
|
|
r = requests.get("https://api.github.com/users/{}".format(args.username))
|
|
if r.status_code != 200:
|
|
sys.exit("HTTP return is {} and not 200 as expected".format(r.status_code))
|
|
if args.force_template_update:
|
|
print("Updating MISP Object templates...")
|
|
update_objects()
|
|
pymisp = PyMISP(misp_url, misp_key, misp_verifycert)
|
|
|
|
misp_object = MISPObject(name="github-user")
|
|
github_user = r.json()
|
|
rfollowers = requests.get(github_user['followers_url'])
|
|
followers = rfollowers.json()
|
|
user_followers = []
|
|
misp_object.add_attributes("follower", *[follower['login'] for follower in followers])
|
|
misp_object.add_attribute('bio', github_user['bio'])
|
|
misp_object.add_attribute('link', github_user['html_url'])
|
|
misp_object.add_attribute('user-fullname', github_user['name'])
|
|
misp_object.add_attribute('username', github_user['login'])
|
|
retcode = pymisp.add_object(args.event, misp_object)
|