vendor-management-api/dao/vendor_dao.py

from utils.UpdateOwnId import updateOwnId
from datetime import datetime
from utils.error_log import insert_into_error_log

def validateVendorCodeDao(v_code,mongo):
    query=mongo["v_tbl_vendor_code"].find({'code':v_code})
    return list(query)

def validatePANDao(v_pan,mongo):
    query=mongo["v_tbl_vendor_master"].find({'v_pan':v_pan})
    return list(query)

def getTotalUserCount(mongo):
    count=0
    try:
        count=mongo["v_tbl_vendor_master"].count_documents({})
    except Exception as err:
        print('getTotalUserCount',err)
    return count

def insertIntoVendorMaster(requestObject,mongo):
    _id=""
    try:
        UpdateResult =mongo["v_tbl_vendor_master"].update_one({'id':str(requestObject['id'])},{'$set':requestObject},upsert=True)
        _id=UpdateResult.raw_result['upserted']
        updateOwnId("v_tbl_vendor_master",_id,mongo)
    except Exception as identifier:
        print(identifier)

    return _id

def updateVendorStatusDao(requestObject,v_id,mongo):
    if 'requestKey' in requestObject:
        del requestObject['requestKey']

    if 'flag' in requestObject:
        del requestObject['flag']
        
    try:
        UpdateResult =mongo["v_tbl_vendor_master"].update_one({'id':v_id},{'$set':requestObject})
        
    except Exception as identifier:
        print(identifier)

    return ""

def updateVendorStatusAdminDao(requestObject,v_id,mongo):
    
    try:
        UpdateResult =mongo["v_tbl_vendor_master"].update_one({'id':v_id},{'$set':requestObject})
        
    except Exception as identifier:
        print(identifier)

    return ""

def getVendorListByUID(u_id,mongo):
    vendors=[]
    try:
        vendors=list(mongo["v_tbl_vendor_master"].find({'user_id':u_id},{'_id':0}))
    except Exception as err:
        print('getVendorListByUID',err)
    return vendors

def getAllVendorListDao(mongo):
    vendors=[]
    try:
        vendors=list(mongo["v_tbl_vendor_master"].find({},{'_id':0}))
    except Exception as err:
        print('getVendorListByUID',err)
    return vendors

def getAllUnregisteredVendorListDao(mongo):
    vendors=[]
    try:
        vendors=list(mongo["v_tbl_voucher_master"].find({'vendor_name':{'$exists':True}},{'_id':0,'vendor_name':1,'vendor_email':1,'vendor_address':1,'vendor_number':1}))
    except Exception as err:
        print('getVendorListByUID',err)
    return vendors

def getVendorDetailsDao(id,mongo):
    vendors=[]
    try:
        vendors=list(mongo["v_tbl_vendor_master"].find({'id':id},{'_id':0}))
    except Exception as err:
        print('getVendorListByUID',err)
    return vendors

def updateVendorRatingDao(requestObject,mongo):
    _id=""
    try:
        mongo["v_tbl_vendor_rating"].update_one({'user_id':str(requestObject['user_id']),'vendor_id':str(requestObject['vendor_id'])},{'$set':requestObject},upsert=True)
        
    except Exception as identifier:
        print(identifier)

    return _id

def getVendorRatingsDao(id,mongo):
    vendors=[]
    try:
        vendors=list(mongo["v_tbl_vendor_rating"].find({'vendor_id':id},{'_id':0}))
    except Exception as err:
        print('getVendorRatingsDao',err)
    return vendors


def getAllUserListDao(mongo):
    vendors=[]
    try:
        vendors=list(mongo["users"].find({},{'_id':0,'key':1,'full_name':1}))
    except Exception as err:
        print('getVendorListByUID',err)
    return vendors