dir and file operate

steven.chen3 · steven.chen3 · commit a119f431097a · 2022-10-26T20:39:51.000-07:00
diff --git a/dir_operate/main.py b/dir_operate/main.py
@@ -0,0 +1,30 @@
+import os
+
+
+def cwd():
+    print(os.getcwd())
+    os.chdir("mysub")
+    print(os.getcwd())
+
+    # os.mkdir("mysub2")
+
+    # os.rename("mysub2", "mysub3")
+
+    os.rmdir("mysub3")
+
+
+def print_dir(root_dir: str):
+    for root, dir_names, file_names in os.walk(root_dir):
+        for dir in dir_names:
+            print(f"{root}/{dir}")
+
+        for file in file_names:
+            print(f"{root}/{file}")
+
+
+def main():
+    print_dir("../../python_tutorial")
+
+
+if __name__ == "__main__":
+    main()
diff --git a/dir_operate/search_file.py b/dir_operate/search_file.py
@@ -0,0 +1,24 @@
+import os
+import sys
+
+
+def search_file(root_dir: str) -> list:
+    result = []
+    for dir, dir_names, file_names in os.walk(root_dir):
+        for file in file_names:
+            if file.lower().endswith(".txt"):
+                result.append(f"{dir}/{file}")
+
+    return result
+
+
+def main():
+    if len(sys.argv) <= 1:
+        print("Please pass dir as an argument")
+    else:
+        file_list = search_file(sys.argv[1])
+        print(file_list)
+
+
+if __name__ == "__main__":
+    main()
