ansible-hardening/doc/source/generate_docs.py
Major Hayden 390ccd8473 Adding benefits faq + config docs
Implements: blueprint security-hardening

Change-Id: I72fd97ff1cdee96d87ad195828e723aad33934d1
2015-11-20 11:12:24 -06:00

97 lines
2.7 KiB
Python

#!/usr/bin/env python
# Copyright 2015, Rackspace US, Inc.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
"""Generates the documentation scaffolding."""
import csv
from jinja2 import Template
from textwrap import fill
def reindent(string_to_indent, numSpaces):
"""Indent strings with spaces."""
s = string_to_indent.splitlines()
s = [(numSpaces * ' ') + line.lstrip() for line in s]
return '\n'.join(s)
stigviewer_url = ("https://www.stigviewer.com/stig/red_hat_enterprise_linux_6/"
"2015-05-26/finding/{0}")
stig_note_template = """{{ title }}
{{ '-' * title | length }}
{{ desc }}
Details: `{{ id }} in STIG Viewer`_.
.. _{{ id }} in STIG Viewer: {{ stigviewer }}
Notes for deployers
~~~~~~~~~~~~~~~~~~~
.. include:: developer-notes/{{ id }}.rst
"""
stigs = []
with open('rhel6stig.csv', 'r') as csvfile:
reader = csv.reader(csvfile)
for row in reader:
metadata = {
'id': row[0],
'title': "{0}: {1}".format(row[0], row[2]),
'desc': fill(row[3], width=78),
'fixtext': row[7],
'checktext': row[9],
'severity': row[1],
'stigviewer': stigviewer_url.format(row[0]),
}
template = Template(stig_note_template)
with open("stig-notes/{0}.rst".format(metadata['id']), 'w') as rstfile:
rstfile.write(template.render(metadata))
stigs.append(metadata)
category_template = """.. include:: <xhtml1-lat1.txt>
`Home <index.html>`__ |raquo| Security hardening for openstack-ansible
Category {{ level }} ({{ name | capitalize }}) controls
================================
.. toctree::
:maxdepth: 1
"""
categories = {
'low': 1,
'medium': 2,
'high': 3,
}
for category_name, category_level in categories.items():
matching_stigs = [x for x in stigs if x['severity'] == category_name]
cat_file = open("controls-cat{0}.rst".format(category_level), 'w')
template = Template(category_template)
cat_file.write(template.render(name=category_name,
level=category_level))
include_template = ".. include:: stig-notes/{0}.rst\n\n"
for matching_stig in sorted(matching_stigs, key=lambda k: k['id']):
cat_file.write(include_template.format(matching_stig['id']))