status-react/test/appium/support/message_reliability_report.py

def write_data_to_file(sent: int, received: int):
    with open('messages.txt', 'a') as f:
        f.write("['sent': %s, 'received': %s]" % (sent, received))


def create_one_to_one_chat_report(one_to_one_chat_data: dict):
    import matplotlib
    matplotlib.use('Agg')
    import matplotlib.pyplot as plt

    user_a = one_to_one_chat_data['user_a']
    user_b = one_to_one_chat_data['user_b']

    fig, ax = plt.subplots(nrows=1, ncols=1, figsize=(15, 7))
    time_1 = sorted(user_a['message_time'])
    ax.plot([i / 60 for i in time_1], [user_a['message_time'][i] for i in time_1],
            'o-', color='#0c0fea', label='user_a')
    time_2 = sorted(user_b['message_time'])
    ax.plot([i / 60 for i in time_2], [user_b['message_time'][i] for i in time_2],
            'o-', color='#f61e06', label='user_b')

    sent_messages = user_a['sent_messages'] + user_b['sent_messages']
    received_messages = len(user_a['message_time']) + len(user_b['message_time'])

    title = "User A: sent messages: {}, received messages: {}" \
            "\nUser B: sent messages: {}, received messages: {}".format(user_a['sent_messages'],
                                                                        len(user_a['message_time']),
                                                                        user_b['sent_messages'],
                                                                        len(user_b['message_time']))
    if sent_messages:
        title += "\nReceived messages: {}%".format(
            round((len(user_a['message_time']) + len(user_b['message_time'])) / sent_messages * 100, ndigits=2))

    plt.title(title)
    plt.xlabel('chat session duration, minutes')
    plt.ylabel('time to receive a message, seconds')
    plt.legend()
    fig.savefig('chart.png')

    write_data_to_file(sent=sent_messages, received=received_messages)


def create_public_chat_report(public_chat_data: dict):
    import matplotlib
    matplotlib.use('Agg')
    import matplotlib.pyplot as plt

    sent_messages = public_chat_data['sent_messages']
    message_time = public_chat_data['message_time']

    fig, ax = plt.subplots(nrows=1, ncols=1, figsize=(15, 7))
    sorted_time = sorted(message_time)
    ax.plot([i / 60 for i in sorted_time], [message_time[i] for i in sorted_time], 'o-', color='#0c0fea')
    title = "Sent messages: {}\nReceived messages: {}".format(sent_messages, len(message_time))
    plt.title(title)
    plt.xlabel('chat session duration, minutes')
    plt.ylabel('time to receive a message, seconds')
    plt.legend()
    fig.savefig('chart.png')

    write_data_to_file(sent=sent_messages, received=len(message_time))