11from neo4j import GraphDatabase , time
22from datetime import datetime
3- from neo4j .time import DateTime
3+ from neo4j .time import DateTime , Date
44import time
55import pytz
66import csv
@@ -25,6 +25,10 @@ def convert_to_datetime(timestamp):
2525 dt = datetime .strptime (timestamp , '%Y-%m-%d' )
2626 return DateTime (dt .year , dt .month , dt .day , 0 , 0 , 0 , pytz .timezone ('GMT' ))
2727
28+ def convert_to_date (timestamp ):
29+ dt = datetime .strptime (timestamp , '%Y-%m-%d' )
30+ return Date (dt .year , dt .month , dt .day )
31+
2832driver = GraphDatabase .driver ("bolt://localhost:7687" )
2933
3034with driver .session () as session :
@@ -37,6 +41,7 @@ def convert_to_datetime(timestamp):
3741 for query_parameters in parameters_csv :
3842 # convert fields based on type designators
3943 query_parameters = {k : int (v ) if re .match ('.*:(ID|LONG)' , k ) else v for k , v in query_parameters .items ()}
44+ query_parameters = {k : convert_to_date (v ) if re .match ('.*:DATE$' , k ) else v for k , v in query_parameters .items ()}
4045 query_parameters = {k : convert_to_datetime (v ) if re .match ('.*:DATETIME' , k ) else v for k , v in query_parameters .items ()}
4146 query_parameters = {k : v .split (';' ) if re .findall ('\[\]$' , k ) else v for k , v in query_parameters .items ()}
4247 # drop type designators
0 commit comments