1
1
import json
2
+ import logging
3
+ from pprint import pformat
2
4
from typing import Any , Dict , Set
3
5
4
6
from aiopg .sa .result import RowProxy
7
9
from ..node_ports .exceptions import InvalidProtocolError
8
10
from .nodeports_v2 import Nodeports
9
11
12
+ log = logging .getLogger (__name__ )
13
+
10
14
NODE_REQUIRED_KEYS : Set [str ] = {
11
15
"schema" ,
12
16
"inputs" ,
@@ -18,6 +22,11 @@ async def load(
18
22
db_manager : DBManager , node_uuid : str , auto_update : bool = False
19
23
) -> Nodeports :
20
24
"""creates a nodeport object from a row from comp_tasks"""
25
+ log .debug (
26
+ "creating node_ports_v2 object from node %s with auto_uptate %s" ,
27
+ node_uuid ,
28
+ auto_update ,
29
+ )
21
30
row : RowProxy = await db_manager .get_ports_configuration_from_node_uuid (node_uuid )
22
31
port_cfg = json .loads (row )
23
32
if any (k not in port_cfg for k in NODE_REQUIRED_KEYS ):
@@ -50,10 +59,18 @@ async def load(
50
59
node_port_creator_cb = load ,
51
60
auto_update = auto_update ,
52
61
)
62
+ log .debug (
63
+ "created node_ports_v2 object %s" ,
64
+ pformat (ports , indent = 2 ),
65
+ )
53
66
return ports
54
67
55
68
56
69
async def dump (nodeports : Nodeports ) -> None :
70
+ log .debug (
71
+ "dumping node_ports_v2 object %s" ,
72
+ pformat (nodeports , indent = 2 ),
73
+ )
57
74
_nodeports_cfg = nodeports .dict (
58
75
include = {"internal_inputs" , "internal_outputs" },
59
76
by_alias = True ,
0 commit comments