ligandsfitting_model.py

from neomodel import StructuredNode, StringProperty, IntegerProperty,UniqueIdProperty, RelationshipTo, DateTimeProperty, BooleanProperty
from uuid import uuid4

class LigandsFitting(StructuredNode):
 	uuid=StringProperty(unique_index=True, default=uuid4)
 	dp_step_name = StringProperty()
 	created_at=DateTimeProperty()
 	updated_at=DateTimeProperty()
 	pipedream_id=IntegerProperty()
 	score=IntegerProperty()
 	fitting_success=BooleanProperty()

 	@property
 	def serialize(self):

 		"""
 		Serializer for node properties
 		"""
 		
 	    return {
 	        'node_properties': {
 	            'uuid': self.uuid,
 	            'dp_step_name': self.dp_step_name,
 	            'created_at': self.created_at,
 	            'updated_at': self.updated_at,	       
 	            'pipedream_id': self.pipedream_id, 	            
 	            'score': self.score,	    
 	            'fitting_success': self.fitting_success,
 	        },
 	    }