mirror of https://github.com/OWASP/Nettacker.git
101 lines
2.8 KiB
Python
101 lines
2.8 KiB
Python
from sqlalchemy import Column, Integer, Text, DateTime
|
|
from sqlalchemy.orm import declarative_base
|
|
|
|
Base = declarative_base()
|
|
|
|
|
|
class Report(Base):
|
|
"""
|
|
This class defines the table schema of the reports table. Any changes
|
|
to the reports table need to be done here.
|
|
"""
|
|
|
|
__tablename__ = "reports"
|
|
|
|
id = Column(Integer, primary_key=True, autoincrement=True)
|
|
date = Column(DateTime)
|
|
scan_unique_id = Column(Text)
|
|
report_path_filename = Column(Text)
|
|
options = Column(Text)
|
|
|
|
def __repr__(self):
|
|
"""
|
|
returns a printable representation of the object of the class Report
|
|
"""
|
|
return "<Report(id={0}, scan_unique_id={1}, date={2}, report_path_filename={3})>".format(
|
|
self.id, self.scan_unique_id, self.date, self.report_path_filename
|
|
)
|
|
|
|
|
|
class TempEvents(Base):
|
|
"""
|
|
This class defines the table schema of the reports table. Any changes to
|
|
the reports table need to be done here.
|
|
"""
|
|
|
|
__tablename__ = "temp_events"
|
|
|
|
id = Column(Integer, primary_key=True, autoincrement=True)
|
|
date = Column(DateTime)
|
|
target = Column(Text)
|
|
module_name = Column(Text)
|
|
scan_unique_id = Column(Text)
|
|
event_name = Column(Text)
|
|
port = Column(Text)
|
|
event = Column(Text)
|
|
data = Column(Text)
|
|
|
|
def __repr__(self):
|
|
"""
|
|
returns a printable representation of the object of the class Report
|
|
"""
|
|
return """
|
|
<scan_events(id={0}, target={1}, date={2}, module_name={3}, scan_unqiue_id={4},
|
|
port={5}, event={6}, data={7})>
|
|
""".format(
|
|
self.id,
|
|
self.target,
|
|
self.date,
|
|
self.module_name,
|
|
self.scan_unique_id,
|
|
self.port,
|
|
self.event,
|
|
self.data,
|
|
)
|
|
|
|
|
|
class HostsLog(Base):
|
|
"""
|
|
This class defines the table schema of the hosts_log table.
|
|
Any changes to the reports hosts_log need to be done here.
|
|
"""
|
|
|
|
__tablename__ = "scan_events"
|
|
|
|
id = Column(Integer, primary_key=True, autoincrement=True)
|
|
date = Column(DateTime)
|
|
target = Column(Text)
|
|
module_name = Column(Text)
|
|
scan_unique_id = Column(Text)
|
|
port = Column(Text)
|
|
event = Column(Text)
|
|
json_event = Column(Text)
|
|
|
|
def __repr__(self):
|
|
"""
|
|
returns a printable representation of the object of the class HostsLog
|
|
"""
|
|
return """
|
|
<scan_events(id={0}, target={1}, date={2}, module_name={3}, scan_unqiue_id={4},
|
|
port={5}, event={6}, json_event={7})>
|
|
""".format(
|
|
self.id,
|
|
self.target,
|
|
self.date,
|
|
self.module_name,
|
|
self.scan_unique_id,
|
|
self.port,
|
|
self.event,
|
|
self.json_event,
|
|
)
|