vendor-management-api/dao/voucher_approval_dao.py

from utils.UpdateOwnId import updateOwnId
from dateutil.relativedelta import relativedelta
from datetime import datetime

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 getVoucherDetailsByCodeDao(voucher_code, mongo):
    vouchers = []
    try:
        vouchers = list(mongo["v_tbl_voucher_master"].find({'voucher_code': voucher_code},
                                                           {'_id': 0}))
    except Exception as err:
        print('getVoucherDetailsByCodeDao', 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_one(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 getApprovedVoucherListForRollbackDao( mongo):
    vouchers = []
    try:
        current_date = datetime.today()
        n = 10
        past_date = current_date - relativedelta(months=n)
        date = int(past_date.timestamp()*1000)
        vouchers = list(mongo["v_tbl_voucher_master"].find(
            {
                '$or': [{
                    'approval_track.finance_level_1.info.updated_on': {'$gte':date},
                    'approval_track.finance_level_1.status': 'APPROVED'
                }, {
                    'approval_track.finance_level_2.info.updated_on': {'$gte':date},
                    'approval_track.finance_level_2.status': 'APPROVED'
                }, {
                    'approval_track.finance_level_3.info.updated_on': {'$gte':date},
                    'approval_track.finance_level_3.status': 'APPROVED'
                }]
            }, {'_id': 0},sort=[('voucher_created_on', -1)]))

       
    except Exception as err:
        print('getApprovedVoucherListForRollbackDao', 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(2000))

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