chefs_soils.py

# pylint: disable=line-too-long
# pylint: disable=no-member
# pylint: disable=unbalanced-tuple-unpacking
# pylint: disable=too-many-function-args
"""
Retrieve CHEFS form data, overwrite it into AGOL CSVs and Layers,
send notification email to subscribers who want to get information for the soil relocation
"""
import os
import logging
import datetime
import pytz
import submission_loader
import submission_mapper
import csv_writer
import agol_updater
import email_sender
import helper

LOGLEVEL = os.getenv('LOGLEVEL')
AGOL_UPDATE_FLAG= helper.get_boolean_env_var('AGOL_UPDATE_FLAG') # the flag to control whether to update AGOL or not
EMAIL_NOTIFY_FLAG = helper.get_boolean_env_var('EMAIL_NOTIFY_FLAG',) # the flag to control whether to send email notification or not

helper.load_env()

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

submissionsJson, hvsJson, subscribersJson, chefsLoaded = submission_loader.load_submissions()

if chefsLoaded:
    currentDate = datetime.datetime.now(tz=pytz.timezone('Canada/Pacific'))

    logging.info('Creating source site, receiving site records...')
    sourceSites, srcRegDistDic, receivingSites, rcvRegDistDic, hvSites, hvRegDistDic = submission_mapper.map_sites(submissionsJson, hvsJson, currentDate)

    logging.info('Creating soil source / receiving / high volume site CSV...')
    csv_writer.site_csv_writer(sourceSites, receivingSites, hvSites)

    if AGOL_UPDATE_FLAG:
        logging.info('Updating source / receiving / high volume site CSV and Layer in AGOL...')
        agol_updater.agol_items_overwrite()
    if EMAIL_NOTIFY_FLAG:
        logging.info('Sending notification emails to subscribers...')
        email_sender.email_to_subscribers(subscribersJson, srcRegDistDic, rcvRegDistDic, hvRegDistDic, currentDate)

    logging.info('Completed Soils data publishing')