from utils.UpdateOwnId import updateOwnId
from datetime import datetime

def getVoucherListByUID(u_id,mongo):
    vouchers=[]
    try:
        vouchers=list(mongo["v_tbl_voucher_master"].find({'deleted':{'$ne':True},'user_id':u_id},{'_id':0}))
    except Exception as err:
        print('getProjectListByUID',err)
    return vouchers

def getVoucherListById(id,mongo):
    voucher_lst=[]
    try:
        voucher_lst=list(mongo["v_tbl_voucher_master"].find({'id':id},{'_id':0}).limit(1))
    except Exception as err:
        print('getVoucherListById',err)
    return voucher_lst

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

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

    return _id

def getVoucherYear(mongo):
    years=''
    try:
        years=list(mongo["v_po_year"].find({},{'_id':0},sort=[('timestamp', 1)]).limit(1))
    except Exception as err:
        print('getProjectListByUID',err)
    return years

def insertVoucherYear(po_year,mongo):
    try:

        mongo["v_po_year"].insert({'year':po_year,'timestamp':int(datetime.now().timestamp() *1000)})
    except Exception as err:
        print(err)

def getInserterdVoucherCount(voucher_year,mongo):
    count=0
    try:
        count=mongo["v_tbl_voucher_master"].count({'voucher_year':voucher_year})
    except Exception as err:
        print(err)
    
    return count

def getVoucherAnalyticsDataDao(from_date,to_date,mongo):
    vouchers=[]
    try:
        vouchers=list(mongo["v_tbl_voucher_master"].find({"voucher_status":"APPROVED","voucher_created_on":{"$gte":int(from_date),"$lte":int(to_date)}},{'_id':0}))
    except Exception as err:
        print('getProjectListByUID',err)
    return vouchers

def updateVoucherDao(voucherCode,date,mongo):
    try:
        mongo["v_tbl_voucher_master"].update_one({'voucher_code':str(voucherCode)},{'$set':{'voucher_created_on':date}})
        
    except Exception as identifier:
        print(identifier)

def updateVoucherPhyCopyStatusDao(id,date,mongo):
    try:
        mongo["v_tbl_voucher_master"].update_one({'id':id},{'$set':{'sub_phy_copy':True,'phy_sub_on':date}})
        
    except Exception as identifier:
        print(identifier)

def updateVoucherDeleteStatusDao(id,date,mongo):
    try:
        mongo["v_tbl_voucher_master"].update_one({'id':id},{'$set':{'deleted':True,'deleted_on':date}})
        
    except Exception as identifier:
        print(identifier)

def getVoucherStatusLogDao(id,mongo):
    voucher_status=[]
    try:
        voucher_status=list(mongo["v_voucher_status_log"].find({'voucher_id':id},{'_id':0}))
    except Exception as err:
        print('getVoucherStatusLogDao',err)
    return voucher_status

def getAssetCodePresentOrNotDao(asset_code_list,mongo):
    asset_count=[]
    try:
        asset_count=mongo["i_tbl_asset_master_new"].count_documents({"asset_code":{"$in":asset_code_list}})
    except Exception as err:
        print('getAssetCodePresentOrNotDao',err)
    return asset_count

def updateAssetDetailsSelected(asset_code_list,update_obj,mongo):
    try:
        mongo["i_tbl_asset_master_new"].update_many({"asset_code":{"$in":asset_code_list}},{'$set':update_obj})
        
    except Exception as identifier:
        print(identifier)

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

    return _id

def voucherFileDeleteDao(query,update_obj,mongo):
    try:
        mongo["v_tbl_voucher_master"].update_one(query,update_obj)
    except Exception as identifier:
        print(identifier)