import os
import json

def get_config_path():
    user_dir = os.path.expanduser('~')
    program_name = 'connect'
    config_filename = 'config.json'

    config_path = os.path.join(user_dir, program_name, config_filename)

    return config_path


def mkdir(path):
    parent_dir = os.path.dirname(path)
    if not os.path.exists(parent_dir):
        mkdir(parent_dir)
    if not os.path.exists(path):
        os.mkdir(path)


def init_config():
    config_path = get_config_path()
    config_dirname = os.path.dirname(config_path)
    mkdir(config_dirname)

    cur_config = {}
    if os.path.exists(config_path):
        cur_config = json.load(open(config_path))

    connect_name = cur_config.get('connect_name', '')
    username = cur_config.get('username', '')
    password = cur_config.get('password', '')

    connect_name = input('请输入连接名称(默认为{})'.format(connect_name)) or connect_name
    username = input('请输入用户名(默认为{})'.format(username)) or username
    password = input('请输入密码(默认为{})'.format(password)) or password

    config = {
        'connect': connect_name,
        'username': username,
        'password': password
    }

    json.dump(config, open(config_path, 'w'))


if __name__ == '__main__':
    init_config()