aboutsummaryrefslogtreecommitdiff
path: root/generalresearch/managers/network/rdns.py
diff options
context:
space:
mode:
Diffstat (limited to 'generalresearch/managers/network/rdns.py')
-rw-r--r--generalresearch/managers/network/rdns.py26
1 files changed, 26 insertions, 0 deletions
diff --git a/generalresearch/managers/network/rdns.py b/generalresearch/managers/network/rdns.py
new file mode 100644
index 0000000..2eed303
--- /dev/null
+++ b/generalresearch/managers/network/rdns.py
@@ -0,0 +1,26 @@
+from typing import Optional
+
+from psycopg import Cursor
+
+from generalresearch.managers.base import PostgresManager
+from generalresearch.models.network.tool_run import RDnsRun
+
+
+class RdnsManager(PostgresManager):
+
+ def _create(self, run: RDnsRun, c: Optional[Cursor] = None) -> None:
+ """
+ Do not use this directly. Must only be used in the context of a toolrun
+ """
+ query = """
+ INSERT INTO network_rdnsresult (
+ run_id, primary_hostname, primary_org,
+ hostname_count, hostnames
+ )
+ VALUES (
+ %(run_id)s, %(primary_hostname)s, %(primary_org)s,
+ %(hostname_count)s, %(hostnames)s
+ );
+ """
+ params = run.model_dump_postgres()
+ c.execute(query, params) \ No newline at end of file