From ae84b2a539f7af3eaff8f710bba9b13798455ef6 Mon Sep 17 00:00:00 2001
From: 4shen0ne <4shen.01@gmail.com>
Date: Tue, 6 Aug 2024 10:58:38 +0800
Subject: [PATCH] =?UTF-8?q?=F0=9F=9A=A7=20taskprogress=20element?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 nmass/models.py | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/nmass/models.py b/nmass/models.py
index f37a52b..5b826fe 100644
--- a/nmass/models.py
+++ b/nmass/models.py
@@ -115,6 +115,14 @@ class HostHint(BaseXmlModel, tag="hosthint"):
     hostnames: Hostnames = element(default=None)
 
 
+class TaskProgress(BaseXmlModel, tag="taskprogress"):
+    task: str = attr()
+    time: str = attr()
+    percent: float = attr()
+    remaining: int = attr(default=None)
+    etc: str = attr(default=None)
+
+
 class NmapRun(BaseXmlModel, tag="nmaprun"):
     """
     This is the data model that maps with the Nmap (also Masscan) XML output.
@@ -136,5 +144,6 @@ class Stats(BaseXmlModel, tag="runstats"):
     verbose: dict[str, int] = element(default=None)  # None for masscan
     debugging: dict[str, int] = element(default=None)  # None for masscan
     hosthint: HostHint = element(default=None)
+    taskprogress: list[TaskProgress] = element(default=None)
     hosts: list[Host]
     stats: Stats