Source code for vivainsights.mt_data

# --------------------------------------------------------------------------------------------
# Copyright (c) Microsoft Corporation. All rights reserved.
# Licensed under the MIT License. See LICENSE.txt in the project root for license information.
# --------------------------------------------------------------------------------------------
"""
This module returns a data frame containing a meeting query.
"""

import pkg_resources
import pandas as pd


[docs]
def load_mt_data():
    # This is a stream-like object. If you want the actual info, call
    # stream.read()
    # stream = pkg_resources.resource_string('vivainsights', 'data/mt_data.csv')
    if(pkg_resources.resource_exists(__name__, 'data/mt_data.csv')):
        stream = pkg_resources.resource_stream(__name__, 'data/mt_data.csv')
    elif(pkg_resources.resource_exists(__name__, '../data/mt_data.csv')):
        stream = pkg_resources.resource_stream(__name__, '../data/mt_data.csv')
    else:
        print('Error: please report issue to repo maintainer')    
    return pd.read_csv(stream, encoding='utf-8')