KeyError при создании HTML-страницы - PullRequest
Новая
       23

KeyError при создании HTML-страницы

0 голосов
/ 19 апреля 2019

Я генерирую HTML-страницу из простого шаблона: 

<div id="dateReference">
    <h1 id="currentDate">Exploit automation completed on: {date_of_attack}</h1>
</div>
<div id="amountOfHost">
    <h4 id="hostCount">Total of {amount_of_hosts} separate hosts attacked</h4>
</div>
<div id="hostDropDown" class="dropdown-content">
    {all_hosts_attacked}
</div>
<div id="successfulAttacks">
    <h4 id="successAttacksBanner">Successful Attacks:</h4>
</div>
{exploit_table}

То, как я создаю теги, заключается в создании простой строки с использованием простого класса: 

import os
import datetime

import lib.settings


class HtmlPageGenerator(object):

    def __init__(self, successes, failures, host_count):
        self.success = successes
        self.fails = failures
        self.host_count = host_count
        self.html_template = lib.settings.HTML_PAGE_TEMPLATE
        self.attacked_hosts_list = open(lib.settings.HOST_FILE).readlines()

    def _generate_html_table(self, headers):
        retval = '<table id="generatedExploitTable"><tr>'
        for header in headers:
            retval += "<th>{}</th>".format(header)
        retval += "</tr><tr>"
        for value in self.success:
            retval += "<td>{}</td>".format(value)
        retval += "</tr></table>"
        return retval

    def _generate_drop_down_menu(self):
        retval = ""
        for host in self.attacked_hosts_list:
            retval += '<a href="#">{}</a>'.format(host.strip())
        return retval

    def generator(self):
        if not os.path.exists(lib.settings.HTML_PAGE_PATH):
            os.makedirs(lib.settings.HTML_PAGE_PATH)
        with open(self.html_template, 'r') as template, open(lib.settings.HTML_PAGE_GENERATION_FILE_PATH, 'a+') as out:
            to_format = template.read()
            out.write(
                to_format.format(
                    date_of_attack=str(datetime.datetime.today()).split(".")[0],
                    exploit_table=self._generate_html_table(["Exploit Paths"]),
                    amount_of_hosts=self.host_count,
                    all_hosts_attacked=self._generate_drop_down_menu()
                )
            )
        return lib.settings.HTML_PAGE_GENERATION_FILE_PATH

Функция _generate_html_table работает правильно и успешно сгенерирована, проблема в том, что когда я пытаюсь сгенерировать теги <a href="#", она выдает следующую ошибку: 

Traceback (most recent call last):
  File "xxxxx.py", line 10, in <module>
    23
  File "/Users/admin/bin/tools/xxxxx/lib/page_generator.py", line 42, in generator
    all_hosts_attacked=self._generate_drop_down_menu()
KeyError: '\n            document'

Что является причиной моей ошибки и как я могу исправить это успешно? Я попытался сгенерировать ссылки как list и вернуть вместо них присоединенное, но оно выдает тот же Exception. Любая помощь с этим была бы великолепна

1 Ответ

0 голосов
/ 19 апреля 2019

Понял, у меня есть javascript в HTML, поэтому, как мы все знаем, javascript использует скобки {} для их анализа: {{}} fixed.

...