01101500

data_processing/data_operation/data_nwp_ftp.py

@@ -196,7 +196,9 @@ def select_file_to_mongo(args):
 
 
 def select_dx_from_nwp(df, args):
-    date = datetime.datetime.strptime(args['dt'], '%Y-%m-%d %H:%M:%S')
+    date, moment = get_moment_next() if args.get('dt') is None else get_moment_next(args.get('dt'))
+    date = datetime.datetime.strptime(date, "%Y%m%d")
+    date = date.strftime('%Y-%m-%d %H:%M:%S')
     date_begin = date + pd.Timedelta(days=int(args.get('day_begin', 'D0')[1:]))
     date_end = date + pd.Timedelta(days=int(args.get('day_end', 'D13')[1:]))
     df['date_time'] = df['date_time'].str.replace("_", " ")