MCPcopy
hub / github.com/Pennyw0rth/NetExec / init_protocol_dbs

Function init_protocol_dbs

nxc/database.py:58–71  ·  view source on GitHub ↗

Check for each protocol if the database exists, if not create it.

(workspace_name, p_loader=None)

Source from the content-addressed store, hash-verified

56
57
58def init_protocol_dbs(workspace_name, p_loader=None):
59 """Check for each protocol if the database exists, if not create it."""
60 if p_loader is None:
61 p_loader = ProtocolLoader()
62 protocols = p_loader.get_protocols()
63 for protocol in protocols:
64 protocol_object = p_loader.load_protocol(protocols[protocol]["dbpath"])
65 proto_db_path = path_join(WORKSPACE_DIR, workspace_name, f"{protocol}.db")
66
67 if not exists(proto_db_path):
68 print(f"[*] Initializing {protocol.upper()} protocol database")
69 db_engine = create_db_engine(proto_db_path)
70 protocol_object.database.db_schema(db_engine)
71 db_engine.dispose()
72
73
74def create_workspace(workspace_name, p_loader=None):

Callers 2

create_workspaceFunction · 0.85
initialize_dbFunction · 0.85

Calls 5

get_protocolsMethod · 0.95
load_protocolMethod · 0.95
ProtocolLoaderClass · 0.90
create_db_engineFunction · 0.85
db_schemaMethod · 0.45

Tested by

no test coverage detected