import os
import commons
import random

install_phase = commons.POST_INSTALL
enabled = True

def execute(name, config, root):
    hostname = config['hostname']

    hostname_file = os.path.join(root, 'etc/hostname')
    hosts_file    = os.path.join(root, 'etc/hosts')

    with open(hostname_file, 'wb') as outfile:
        outfile.write(hostname)

    commons.replace_string_in_file(hosts_file, r'127\.0\.0\.1\s+localhost\s*\Z', '127.0.0.1\tlocalhost\n127.0.0.1\t' + hostname)