from neomodel import StructuredNode, StringProperty, DateTimeFormatProperty, IntegerProperty, BooleanProperty, \
    RelationshipTo, StructuredRel


class RelResolver53(StructuredRel):
    W = IntegerProperty()
    LTIME = DateTimeFormatProperty(default_now=True, format="%Y-%m-%d %H:%M:%S")


class RelResolverQuery(StructuredRel):
    W = IntegerProperty()


class NodeResolverQuery(StructuredNode):
    QNAME = StringProperty(required=True)
    QTYPE = StringProperty()


class NodeResolver53(StructuredNode):
    IP = StringProperty(required=True, unique_index=True)
    AS = StringProperty()
    ISP = StringProperty()
    COU = StringProperty()
    CCODE = StringProperty()
    PROV = StringProperty()
    LAT = StringProperty()
    LNG = StringProperty()
    IPType = StringProperty()
    FTIME = DateTimeFormatProperty(format="%Y-%m-%d %H:%M:%S")
    LTIME = DateTimeFormatProperty(default_now=True, format="%Y-%m-%d %H:%M:%S")
    W = IntegerProperty()
    ISPUBLIC = BooleanProperty(default=False)
    LINK = RelationshipTo("NodeResolver53", "IP_LINK", model=RelResolver53)
    QLINK = RelationshipTo("NodeResolverQuery", "Q_LINK", model=RelResolverQuery)