mirror of
https://github.com/CyberSecurityUP/NeuroSploit.git
synced 2026-08-21 00:47:13 +02:00
Add files via upload
This commit is contained in:
@@ -0,0 +1,47 @@
|
||||
"""
|
||||
NeuroSploit v3 - Target Model
|
||||
"""
|
||||
from datetime import datetime
|
||||
from typing import Optional
|
||||
from sqlalchemy import String, Integer, DateTime, ForeignKey
|
||||
from sqlalchemy.orm import Mapped, mapped_column, relationship
|
||||
from backend.db.database import Base
|
||||
import uuid
|
||||
|
||||
|
||||
class Target(Base):
|
||||
"""Target URL model"""
|
||||
__tablename__ = "targets"
|
||||
|
||||
id: Mapped[str] = mapped_column(String(36), primary_key=True, default=lambda: str(uuid.uuid4()))
|
||||
scan_id: Mapped[str] = mapped_column(String(36), ForeignKey("scans.id", ondelete="CASCADE"))
|
||||
|
||||
# URL details
|
||||
url: Mapped[str] = mapped_column(String(2048))
|
||||
hostname: Mapped[Optional[str]] = mapped_column(String(255), nullable=True)
|
||||
port: Mapped[Optional[int]] = mapped_column(Integer, nullable=True)
|
||||
protocol: Mapped[Optional[str]] = mapped_column(String(10), nullable=True)
|
||||
path: Mapped[Optional[str]] = mapped_column(String(2048), nullable=True)
|
||||
|
||||
# Status
|
||||
status: Mapped[str] = mapped_column(String(50), default="pending") # pending, scanning, completed, failed
|
||||
|
||||
# Timestamps
|
||||
created_at: Mapped[datetime] = mapped_column(DateTime, default=datetime.utcnow)
|
||||
|
||||
# Relationship
|
||||
scan: Mapped["Scan"] = relationship("Scan", back_populates="targets")
|
||||
|
||||
def to_dict(self) -> dict:
|
||||
"""Convert to dictionary"""
|
||||
return {
|
||||
"id": self.id,
|
||||
"scan_id": self.scan_id,
|
||||
"url": self.url,
|
||||
"hostname": self.hostname,
|
||||
"port": self.port,
|
||||
"protocol": self.protocol,
|
||||
"path": self.path,
|
||||
"status": self.status,
|
||||
"created_at": self.created_at.isoformat() if self.created_at else None
|
||||
}
|
||||
Reference in New Issue
Block a user