from datetime import datetime
import sys
from dateutil.tz import UTC
from buyercall.integrations.elasticsearch.utilities import insert_activity_data, insert_error_data

# Insert data to error log

error_data = {
    'environment': 'DEVELOPMENT',
    'hostname': 'buyercall.com',
    'ip': '122.223.225.2',
    'created': datetime.now(tz=UTC),
    'method': 'GET',
    'is_api': False,
    'params': {
        'key': 'name',
        'value': 'Jay'        
    },
    'url': 'https://buyercall.com/elasticsearch/',
    'status_code': 200,
    'status_code_root': 200,
    'error_detail': 'No errors',
    'user_id': 'fr2ra-55sfro-85sd1f4-qgv44f',
    'partnership_id': 'Ogg4a-55sfro-85sd1f4-qgv44o',
    'account_id': 'Gs9ra-5n8sro-85sd1f4-qgv447'
}

activity_data = {
    'user_id': 'fr2ra-55sfro-85sd1f4-qgv44f',
    'partnership_id': 'Ogg4a-55sfro-85sd1f4-qgv44o',
    'account_id': 'Gs9ra-5n8sro-85sd1f4-qgv447',
    'message': 'This is a sample log data',
    'created': datetime.now(tz=UTC),
    'activity_type': 'chat',
    'hyperlink': 'https://buyercall.com/chat/dfg55-5ryjfgj-kljioJ5-4s5d64g',
    'is_read': True
}

if __name__ == '__main__':

    if sys.argv[1] == 'error':
        resp = insert_error_data(error_data)
        print(resp)
    elif sys.argv[1] == 'activity':
        resp = insert_activity_data(activity_data)
        print(resp)
    else:
        print('Options : error, activity')