version.py

"""Generates a configuration file containing the version number."""
from __future__ import absolute_import
import os
from retriever import VERSION, MODULE_LIST


def get_module_version():
    """This function gets the version number of the scripts and returns them in array form."""
    modules = MODULE_LIST()
    scripts = []
    for module in modules:
        if module.SCRIPT.public:
            if os.path.isfile('.'.join(module.__file__.split('.')[:-1]) + '.json') and module.SCRIPT.version:
                module_name = module.__name__ + '.json'
                scripts.append(','.join([module_name, str(module.SCRIPT.version)]))
            elif os.path.isfile('.'.join(module.__file__.split('.')[:-1]) + '.py') and \
                    not os.path.isfile('.'.join(module.__file__.split('.')[:-1]) + '.json'):
                module_name = module.__name__ + '.py'
                scripts.append(','.join([module_name, str(module.SCRIPT.version)]))

    scripts = sorted(scripts, key = str.lower)
    return scripts

scripts = get_module_version()

if os.path.isfile("version.txt"):
    os.remove("version.txt")

with open("version.txt", "w") as version_file:
    version_file.write(VERSION)
    for script in scripts:
        version_file.write('\n' + script)