Error in datetime module in python script for ni-fi

Question

So I am trying to implement this script

from datetime import datetime, timedelta, date                                         

flowfile = session.get()
current_date= datetime.today().date()
if flowfile != None:
	src_table_name = flowfile.getAttribute('src_table_name')
	date_filter = str(current_date.replace(day =11))
	if src_table_name in ['Income_project_BP', 'Plan_final_scenario', 'Plan_primary_scenario', 'ProductionCalendar', 'Subcontracting_BP']:	
		flowfile = session.putAttribute(flowfile, 'date_filter', date_filter)
		session.transfer(flowfile, REL_SUCCESS)
	else:
		session.transfer(flowfile, REL_FAILURE)

And I've tried a lot of different ways but when it comes to "replace" method I'm always getting this error. Same thing when I am trying to use "strftime" method

What should I do? I really need this method.

Thank you beforehand!

SAMSAL · Accepted Answer

Hi,I think something is happening when nifi is trying to interpret the correct data type in python. to make it work make the current_date string as follows:current_date= str(datetime.today().date())and let the python engine do the conversion to the correct datetime type when you set the date_filter as follows:date_filter = str(datetime.strptime(current_date,'%Y-%m-%d').replace(day =11)) Not sure if this is a bug in Nifi, but at least this should be a work around.Hope that helps.

SAMSAL · Answer

set the date filter as follows:date_filter = datetime.strptime(current_date,'%Y-%m-%d').replace(day =11).strftime('%Y%m%d') If that solves your problem please accept solution.Thanks

Sign up

Please use your Cloudera account.

Login to the community

Please use your Cloudera account.