feat: implement change_entity_label() (first working version)

# Last 3 commits:

# 6d41820 feat: implement `change_entity_label()` (first working version)
# 1a3943d feat: implement `change_entity_label()` (wip)
# 11b3b6d feat (test_refactor_tools): add `make_temp_copy_of_file()`

# Last 3 commits:

# f2b9050 feat: implement `change_entity_label()` (wip) 
# 7af0110 feat (test_refactor_tools): add `make_temp_copy_of_file()` 
# 2db8f37 feat (script.py): add `--refactor-entity-label` option stub

Author: cknoll

src/pyirk/refactor_tools.py

@@ -38,7 +38,16 @@ def change_entity_label(key, new_label, fpath):
 
     src_lines[start_idx + i] = new_defining_line
 
-    with open(fpath, "w", encoding="utf-8") as fp:
-        fp.writelines(src_lines)
+    new_src = "".join(src_lines)
+
+    replacements = [
+        (f'{key}["{original_label}"]', f'{key}["{new_label}"]'),
+        (f"{key}['{original_label}']", f"{key}['{new_label}']"),
+        # TODO: add underscore_version
+    ]
 
-    IPS()
+    for rplm in replacements:
+        new_src = new_src.replace(*rplm)
+
+    with open(fpath, "w", encoding="utf-8") as fp:
+        fp.write(new_src)

src/pyirk/script.py

@@ -197,9 +197,8 @@ def main():
     if args.refactor_entity_label:
         from . import refactor_tools as rt
         assert args.inputfile is not None
-        rt.change_entity_label(1, 2, args.inputfile)
-
-        IPS()
+        assert len(args.refactor_entity_label) == 2
+        rt.change_entity_label(*args.refactor_entity_label, args.inputfile)
         exit()
 
     if args.version:

tests/test_refactor_tools.py

@@ -1,3 +1,4 @@
+import os
 import unittest
 import tempfile
 import shutil
@@ -17,19 +18,31 @@
 
 # noinspection PyPep8Naming
 class Test_01_Script(HousekeeperMixin, unittest.TestCase):
-    def test_rt_a01__(self):
+    def test_rt_a01__change_entity_label(self):
 
-        tmp_mod_fpath0 = make_temp_copy_of_file(ocse_subset_agents_path)
         tmp_mod_fpath = make_temp_copy_of_file(ocse_subset_agents_path)
 
         # change I9942["Stanford University"] to I9942["Renamed Stanford University"]
         rt.change_entity_label(key="I9942", new_label="Renamed Stanford University", fpath=tmp_mod_fpath)
 
-        # for manually checking the difference
-        import os
-        os.system(f"kdiff3 {tmp_mod_fpath0} {tmp_mod_fpath}")
+        manually_checking_difference = False
+        if manually_checking_difference:
+            tmp_mod_fpath0 = make_temp_copy_of_file(ocse_subset_agents_path)
+            os.system(f"kdiff3 {tmp_mod_fpath0} {tmp_mod_fpath}")
+
+
+    def test_rt_a02__change_entity_label_from_cli(self):
+
+        tmp_mod_fpath = make_temp_copy_of_file(ocse_subset_agents_path)
+
+        cmd = f'pyirk --refactor-entity-label I9942 "Renamed Stanford University" {tmp_mod_fpath}'
+        os.system(cmd)
 
 
+        manually_checking_difference = False
+        if manually_checking_difference:
+            tmp_mod_fpath0 = make_temp_copy_of_file(ocse_subset_agents_path)
+            os.system(f"kdiff3 {tmp_mod_fpath0} {tmp_mod_fpath}")
 #
 # Auxiliary functions:
 #