mirror of
https://github.com/valitydev/atomic-threat-coverage.git
synced 2024-11-06 09:35:21 +00:00
75 lines
2.5 KiB
Python
Executable File
75 lines
2.5 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
from atcutils import ATCutils
|
|
from requests.auth import HTTPBasicAuth
|
|
import getpass
|
|
|
|
|
|
def main(c_auth=None):
|
|
|
|
try:
|
|
ATCconfig = ATCutils.load_config("config.yml")
|
|
confluence_space_name = ATCconfig.get('confluence_space_name')
|
|
confluence_space_home_page_name = ATCconfig.get(
|
|
'confluence_space_home_page_name')
|
|
confluence_rest_api_url = ATCconfig.get('confluence_rest_api_url')
|
|
confluence_name_of_root_directory = ATCconfig.get(
|
|
'confluence_name_of_root_directory')
|
|
|
|
except Exception as e:
|
|
raise e
|
|
pass
|
|
|
|
if not c_auth:
|
|
mail = input("Login: ")
|
|
password = getpass.getpass(prompt='Password: ', stream=None)
|
|
auth = HTTPBasicAuth(mail, password)
|
|
else:
|
|
auth = c_auth
|
|
|
|
url = confluence_rest_api_url
|
|
content = ""
|
|
|
|
print("Creating ATC page..")
|
|
# print(str(ATCutils.confluence_get_page_id(url,
|
|
# auth, confluence_space_name, confluence_space_home_page_name)))
|
|
data = {
|
|
"title": confluence_name_of_root_directory,
|
|
"spacekey": confluence_space_name,
|
|
"parentid": str(ATCutils.confluence_get_page_id(
|
|
url, auth, confluence_space_name,
|
|
confluence_space_home_page_name)),
|
|
"confluencecontent": content,
|
|
}
|
|
|
|
# print(push_to_confluence(data, url, auth))
|
|
if not ATCutils.push_to_confluence(data, url, auth):
|
|
raise Exception("Could not create or update the page. " +
|
|
"Is the parent name correct?")
|
|
|
|
spaces = ["Detection Rules", "Logging Policies",
|
|
"Data Needed", "Triggers", "Response Actions",
|
|
"Response Playbooks", "Enrichments", "Customers",
|
|
"Mitigation Systems", "Mitigation Policies",
|
|
"Hardening Policies"]
|
|
|
|
for space in spaces:
|
|
print("Creating %s.." % space)
|
|
data = {
|
|
"title": space,
|
|
"spacekey": confluence_space_name,
|
|
"parentid": str(ATCutils.confluence_get_page_id(
|
|
url, auth, confluence_space_name,
|
|
confluence_name_of_root_directory)),
|
|
"confluencecontent": content,
|
|
}
|
|
# print(push_to_confluence(data, url, auth))
|
|
if not ATCutils.push_to_confluence(data, url, auth):
|
|
raise Exception("Could not create or update the page. " +
|
|
"Is the parent name correct?")
|
|
print("Done!")
|
|
return True
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|