from utils.UpdateOwnId import updateOwnId


def updateSeniorDataMaster(insert_obj, mongo):
    _id = ""
    try:
        UpdateResult = mongo["v_tbl_voucher_senior_master"].update_one(
            {'emp_key': str(insert_obj['emp_key'])}, {'$set': insert_obj},
            upsert=True)
        _id = UpdateResult.raw_result['upserted']
        updateOwnId("v_tbl_voucher_senior_master", _id, mongo)
    except Exception as identifier:
        print(identifier)

    return _id

def getVoucherSeniorListDao(emp_key, mongo):
    voucher_senior_list = []
    try:
        voucher_senior_list = list(mongo["v_tbl_voucher_senior_master"].find(
            {'emp_key': emp_key}, {'_id': 0}))
    except Exception as err:
        print('getVoucherSeniorListDao', err)
    return voucher_senior_list

def getVoucherDetailsDao(id, mongo):
    vouchers = []
    try:
        vouchers = list(mongo["v_tbl_voucher_master"].find({'id': id},
                                                           {'_id': 0}))
    except Exception as err:
        print('getVoucherDetailsDao', err)

    return vouchers

def updateVoucherApprovalStatus(requestObject, voucher_id, mongo):
    try:
        mongo["v_tbl_voucher_master"].update_one({'id': str(voucher_id)},
                                                 {'$set': requestObject})

    except Exception as identifier:
        print(identifier)

def insertVoucherStatusLog(status_obj, mongo):
    try:
        mongo["v_voucher_status_log"].insert(status_obj)
    except Exception as err:
        print(err)

def getPendingVoucherForApprovalDao(u_id, mongo):
    vouchers = []
    try:
        vouchers = list(mongo["v_tbl_voucher_master"].find(
            {
                'voucher_status':'PENDING',
                '$or': [{
                    'approval_track.senior_level_1.id': u_id,
                    'approval_track.senior_level_1.status': 'PENDING'
                }, {
                    'approval_track.senior_level_2.id': u_id,
                    'approval_track.senior_level_2.status': 'PENDING'
                }, {
                    'approval_track.senior_level_3.id': u_id,
                    'approval_track.senior_level_3.status': 'PENDING'
                }, {
                    'approval_track.finance_level_1.id': u_id,
                    'approval_track.finance_level_1.status': 'PENDING'
                }, {
                    'approval_track.finance_level_2.id': u_id,
                    'approval_track.finance_level_2.status': 'PENDING'
                }]
            }, {'_id': 0}))

       
    except Exception as err:
        print('getPendingVoucherForApprovalDao', err)
    return vouchers

def getArchiveVoucherForApprovalDao(u_id, mongo):
    vouchers = []
    try:
        vouchers = list(mongo["v_tbl_voucher_master"].find(
            {
                '$or': [{
                    'approval_track.senior_level_1.id': u_id,
                    'approval_track.senior_level_1.status': {'$ne':'PENDING'}
                }, {
                    'approval_track.senior_level_2.id': u_id,
                    'approval_track.senior_level_2.status': {'$ne':'PENDING'}
                }, {
                    'approval_track.senior_level_3.id': u_id,
                    'approval_track.senior_level_3.status': {'$ne':'PENDING'}
                }, {
                    'approval_track.finance_level_1.id': u_id,
                    'approval_track.finance_level_1.status': {'$ne':'PENDING'}
                }, {
                    'approval_track.finance_level_2.id': u_id,
                    'approval_track.finance_level_2.status': {'$ne':'PENDING'}
                }]
            }, {'_id': 0},sort=[('voucher_created_on', -1)]).limit(100))

       
    except Exception as err:
        print('getPendingVoucherForApprovalDao', err)
    return vouchers