Authenticator/tools/yaml2json.py

#!/usr/bin/env python3
"""
YAML database to JSON converter.
"""
from subprocess import call
import json
import tempfile
from os import path, remove
from glob import glob
from shutil import rmtree
from collections import OrderedDict

import yaml

GIT_CLONE_URI = "https://github.com/2factorauth/Authenticator.git"
TMP_FOLDER = path.join(tempfile.gettempdir(), "Authenticator")
DATA_DIR = path.join(TMP_FOLDER, "_data")
OUTPUT_DIR = path.join(path.dirname(
    path.realpath(__file__)), "../data/data.json")


print("Cloning the repository...")
if path.exists(TMP_FOLDER):
    rmtree(TMP_FOLDER)
call(["git", "clone", GIT_CLONE_URI, TMP_FOLDER])

if path.exists(OUTPUT_DIR):
    remove(OUTPUT_DIR)  


def is_valid(provider):
    if set(["tfa", "software"]).issubset(provider.keys()):
        return provider["tfa"] and provider["software"]
    else:
        return False


output = {}

for db_file in glob(DATA_DIR + "/*.yml"):
    with open(db_file, 'r') as file_data:
        try:
            providers = yaml.load(file_data)["websites"]
            for provider in providers:
                if is_valid(provider):
                    output[provider.get("name")] = path.splitext(
                        provider.get("img"))[0]
        except (yaml.YAMLError, TypeError, KeyError) as error:
            pass
output = OrderedDict(sorted(output.items(), key=lambda x: x[0].lower()))


with open(OUTPUT_DIR, 'w') as data:
    json.dump(output, data, indent=4)

rmtree(TMP_FOLDER)