Skip to content

Commit 022f269

Browse files
committed
added nb utils get_cell_id and register_pre_run to handle nb cell pyjviz case
1 parent a03c3c2 commit 022f269

File tree

4 files changed

+165
-50
lines changed

4 files changed

+165
-50
lines changed

examples/notebooks/why-janitor.ipynb

Lines changed: 106 additions & 47 deletions
Original file line numberDiff line numberDiff line change
@@ -10,10 +10,13 @@
1010
"name": "stdout",
1111
"output_type": "stream",
1212
"text": [
13+
"register_pre_run\n",
14+
"pre_run_cell: This function will be called before every cell is executed.\n",
15+
"cell_id: 80990326-1ef0-4588-8e28-0f377bd4b5bd\n",
1316
"call dropna\n",
14-
"my rename 140465675825360\n",
17+
"my rename 140163077598768\n",
1518
"new copy: () {'deep': None}\n",
16-
"my rename 140465675826032\n",
19+
"my rename 140163077599296\n",
1720
"new copy: () {'deep': None}\n",
1821
"new copy: () {}\n"
1922
]
@@ -34,17 +37,17 @@
3437
" <polygon fill=\"white\" points=\"-4,4 -4,-113 1995,-113 1995,4 -4,4\" stroke=\"transparent\"/>\n",
3538
" <g class=\"cluster\" id=\"clust1\">\n",
3639
" <title>\n",
37-
" cluster_d6921421676962807575\n",
40+
" cluster_d1094564498426419428\n",
3841
" </title>\n",
3942
" <polygon fill=\"none\" points=\"8,-8 8,-101 1983,-101 1983,-8 8,-8\" stroke=\"black\"/>\n",
4043
" <text font-family=\"Helvetica,Arial,sans-serif\" font-size=\"14.00\" text-anchor=\"middle\" x=\"995.5\" y=\"-85.8\">\n",
4144
" WHY JANITOR?\n",
4245
" </text>\n",
4346
" </g>\n",
44-
" <!-- node_7350196740953114244 -->\n",
47+
" <!-- node_1227845970559210097 -->\n",
4548
" <g class=\"node\" id=\"node1\">\n",
4649
" <title>\n",
47-
" node_7350196740953114244\n",
50+
" node_1227845970559210097\n",
4851
" </title>\n",
4952
" <g cursor=\"pointer\" id=\"a_node1\" onclick=\" window.open('', '_blank', 'width=800,height=200').document.body.innerHTML = `&lt;table border=&quot;1&quot; class=&quot;dataframe&quot;&gt;\n",
5053
" &lt;thead&gt;\n",
@@ -105,31 +108,31 @@
105108
" </text>\n",
106109
" </g>\n",
107110
" </g>\n",
108-
" <!-- node_d8297656796017466239 -->\n",
111+
" <!-- node_d6555065156464757206 -->\n",
109112
" <g class=\"node\" id=\"node7\">\n",
110113
" <title>\n",
111-
" node_d8297656796017466239\n",
114+
" node_d6555065156464757206\n",
112115
" </title>\n",
113116
" <polygon fill=\"lightgrey\" points=\"365,-61 197,-61 197,-25 365,-25 365,-61\" stroke=\"transparent\"/>\n",
114117
" <text font-family=\"Helvetica,Arial,sans-serif\" font-size=\"14.00\" text-anchor=\"start\" x=\"209\" y=\"-38.8\">\n",
115118
" remove_columns(df, ...)\n",
116119
" </text>\n",
117120
" </g>\n",
118-
" <!-- node_7350196740953114244&#45;&gt;node_d8297656796017466239 -->\n",
121+
" <!-- node_1227845970559210097&#45;&gt;node_d6555065156464757206 -->\n",
119122
" <g class=\"edge\" id=\"edge1\">\n",
120123
" <title>\n",
121-
" node_7350196740953114244-&gt;node_d8297656796017466239\n",
124+
" node_1227845970559210097-&gt;node_d6555065156464757206\n",
122125
" </title>\n",
123126
" <path d=\"M149.2,-43C161.27,-43 174.06,-43 186.75,-43\" fill=\"none\" stroke=\"black\"/>\n",
124127
" <polygon fill=\"black\" points=\"186.89,-46.5 196.89,-43 186.89,-39.5 186.89,-46.5\" stroke=\"black\"/>\n",
125128
" <text font-family=\"Helvetica,Arial,sans-serif\" font-size=\"14.00\" text-anchor=\"middle\" x=\"173\" y=\"-46.8\">\n",
126129
" df\n",
127130
" </text>\n",
128131
" </g>\n",
129-
" <!-- node_6080578892279087024 -->\n",
132+
" <!-- node_d985045474409174850 -->\n",
130133
" <g class=\"node\" id=\"node2\">\n",
131134
" <title>\n",
132-
" node_6080578892279087024\n",
135+
" node_d985045474409174850\n",
133136
" </title>\n",
134137
" <g cursor=\"pointer\" id=\"a_node2\" onclick=\" window.open('', '_blank', 'width=800,height=200').document.body.innerHTML = `&lt;table border=&quot;1&quot; class=&quot;dataframe&quot;&gt;\n",
135138
" &lt;thead&gt;\n",
@@ -185,31 +188,31 @@
185188
" </text>\n",
186189
" </g>\n",
187190
" </g>\n",
188-
" <!-- node_3009124502050678936 -->\n",
191+
" <!-- node_1904243765544493615 -->\n",
189192
" <g class=\"node\" id=\"node8\">\n",
190193
" <title>\n",
191-
" node_3009124502050678936\n",
194+
" node_1904243765544493615\n",
192195
" </title>\n",
193196
" <polygon fill=\"lightgrey\" points=\"689,-61 583,-61 583,-25 689,-25 689,-61\" stroke=\"transparent\"/>\n",
194197
" <text font-family=\"Helvetica,Arial,sans-serif\" font-size=\"14.00\" text-anchor=\"start\" x=\"595\" y=\"-38.8\">\n",
195198
" dropna(df, ...)\n",
196199
" </text>\n",
197200
" </g>\n",
198-
" <!-- node_6080578892279087024&#45;&gt;node_3009124502050678936 -->\n",
201+
" <!-- node_d985045474409174850&#45;&gt;node_1904243765544493615 -->\n",
199202
" <g class=\"edge\" id=\"edge3\">\n",
200203
" <title>\n",
201-
" node_6080578892279087024-&gt;node_3009124502050678936\n",
204+
" node_d985045474409174850-&gt;node_1904243765544493615\n",
202205
" </title>\n",
203206
" <path d=\"M535.39,-43C547.71,-43 560.55,-43 572.76,-43\" fill=\"none\" stroke=\"black\"/>\n",
204207
" <polygon fill=\"black\" points=\"572.85,-46.5 582.85,-43 572.85,-39.5 572.85,-46.5\" stroke=\"black\"/>\n",
205208
" <text font-family=\"Helvetica,Arial,sans-serif\" font-size=\"14.00\" text-anchor=\"middle\" x=\"559\" y=\"-46.8\">\n",
206209
" df\n",
207210
" </text>\n",
208211
" </g>\n",
209-
" <!-- node_d8893236431485775077 -->\n",
212+
" <!-- node_d4843245280907125723 -->\n",
210213
" <g class=\"node\" id=\"node3\">\n",
211214
" <title>\n",
212-
" node_d8893236431485775077\n",
215+
" node_d4843245280907125723\n",
213216
" </title>\n",
214217
" <g cursor=\"pointer\" id=\"a_node3\" onclick=\" window.open('', '_blank', 'width=800,height=200').document.body.innerHTML = `&lt;table border=&quot;1&quot; class=&quot;dataframe&quot;&gt;\n",
215218
" &lt;thead&gt;\n",
@@ -259,31 +262,31 @@
259262
" </text>\n",
260263
" </g>\n",
261264
" </g>\n",
262-
" <!-- node_d2412246820646920085 -->\n",
265+
" <!-- node_d2524308946617689270 -->\n",
263266
" <g class=\"node\" id=\"node9\">\n",
264267
" <title>\n",
265-
" node_d2412246820646920085\n",
268+
" node_d2524308946617689270\n",
266269
" </title>\n",
267270
" <polygon fill=\"lightgrey\" points=\"1069,-61 907,-61 907,-25 1069,-25 1069,-61\" stroke=\"transparent\"/>\n",
268271
" <text font-family=\"Helvetica,Arial,sans-serif\" font-size=\"14.00\" text-anchor=\"start\" x=\"919\" y=\"-38.8\">\n",
269272
" rename_column(df, ...)\n",
270273
" </text>\n",
271274
" </g>\n",
272-
" <!-- node_d8893236431485775077&#45;&gt;node_d2412246820646920085 -->\n",
275+
" <!-- node_d4843245280907125723&#45;&gt;node_d2524308946617689270 -->\n",
273276
" <g class=\"edge\" id=\"edge5\">\n",
274277
" <title>\n",
275-
" node_d8893236431485775077-&gt;node_d2412246820646920085\n",
278+
" node_d4843245280907125723-&gt;node_d2524308946617689270\n",
276279
" </title>\n",
277280
" <path d=\"M859.01,-43C871.12,-43 883.97,-43 896.66,-43\" fill=\"none\" stroke=\"black\"/>\n",
278281
" <polygon fill=\"black\" points=\"896.79,-46.5 906.79,-43 896.79,-39.5 896.79,-46.5\" stroke=\"black\"/>\n",
279282
" <text font-family=\"Helvetica,Arial,sans-serif\" font-size=\"14.00\" text-anchor=\"middle\" x=\"883\" y=\"-46.8\">\n",
280283
" df\n",
281284
" </text>\n",
282285
" </g>\n",
283-
" <!-- node_d2694599867417863562 -->\n",
286+
" <!-- node_d2973361350997812975 -->\n",
284287
" <g class=\"node\" id=\"node4\">\n",
285288
" <title>\n",
286-
" node_d2694599867417863562\n",
289+
" node_d2973361350997812975\n",
287290
" </title>\n",
288291
" <g cursor=\"pointer\" id=\"a_node4\" onclick=\" window.open('', '_blank', 'width=800,height=200').document.body.innerHTML = `&lt;table border=&quot;1&quot; class=&quot;dataframe&quot;&gt;\n",
289292
" &lt;thead&gt;\n",
@@ -333,31 +336,31 @@
333336
" </text>\n",
334337
" </g>\n",
335338
" </g>\n",
336-
" <!-- node_2452196101976357619 -->\n",
339+
" <!-- node_7444217268540326659 -->\n",
337340
" <g class=\"node\" id=\"node10\">\n",
338341
" <title>\n",
339-
" node_2452196101976357619\n",
342+
" node_7444217268540326659\n",
340343
" </title>\n",
341344
" <polygon fill=\"lightgrey\" points=\"1449,-61 1287,-61 1287,-25 1449,-25 1449,-61\" stroke=\"transparent\"/>\n",
342345
" <text font-family=\"Helvetica,Arial,sans-serif\" font-size=\"14.00\" text-anchor=\"start\" x=\"1299\" y=\"-38.8\">\n",
343346
" rename_column(df, ...)\n",
344347
" </text>\n",
345348
" </g>\n",
346-
" <!-- node_d2694599867417863562&#45;&gt;node_2452196101976357619 -->\n",
349+
" <!-- node_d2973361350997812975&#45;&gt;node_7444217268540326659 -->\n",
347350
" <g class=\"edge\" id=\"edge7\">\n",
348351
" <title>\n",
349-
" node_d2694599867417863562-&gt;node_2452196101976357619\n",
352+
" node_d2973361350997812975-&gt;node_7444217268540326659\n",
350353
" </title>\n",
351354
" <path d=\"M1239.01,-43C1251.12,-43 1263.97,-43 1276.66,-43\" fill=\"none\" stroke=\"black\"/>\n",
352355
" <polygon fill=\"black\" points=\"1276.79,-46.5 1286.79,-43 1276.79,-39.5 1276.79,-46.5\" stroke=\"black\"/>\n",
353356
" <text font-family=\"Helvetica,Arial,sans-serif\" font-size=\"14.00\" text-anchor=\"middle\" x=\"1263\" y=\"-46.8\">\n",
354357
" df\n",
355358
" </text>\n",
356359
" </g>\n",
357-
" <!-- node_d7379048532167015300 -->\n",
360+
" <!-- node_d6383385337535756017 -->\n",
358361
" <g class=\"node\" id=\"node5\">\n",
359362
" <title>\n",
360-
" node_d7379048532167015300\n",
363+
" node_d6383385337535756017\n",
361364
" </title>\n",
362365
" <g cursor=\"pointer\" id=\"a_node5\" onclick=\" window.open('', '_blank', 'width=800,height=200').document.body.innerHTML = `&lt;table border=&quot;1&quot; class=&quot;dataframe&quot;&gt;\n",
363366
" &lt;thead&gt;\n",
@@ -407,31 +410,31 @@
407410
" </text>\n",
408411
" </g>\n",
409412
" </g>\n",
410-
" <!-- node_d124244031066527821 -->\n",
413+
" <!-- node_8835876692805255874 -->\n",
411414
" <g class=\"node\" id=\"node11\">\n",
412415
" <title>\n",
413-
" node_d124244031066527821\n",
416+
" node_8835876692805255874\n",
414417
" </title>\n",
415418
" <polygon fill=\"lightgrey\" points=\"1805,-61 1667,-61 1667,-25 1805,-25 1805,-61\" stroke=\"transparent\"/>\n",
416419
" <text font-family=\"Helvetica,Arial,sans-serif\" font-size=\"14.00\" text-anchor=\"start\" x=\"1679\" y=\"-38.8\">\n",
417420
" add_column(df, ...)\n",
418421
" </text>\n",
419422
" </g>\n",
420-
" <!-- node_d7379048532167015300&#45;&gt;node_d124244031066527821 -->\n",
423+
" <!-- node_d6383385337535756017&#45;&gt;node_8835876692805255874 -->\n",
421424
" <g class=\"edge\" id=\"edge9\">\n",
422425
" <title>\n",
423-
" node_d7379048532167015300-&gt;node_d124244031066527821\n",
426+
" node_d6383385337535756017-&gt;node_8835876692805255874\n",
424427
" </title>\n",
425428
" <path d=\"M1619.03,-43C1631.28,-43 1644.21,-43 1656.81,-43\" fill=\"none\" stroke=\"black\"/>\n",
426429
" <polygon fill=\"black\" points=\"1656.82,-46.5 1666.82,-43 1656.82,-39.5 1656.82,-46.5\" stroke=\"black\"/>\n",
427430
" <text font-family=\"Helvetica,Arial,sans-serif\" font-size=\"14.00\" text-anchor=\"middle\" x=\"1643\" y=\"-46.8\">\n",
428431
" df\n",
429432
" </text>\n",
430433
" </g>\n",
431-
" <!-- node_4691575139530742294 -->\n",
434+
" <!-- node_4042741298201303901 -->\n",
432435
" <g class=\"node\" id=\"node6\">\n",
433436
" <title>\n",
434-
" node_4691575139530742294\n",
437+
" node_4042741298201303901\n",
435438
" </title>\n",
436439
" <g cursor=\"pointer\" id=\"a_node6\" onclick=\" window.open('', '_blank', 'width=800,height=200').document.body.innerHTML = `&lt;table border=&quot;1&quot; class=&quot;dataframe&quot;&gt;\n",
437440
" &lt;thead&gt;\n",
@@ -485,42 +488,42 @@
485488
" </text>\n",
486489
" </g>\n",
487490
" </g>\n",
488-
" <!-- node_d8297656796017466239&#45;&gt;node_6080578892279087024 -->\n",
491+
" <!-- node_d6555065156464757206&#45;&gt;node_d985045474409174850 -->\n",
489492
" <g class=\"edge\" id=\"edge2\">\n",
490493
" <title>\n",
491-
" node_d8297656796017466239-&gt;node_6080578892279087024\n",
494+
" node_d6555065156464757206-&gt;node_d985045474409174850\n",
492495
" </title>\n",
493496
" <path d=\"M365.25,-43C374.05,-43 382.95,-43 391.65,-43\" fill=\"none\" stroke=\"black\"/>\n",
494497
" <polygon fill=\"black\" points=\"391.76,-46.5 401.76,-43 391.76,-39.5 391.76,-46.5\" stroke=\"black\"/>\n",
495498
" </g>\n",
496-
" <!-- node_3009124502050678936&#45;&gt;node_d8893236431485775077 -->\n",
499+
" <!-- node_1904243765544493615&#45;&gt;node_d4843245280907125723 -->\n",
497500
" <g class=\"edge\" id=\"edge4\">\n",
498501
" <title>\n",
499-
" node_3009124502050678936-&gt;node_d8893236431485775077\n",
502+
" node_1904243765544493615-&gt;node_d4843245280907125723\n",
500503
" </title>\n",
501504
" <path d=\"M689.31,-43C697.83,-43 706.81,-43 715.76,-43\" fill=\"none\" stroke=\"black\"/>\n",
502505
" <polygon fill=\"black\" points=\"715.82,-46.5 725.82,-43 715.82,-39.5 715.82,-46.5\" stroke=\"black\"/>\n",
503506
" </g>\n",
504-
" <!-- node_d2412246820646920085&#45;&gt;node_d2694599867417863562 -->\n",
507+
" <!-- node_d2524308946617689270&#45;&gt;node_d2973361350997812975 -->\n",
505508
" <g class=\"edge\" id=\"edge6\">\n",
506509
" <title>\n",
507-
" node_d2412246820646920085-&gt;node_d2694599867417863562\n",
510+
" node_d2524308946617689270-&gt;node_d2973361350997812975\n",
508511
" </title>\n",
509512
" <path d=\"M1069.05,-43C1077.86,-43 1086.81,-43 1095.57,-43\" fill=\"none\" stroke=\"black\"/>\n",
510513
" <polygon fill=\"black\" points=\"1095.74,-46.5 1105.74,-43 1095.74,-39.5 1095.74,-46.5\" stroke=\"black\"/>\n",
511514
" </g>\n",
512-
" <!-- node_2452196101976357619&#45;&gt;node_d7379048532167015300 -->\n",
515+
" <!-- node_7444217268540326659&#45;&gt;node_d6383385337535756017 -->\n",
513516
" <g class=\"edge\" id=\"edge8\">\n",
514517
" <title>\n",
515-
" node_2452196101976357619-&gt;node_d7379048532167015300\n",
518+
" node_7444217268540326659-&gt;node_d6383385337535756017\n",
516519
" </title>\n",
517520
" <path d=\"M1449.05,-43C1457.86,-43 1466.81,-43 1475.57,-43\" fill=\"none\" stroke=\"black\"/>\n",
518521
" <polygon fill=\"black\" points=\"1475.74,-46.5 1485.74,-43 1475.74,-39.5 1475.74,-46.5\" stroke=\"black\"/>\n",
519522
" </g>\n",
520-
" <!-- node_d124244031066527821&#45;&gt;node_4691575139530742294 -->\n",
523+
" <!-- node_8835876692805255874&#45;&gt;node_4042741298201303901 -->\n",
521524
" <g class=\"edge\" id=\"edge10\">\n",
522525
" <title>\n",
523-
" node_d124244031066527821-&gt;node_4691575139530742294\n",
526+
" node_8835876692805255874-&gt;node_4042741298201303901\n",
524527
" </title>\n",
525528
" <path d=\"M1805.37,-43C1814.02,-43 1822.92,-43 1831.69,-43\" fill=\"none\" stroke=\"black\"/>\n",
526529
" <polygon fill=\"black\" points=\"1831.89,-46.5 1841.89,-43 1831.89,-39.5 1831.89,-46.5\" stroke=\"black\"/>\n",
@@ -534,6 +537,13 @@
534537
},
535538
"metadata": {},
536539
"output_type": "display_data"
540+
},
541+
{
542+
"name": "stdout",
543+
"output_type": "stream",
544+
"text": [
545+
"80990326-1ef0-4588-8e28-0f377bd4b5bd\n"
546+
]
537547
}
538548
],
539549
"source": [
@@ -560,14 +570,63 @@
560570
" .add_column(\"Google\", [450.0, 550.0, 800.0])\n",
561571
" )\n",
562572
"\n",
563-
"pyjviz.show(vertical=False)\n"
573+
"pyjviz.show(vertical=False)\n",
574+
"print(pyjviz.nb_utils.get_cell_id())"
564575
]
565576
},
566577
{
567578
"cell_type": "code",
568-
"execution_count": null,
579+
"execution_count": 2,
569580
"id": "db8f1e76-5296-4c61-aed9-f22d5f3a6ccf",
570581
"metadata": {},
582+
"outputs": [
583+
{
584+
"name": "stdout",
585+
"output_type": "stream",
586+
"text": [
587+
"pre_run_cell: This function will be called before every cell is executed.\n",
588+
"cell_id: db8f1e76-5296-4c61-aed9-f22d5f3a6ccf\n"
589+
]
590+
},
591+
{
592+
"data": {
593+
"text/plain": [
594+
"'db8f1e76-5296-4c61-aed9-f22d5f3a6ccf'"
595+
]
596+
},
597+
"execution_count": 2,
598+
"metadata": {},
599+
"output_type": "execute_result"
600+
}
601+
],
602+
"source": [
603+
"pyjviz.nb_utils.get_cell_id()"
604+
]
605+
},
606+
{
607+
"cell_type": "code",
608+
"execution_count": 6,
609+
"id": "7e2a1879-97b8-41e5-a493-96a64d07188b",
610+
"metadata": {},
611+
"outputs": [
612+
{
613+
"name": "stdout",
614+
"output_type": "stream",
615+
"text": [
616+
"pre_run_cell: This function will be called before every cell is executed.\n",
617+
"hi 7e2a1879-97b8-41e5-a493-96a64d07188b\n"
618+
]
619+
}
620+
],
621+
"source": [
622+
"print('hi', pyjviz.nb_utils.get_cell_id())"
623+
]
624+
},
625+
{
626+
"cell_type": "code",
627+
"execution_count": null,
628+
"id": "d06bd090-f942-4cf0-9ed6-a9d756a5b973",
629+
"metadata": {},
571630
"outputs": [],
572631
"source": []
573632
}

0 commit comments

Comments
 (0)