tpx_status.py

import instrument
import nires
import hires
import deimos
import lris
import kcwi
import nirc2
import nirspec
import osiris
import esi
import mosfire
import weather
import json
import db_conn

from flask import Flask, render_template, request, redirect, url_for, session

import logging
log = logging.getLogger('koaapi')


# Create a dictionary of the instrument Constructors
INSTRUMENTS = {
        'deimos':deimos.Deimos,
        'esi':esi.Esi,
        'hires':hires.Hires,
        'kcwi':kcwi.Kcwi,
        'lris':lris.Lris,
        'mosfire':mosfire.Mosfire,
        'nirc2':nirc2.Nirc2,
        'nires':nires.Nires,
        'nirspec':nirspec.Nirspec,
        'osiris':osiris.Osiris,
        'weather':weather.Weather
        }


def tpx_status(dev=False):
    '''
    Method to update the transfer status of koa data

    Arguments for the update will be passed via GET or POST
    instr: the instrument that created the data
    @type instr: string
    date: the utdate the instrument took the data
    @type date: string
    statusType: the type of file transfered
    @type statusType: string
    status: the status of the ingestion from IPAC
    @type status: string
    '''
    # get the arguments passed as a get or post
    args = request.args
    instr = args['instr'].lower()
    date = args['date']
    statusType = args['statusType']
    status = args['status']
    statusMessage = args.get('statusMessage', 'NULL')
    response = ''
    # change dev here for testonly paramater
    try:
        if args['testonly'].lower() == 'true': dev=True
    except:
        pass

    # Create the instrument subclass object based on instr 
    try:
        # Here we create our instrumentStatus object that will handle any differences between instruments
        # INSTRUMENTS is the dictionary of instrument constructors
        instrumentStatus = INSTRUMENTS[instr](instr, date, statusType, status, statusMessage, dev)
    except Exception as e:
        log.error('error creating the object: ' + str(e))
        response = {'APIStatus':'ERROR', 'Message':f"error creating the object for '{instr}'"}
    else:
        # execute the status function based on statusType
        # statusType is defined in the instrument.py file
        try:
            instrumentStatus.myDict['Instrument'] = instrumentStatus.instr
            response = instrumentStatus.types[instrumentStatus.statusType]()
        except Exception as e:
            log.error('error executing the status type: ' + str(e))
            response = {'APIStatus':'ERROR', 'Message':f"error executing the status type '{statusType}'"}
        else:
            print(response)
    return json.dumps(response)