Export sumologic users

export_sumologic_users.py

import sys
import requests
import csv
from requests.auth import HTTPBasicAuth

# how to get keys --> https://api.sumologic.com/docs/#section/Getting-Started/Authentication
sumo_key = ''
sumo_secret = ''
sumo_region = 'us2'

#Script will need change to support more than 1000 users
print("Getting users")
response = requests.get(f'https://api.{sumo_region}.sumologic.com/api/v1/users?limit=1000', auth = HTTPBasicAuth(sumo_key, sumo_secret))

if response.status_code != 200:
    print(f"Failed to get user from sumologic reason --> {response.reason}")
    sys.exit(1)

with open('users.csv', mode='w') as csv_file:
    fieldnames = ['firstName', 'lastName', 'email', 'createdAt', 'createdBy', 'modifiedAt', 'modifiedBy', 'id', 'isActive', 'isLocked', 'isMfaEnabled', 'lastLoginTimestamp']
    writer = csv.DictWriter(csv_file, fieldnames=fieldnames)

    writer.writeheader()
    for user in response.json()['data']:
        user.pop('roleIds')
        writer.writerow(user)

print("Created CSV")