api_Onedrive.py

# -*- coding: utf-8 -*-
"""
Created on Tue Mar 24 15:25:23 2020

@author: ceshuca
"""

import os
import json

def get_onedrive_location(account_type='business'):
    """
    Returns a string of the filepath of the Onedrive for this user

    :param account_type: str, 'business' or 'personal'
    """
    info_path = _get_onedrive_info_path()
    info_dict = _get_dictionary_from_path_to_json(info_path)
    return _get_onedrive_path_from_dictionary(info_dict, account_type)

def _get_onedrive_info_path():
    """
    Returns filepath of Onedrive file info.json
    """
    #path = _create_onedrive_info_path('APPDATA')
    #if path:
    #    return path
    return _create_onedrive_info_path('LOCALAPPDATA')

def _create_onedrive_info_path(appdata_str):
    r"""
    Looks up the environment variable given by appdata_str and combines with \Onedrive\info.json

    Then checks if the info.json exists at that path, and if so returns the filepath, otherwise
    returns False
    """
    path = os.path.join(os.environ[appdata_str], r'Onedrive')
    if os.path.exists(path):
        return path
    return False

def _get_dictionary_from_path_to_json(info_path):
    """
    Loads a json file and returns as a dictionary
    """
    with open(info_path, 'r') as f:
        text = f.read()

    return json.loads(text)

def _get_onedrive_path_from_dictionary(info_dict, account_type):
    """
    Returns the 'path' value under the account_type dictionary within the main dictionary
    """
    return info_dict[account_type]['path']



test = _get_onedrive_info_path()#get_onedrive_location(account_type = 'personal')
print(test)