diff --git a/neomodel/scripts/neomodel_inspect_database.py b/neomodel/scripts/neomodel_inspect_database.py index bb2d8452..ca3de5ad 100644 --- a/neomodel/scripts/neomodel_inspect_database.py +++ b/neomodel/scripts/neomodel_inspect_database.py @@ -55,6 +55,9 @@ def parse_prop_class(prop_type): elif prop_type == "BOOLEAN": _import = "BooleanProperty" prop_class = f"{_import}(" + elif prop_type == "DATE": + _import = "DateProperty" + prop_class = f"{_import}(" elif prop_type == "DATE_TIME": _import = "DateTimeProperty" prop_class = f"{_import}(" diff --git a/test/data/neomodel_inspect_database_output.txt b/test/data/neomodel_inspect_database_output.txt index 3a532dd6..8f0ea4e0 100644 --- a/test/data/neomodel_inspect_database_output.txt +++ b/test/data/neomodel_inspect_database_output.txt @@ -1,4 +1,4 @@ -from neomodel import StructuredNode, StringProperty, RelationshipTo, StructuredRel, ZeroOrOne, ArrayProperty, FloatProperty, BooleanProperty, DateTimeProperty, IntegerProperty +from neomodel import StructuredNode, StringProperty, RelationshipTo, StructuredRel, ZeroOrOne, ArrayProperty, FloatProperty, BooleanProperty, DateProperty, DateTimeProperty, IntegerProperty from neomodel.contrib.spatial_properties import PointProperty class ScriptsTestNode(StructuredNode): @@ -19,6 +19,7 @@ class EveryPropertyTypeNode(StructuredNode): boolean_property = BooleanProperty() point_property = PointProperty(crs='wgs-84') string_property = StringProperty() + date_property = DateProperty() datetime_property = DateTimeProperty() integer_property = IntegerProperty() diff --git a/test/data/neomodel_inspect_database_output_light.txt b/test/data/neomodel_inspect_database_output_light.txt index a7fd16ec..556a9ca9 100644 --- a/test/data/neomodel_inspect_database_output_light.txt +++ b/test/data/neomodel_inspect_database_output_light.txt @@ -1,4 +1,4 @@ -from neomodel import StructuredNode, StringProperty, RelationshipTo, ArrayProperty, FloatProperty, BooleanProperty, DateTimeProperty, IntegerProperty +from neomodel import StructuredNode, StringProperty, RelationshipTo, ArrayProperty, FloatProperty, BooleanProperty, DateProperty, DateTimeProperty, IntegerProperty from neomodel.contrib.spatial_properties import PointProperty class ScriptsTestNode(StructuredNode): @@ -14,6 +14,7 @@ class EveryPropertyTypeNode(StructuredNode): boolean_property = BooleanProperty() point_property = PointProperty(crs='wgs-84') string_property = StringProperty() + date_property = DateProperty() datetime_property = DateTimeProperty() integer_property = IntegerProperty() diff --git a/test/data/neomodel_inspect_database_output_pre_5_7.txt b/test/data/neomodel_inspect_database_output_pre_5_7.txt index b025411e..dc9c4217 100644 --- a/test/data/neomodel_inspect_database_output_pre_5_7.txt +++ b/test/data/neomodel_inspect_database_output_pre_5_7.txt @@ -1,4 +1,4 @@ -from neomodel import StructuredNode, StringProperty, RelationshipTo, StructuredRel, ZeroOrOne, ArrayProperty, FloatProperty, BooleanProperty, DateTimeProperty, IntegerProperty +from neomodel import StructuredNode, StringProperty, RelationshipTo, StructuredRel, ZeroOrOne, ArrayProperty, FloatProperty, BooleanProperty, DateProperty, DateTimeProperty, IntegerProperty from neomodel.contrib.spatial_properties import PointProperty class ScriptsTestNode(StructuredNode): @@ -19,6 +19,7 @@ class EveryPropertyTypeNode(StructuredNode): boolean_property = BooleanProperty() point_property = PointProperty(crs='wgs-84') string_property = StringProperty() + date_property = DateProperty() datetime_property = DateTimeProperty() integer_property = IntegerProperty() diff --git a/test/data/neomodel_inspect_database_output_pre_5_7_light.txt b/test/data/neomodel_inspect_database_output_pre_5_7_light.txt index a7fd16ec..556a9ca9 100644 --- a/test/data/neomodel_inspect_database_output_pre_5_7_light.txt +++ b/test/data/neomodel_inspect_database_output_pre_5_7_light.txt @@ -1,4 +1,4 @@ -from neomodel import StructuredNode, StringProperty, RelationshipTo, ArrayProperty, FloatProperty, BooleanProperty, DateTimeProperty, IntegerProperty +from neomodel import StructuredNode, StringProperty, RelationshipTo, ArrayProperty, FloatProperty, BooleanProperty, DateProperty, DateTimeProperty, IntegerProperty from neomodel.contrib.spatial_properties import PointProperty class ScriptsTestNode(StructuredNode): @@ -14,6 +14,7 @@ class EveryPropertyTypeNode(StructuredNode): boolean_property = BooleanProperty() point_property = PointProperty(crs='wgs-84') string_property = StringProperty() + date_property = DateProperty() datetime_property = DateTimeProperty() integer_property = IntegerProperty() diff --git a/test/test_scripts.py b/test/test_scripts.py index 07a5a14b..6f8fcf4a 100644 --- a/test/test_scripts.py +++ b/test/test_scripts.py @@ -125,6 +125,7 @@ def test_neomodel_inspect_database(script_flavour): CREATE (:EveryPropertyTypeNode { string_property: "Hello World", boolean_property: true, + date_property: date("2020-01-01"), datetime_property: datetime("2020-01-01T00:00:00.000Z"), integer_property: 1, float_property: 1.0,