MetaPhase-Consulting/State-TalentMAP-API

import logging
import jwt
import pydash
from functools import partial
from django.conf import settings
from django.http import QueryDict
from django.http import HttpResponse
from django.utils.encoding import smart_str

from talentmap_api.fsbid.services import common as services

API_ROOT = settings.WS_ROOT_API_URL

logger = logging.getLogger(__name__)



def create_remark_and_remark_insert(query={}, jwt_token=None, host=None):
    '''
    Create remark and inserts
    '''
    hru_id = jwt.decode(jwt_token, verify=False).get('sub')
    query['create_id'] = hru_id
    query['update_id'] = hru_id
    remark = create_remark(query, jwt_token)
    rmrk_seq_num = pydash.get(remark, '[0].rmrk_seq_num')

    if rmrk_seq_num:
        for x in query['rmrkInsertionList']:
            formattedInsert = {}
            formattedInsert['ritext'] = x
            formattedInsert['ricreateid'] = hru_id
            formattedInsert['riupdateid'] = hru_id
            formattedInsert['rirolerestrictedind'] = 'N'
            formattedInsert['rirmrkseqnum'] = rmrk_seq_num
            create_remark_insert(rmrk_seq_num, formattedInsert, jwt_token)
        else:
            logger.error("Create remark insert failed")
    else:
        logger.error("Create remark failed")


def create_remark(query, jwt_token):
    '''
    Create Remark
    '''
    args = {
        "uri": "v1/remarks",
        "query": query,
        "query_mapping_function": convert_remark_query,
        "jwt_token": jwt_token,
        "mapping_function": "",
    }

    return services.get_results_with_post(
        **args
    )


def create_remark_insert(rmrk_seq_num, query, jwt_token):
    '''
    Create Remark Insert 
    '''
    args = {
        "uri": f"v1/remarks/{rmrk_seq_num}/inserts",
        "query": query,
        "query_mapping_function": convert_remark_insert_query,
        "jwt_token": jwt_token,
        "mapping_function": "",
    }

    return services.get_results_with_post(
        **args
    )


def convert_remark_query(query, is_edit=False):
    '''
    Converts TalentMap Remarks into FSBid Remarks
    '''
    formatted_query = {
        'rmrkrccode': query.get('rmrkCategory'),
        'rmrkordernum': 0,
        'rmrkshortdesctext': query.get('shortDescription') or 'None Listed',
        'rmrkmutuallyexclusiveind': 'N',
        'rmrktext': query.get('longDescription'),
        'rmrkactiveind': 'Y' if query.get('activeIndicator') else 'N',
        'rmrkcreateid': query.get('create_id'),
        'rmrkupdateid': query.get('update_id'),
        # Need to format the date for fsbid service, expects space b/w date and time, not T
        'rmrkupdatedate': query.get('update_date', '').replace('T', ' '),
    }
    if is_edit:
        formatted_query['rmrkseqnum'] = query.get('seq_num')
    return formatted_query


def convert_remark_insert_query(query):
    '''
    Converts TalentMap Remark Inserts into FSBid Remark Inserts
    '''
    return {
        'riseqnum': query.get('riseqnum'),
        'rirmrkseqnum': query.get('rirmrkseqnum'),
        'riinsertiontext': query.get('ritext'),
        'rirolerestrictedind': query.get('rirolerestrictedind'),
        'ricreateid': query.get('ricreateid'), 
        'ricreatedate': query.get('ricreatedate'), 
        'riupdateid': query.get('riupdateid'),
        'riupdatedate': query.get('riupdatedate'),
    }


def edit_remark(pk, query, jwt_token):
    '''
    Edit Remark
    '''
    hru_id = jwt.decode(jwt_token, verify=False).get('sub')
    query['update_id'] = hru_id
    args = {
        "uri": f"v1/remarks/{pk}",
        "query": query,
        "query_mapping_function": partial(convert_remark_query, is_edit=True),
        "jwt_token": jwt_token,
        "mapping_function": "",
    }

    return services.send_post_request(
        **args
    )