submission_mapper.py

# pylint: disable=line-too-long
# pylint: disable=too-many-locals
# pylint: disable=no-member
"""
Map submission data from CHEFS the pre-defined submision dictionaries by sites
"""
import os
import logging
import helper

NOTICE_STANDARD_TIME = os.getenv('NOTICE_STANDARD_TIME')
LOGLEVEL = os.getenv('LOGLEVEL')

logging.basicConfig(level=LOGLEVEL, format='%(asctime)s [%(levelname)s] %(message)s')

def map_sites(submissions_json, hvs_json, current_date):
    """Map submission data from CHEFS the pre-defined submision dictionaries by sites"""
    source_sites = []
    src_reg_dist_dic = {}
    receiving_sites = []
    rcv_reg_dist_dic = {}
    hv_sites = []
    hv_reg_dist_dic = {}
    for submission in submissions_json:
        logging.debug('Mapping submission data to the source site...')
        _src_dic = helper.map_source_site(submission)
        if _src_dic:
            source_sites.append(_src_dic)
            _diff_src = helper.get_difference_datetimes_in_hour(current_date, _src_dic['createAt'])
            if (_diff_src is not None and _diff_src <= int(NOTICE_STANDARD_TIME)):  #within the last this hours.
                helper.add_regional_district_dic(_src_dic, src_reg_dist_dic)

        logging.debug('Mapping submission data to the receiving site...')
        _1rcv_dic = helper.map_rcv_site(submission, 1)
        if _1rcv_dic:
            receiving_sites.append(_1rcv_dic)
            _diff_1rcv = helper.get_difference_datetimes_in_hour(current_date, _1rcv_dic['createAt'])
            if (_diff_1rcv is not None and _diff_1rcv <= int(NOTICE_STANDARD_TIME)):
                helper.add_regional_district_dic(_1rcv_dic, rcv_reg_dist_dic)

        _2rcv_dic = helper.map_rcv_site(submission, 2)
        if _2rcv_dic:
            receiving_sites.append(_2rcv_dic)
            _diff_2rcv = helper.get_difference_datetimes_in_hour(current_date, _2rcv_dic['createAt'])
            if (_diff_2rcv is not None and _diff_2rcv <= int(NOTICE_STANDARD_TIME)):
                helper.add_regional_district_dic(_2rcv_dic, rcv_reg_dist_dic)

        _3rcv_dic = helper.map_rcv_site(submission, 3)
        if _3rcv_dic:
            receiving_sites.append(_3rcv_dic)
            _diff_3rcv = helper.get_difference_datetimes_in_hour(current_date, _3rcv_dic['createAt'])
            if (_diff_3rcv is not None and _diff_3rcv <= int(NOTICE_STANDARD_TIME)):
                helper.add_regional_district_dic(_3rcv_dic, rcv_reg_dist_dic)

    logging.debug('Mapping additional work between source site and receiving sites...')
    source_sites, receiving_sites = helper.map_source_receiving_site(source_sites, receiving_sites)

    logging.info('Creating high volume site records...')
    for hvs in hvs_json:
        logging.debug('Mapping hv data to the hv site...')
        _hv_dic = helper.map_hv_site(hvs)
        if _hv_dic:
            hv_sites.append(_hv_dic)
            _diff_hv = helper.get_difference_datetimes_in_hour(current_date, _hv_dic['createAt'])
            if (_diff_hv is not None and _diff_hv <= int(NOTICE_STANDARD_TIME)):
                helper.add_regional_district_dic(_hv_dic, hv_reg_dist_dic)

    return source_sites, src_reg_dist_dic, receiving_sites, rcv_reg_dist_dic, hv_sites, hv_reg_dist_dic