diff --git a/prismedia/genconfig.py b/prismedia/genconfig.py index 03273ac..d3a29c7 100644 --- a/prismedia/genconfig.py +++ b/prismedia/genconfig.py @@ -6,19 +6,18 @@ logger = logging.getLogger('Prismedia') from . import utils + def genconfig(): path = join(dirname(__file__), 'config') files = [f for f in listdir(path) if isfile(join(path, f))] for f in files: final_f = f.replace(".sample", "") - overwrite = True - if exists(final_f): - overwrite = utils.ask_overwrite(final_f + " already exists. Do you want to overwrite it?") + if exists(final_f) and not utils.ask_overwrite(final_f + " already exists. Do you want to overwrite it?"): + continue - if overwrite: - copyfile(join(path, f), final_f) - logger.info(str(final_f) + " correctly generated, you may now edit it to fill your credentials.") + copyfile(join(path, f), final_f) + logger.info(str(final_f) + " correctly generated, you may now edit it to fill your credentials.") if __name__ == '__main__':