@@ -23,7 +23,6 @@ class MigrationStep:
23
23
24
24
@dataclass
25
25
class MigrationNode :
26
- last_node_id = 0
27
26
node_id : int
28
27
object_type : str
29
28
object_id : str
@@ -69,6 +68,7 @@ class MigrationSequencer:
69
68
70
69
def __init__ (self , ws : WorkspaceClient ):
71
70
self ._ws = ws
71
+ self ._last_node_id = 0
72
72
self ._root = MigrationNode (
73
73
node_id = 0 , object_type = "ROOT" , object_id = "ROOT" , object_name = "ROOT" , object_owner = "NONE"
74
74
)
@@ -79,9 +79,9 @@ def register_workflow_task(self, task: jobs.Task, job: jobs.Job, _graph: Depende
79
79
if task_node :
80
80
return task_node
81
81
job_node = self .register_workflow_job (job )
82
- MigrationNode . last_node_id += 1
82
+ self . _last_node_id += 1
83
83
task_node = MigrationNode (
84
- node_id = MigrationNode . last_node_id ,
84
+ node_id = self . _last_node_id ,
85
85
object_type = "TASK" ,
86
86
object_id = task_id ,
87
87
object_name = task .task_key ,
@@ -100,10 +100,10 @@ def register_workflow_job(self, job: jobs.Job) -> MigrationNode:
100
100
job_node = self ._find_node (object_type = "JOB" , object_id = str (job .job_id ))
101
101
if job_node :
102
102
return job_node
103
- MigrationNode . last_node_id += 1
103
+ self . _last_node_id += 1
104
104
job_name = job .settings .name if job .settings and job .settings .name else str (job .job_id )
105
105
job_node = MigrationNode (
106
- node_id = MigrationNode . last_node_id ,
106
+ node_id = self . _last_node_id ,
107
107
object_type = "JOB" ,
108
108
object_id = str (job .job_id ),
109
109
object_name = job_name ,
@@ -132,9 +132,9 @@ def register_cluster(self, cluster_key: str) -> MigrationNode:
132
132
details = self ._ws .clusters .get (cluster_key )
133
133
object_name = details .cluster_name if details and details .cluster_name else cluster_key
134
134
object_owner = details .creator_user_name if details and details .creator_user_name else "<UNKNOWN>"
135
- MigrationNode . last_node_id += 1
135
+ self . _last_node_id += 1
136
136
cluster_node = MigrationNode (
137
- node_id = MigrationNode . last_node_id ,
137
+ node_id = self . _last_node_id ,
138
138
object_type = "CLUSTER" ,
139
139
object_id = cluster_key ,
140
140
object_name = object_name ,
0 commit comments