diff options
Diffstat (limited to 'generalresearch/managers/network/rdns.py')
| -rw-r--r-- | generalresearch/managers/network/rdns.py | 26 |
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 |
