dns-generator/namebench.py

from entry import entry

class namebench:

    def __init__(self):
        self.buffer = list()

    def generate_to_buffer(self, domain):
        self.buffer.append('{0} {1}'.format(domain.type, domain.domain))

    def write_buffer(self):
        with open('namebench.txt', 'w') as f:
            for line in self.buffer:
                f.write('{0}\n'.format(line))