ref: Use pathlib read/write functions (#4177)

Use pathlib read/write functions

src/backend/base/langflow/base/data/utils.py

@@ -109,16 +109,14 @@ def partition_file_to_data(file_path: str, *, silent_errors: bool) -> Data | Non
 
 def read_text_file(file_path: str) -> str:
     _file_path = Path(file_path)
-    with _file_path.open("rb") as f:
+    raw_data = _file_path.read_bytes()
-        raw_data = f.read()
     result = chardet.detect(raw_data)
     encoding = result["encoding"]
 
     if encoding in {"Windows-1252", "Windows-1254", "MacRoman"}:
         encoding = "utf-8"
 
-    with _file_path.open(encoding=encoding) as f:
+    return _file_path.read_text(encoding=encoding)
-        return f.read()
 
 
 def read_docx_file(file_path: str) -> str:

src/backend/base/langflow/components/agents/JsonAgent.py

@@ -23,8 +23,8 @@ class JsonAgentComponent(LCAgentComponent):
 
     def build_agent(self) -> AgentExecutor:
         path = Path(self.path)
-        if self.path.endswith("yaml") or self.path.endswith("yml"):
+        if path.suffix in ("yaml", "yml"):
-            with path.open() as file:
+            with path.open(encoding="utf-8") as file:
                 yaml_dict = yaml.load(file, Loader=yaml.FullLoader)
             spec = JsonSpec(dict_=yaml_dict)
         else:

src/backend/base/langflow/components/agents/OpenAPIAgent.py

@@ -25,8 +25,8 @@ class OpenAPIAgentComponent(LCAgentComponent):
 
     def build_agent(self) -> AgentExecutor:
         path = Path(self.path)
-        if self.path.endswith("yaml") or self.path.endswith("yml"):
+        if path.suffix in ("yaml", "yml"):
-            with path.open() as file:
+            with path.open(encoding="utf-8") as file:
                 yaml_dict = yaml.load(file, Loader=yaml.FullLoader)
             spec = JsonSpec(dict_=yaml_dict)
         else:

src/backend/base/langflow/components/documentloaders/GitLoader.py

@@ -88,8 +88,7 @@ class GitLoaderComponent(Component):
             content_regex = re.compile(content_filter_pattern)
 
             def content_filter(file_path: Path) -> bool:
-                with file_path.open("r", encoding="utf-8", errors="ignore") as file:
+                content = file_path.read_text(encoding="utf-8", errors="ignore")
-                    content = file.read()
                 return bool(content_regex.search(content))
 
             file_filters.append(content_filter)

src/backend/base/langflow/components/helpers/JSONtoData.py

@@ -55,16 +55,14 @@ class JSONToDataComponent(Component):
                 if file_path.suffix.lower() != ".json":
                     self.status = "The provided file must be a JSON file."
                 else:
-                    with file_path.open(encoding="utf-8") as jsonfile:
+                    json_data = file_path.read_text(encoding="utf-8")
-                        json_data = jsonfile.read()
 
             elif self.json_path:
                 file_path = Path(self.json_path)
                 if file_path.suffix.lower() != ".json":
                     self.status = "The provided file must be a JSON file."
                 else:
-                    with file_path.open(encoding="utf-8") as jsonfile:
+                    json_data = file_path.read_text(encoding="utf-8")
-                        json_data = jsonfile.read()
 
             else:
                 json_data = self.json_string

src/backend/base/langflow/components/vectorstores/Redis.py

@@ -53,8 +53,7 @@ class RedisVectorStoreComponent(LCVectorStoreComponent):
                 documents.append(_input.to_lc_document())
             else:
                 documents.append(_input)
-        with Path("docuemnts.txt").open("w", encoding="utf-8") as f:
+        Path("docuemnts.txt").write_text(str(documents), encoding="utf-8")
-            f.write(str(documents))
 
         if not documents:
             if self.schema is None:

src/backend/base/langflow/custom/code_parser/code_parser.py

@@ -32,8 +32,7 @@ def find_class_ast_node(class_obj):
         return None, []
 
     # Read the source code from the file
-    with Path(source_file).open(encoding="utf-8") as file:
+    source_code = Path(source_file).read_text(encoding="utf-8")
-        source_code = file.read()
 
     # Parse the source code into an AST
     tree = ast.parse(source_code)

src/backend/base/langflow/initial_setup/setup.py

@@ -362,9 +362,9 @@ def load_starter_projects(retries=3, delay=1) -> list[tuple[Path, dict]]:
     for file in folder.glob("*.json"):
         attempt = 0
         while attempt < retries:
-            with file.open(encoding="utf-8") as f:
+            content = file.read_text(encoding="utf-8")
             try:
-                    project = orjson.loads(f.read())
+                project = orjson.loads(content)
                 starter_projects.append((file, project))
                 logger.info(f"Loaded starter project {file}")
                 break  # Break if load is successful
@@ -427,8 +427,7 @@ def get_project_data(project):
 
 def update_project_file(project_path: Path, project: dict, updated_project_data):
     project["data"] = updated_project_data
-    with project_path.open("w", encoding="utf-8") as f:
+    project_path.write_text(orjson.dumps(project, option=ORJSON_OPTIONS).decode(), encoding="utf-8")
-        f.write(orjson.dumps(project, option=ORJSON_OPTIONS).decode())
     logger.info(f"Updated starter project {project['name']} file")
 
 
@@ -539,8 +538,8 @@ def load_flows_from_directory():
             if f.suffix != ".json":
                 continue
             logger.info(f"Loading flow from file: {f.name}")
-            with f.open(encoding="utf-8") as file:
+            content = f.read_text(encoding="utf-8")
-                flow = orjson.loads(file.read())
+            flow = orjson.loads(content)
             no_json_name = f.stem
             flow_endpoint_name = flow.get("endpoint_name")
             if _is_valid_uuid(no_json_name):

src/backend/base/langflow/interface/utils.py

@@ -21,7 +21,7 @@ def load_file_into_dict(file_path: str) -> dict:
         raise FileNotFoundError(msg)
 
     # Files names are UUID, so we can't find the extension
-    with _file_path.open() as file:
+    with _file_path.open(encoding="utf-8") as file:
         try:
             data = json.load(file)
         except json.JSONDecodeError:

src/backend/base/langflow/services/cache/utils.py

@@ -104,8 +104,7 @@ def save_binary_file(content: str, file_name: str, accepted_types: list[str]) ->
     file_path = cache_path / file_name
 
     # Save the binary content to the file
-    with file_path.open("wb") as file:
+    file_path.write_bytes(decoded_bytes)
-        file.write(decoded_bytes)
 
     return str(file_path)
 

src/backend/base/langflow/services/database/service.py

@@ -178,7 +178,7 @@ class DatabaseService(Service):
         # which is a buffer
         # I don't want to output anything
         # subprocess.DEVNULL is an int
-        with (self.script_location / "alembic.log").open("w") as buffer:
+        with (self.script_location / "alembic.log").open("w", encoding="utf-8") as buffer:
             alembic_cfg = Config(stdout=buffer)
             # alembic_cfg.attributes["connection"] = session
             alembic_cfg.set_main_option("script_location", str(self.script_location))

src/backend/base/langflow/services/settings/base.py

@@ -382,7 +382,7 @@ def load_settings_from_yaml(file_path: str) -> Settings:
     else:
         _file_path = Path(file_path)
 
-    with _file_path.open() as f:
+    with _file_path.open(encoding="utf-8") as f:
         settings_dict = yaml.safe_load(f)
         settings_dict = {k.upper(): v for k, v in settings_dict.items()}
 

src/backend/base/langflow/services/settings/manager.py

@@ -28,7 +28,7 @@ class SettingsService(Service):
         else:
             _file_path = Path(file_path)
 
-        with _file_path.open() as f:
+        with _file_path.open(encoding="utf-8") as f:
             settings_dict = yaml.safe_load(f)
             settings_dict = {k.upper(): v for k, v in settings_dict.items()}
 

src/backend/base/langflow/services/settings/utils.py

@@ -29,8 +29,7 @@ def set_secure_permissions(file_path: Path):
 
 
 def write_secret_to_file(path: Path, value: str) -> None:
-    with path.open("wb") as f:
+    path.write_text(value, encoding="utf-8")
-        f.write(value.encode("utf-8"))
     try:
         set_secure_permissions(path)
     except Exception:  # noqa: BLE001
@@ -38,5 +37,4 @@ def write_secret_to_file(path: Path, value: str) -> None:
 
 
 def read_secret_from_file(path: Path) -> str:
-    with path.open("r") as f:
+    return path.read_text(encoding="utf-8")
-        return f.read()

src/backend/base/langflow/services/storage/local.py

@@ -34,8 +34,7 @@ class LocalStorageService(StorageService):
         file_path = folder_path / file_name
 
         def write_file(file_path: Path, data: bytes) -> None:
-            with Path(file_path).open("wb") as f:
+            file_path.write_bytes(data)
-                f.write(data)
 
         try:
             await asyncio.get_event_loop().run_in_executor(None, write_file, file_path, data)
@@ -59,8 +58,7 @@ class LocalStorageService(StorageService):
             raise FileNotFoundError(msg)
 
         def read_file(file_path: Path) -> bytes:
-            with Path(file_path).open("rb") as f:
+            return file_path.read_bytes()
-                return f.read()
 
         content = await asyncio.get_event_loop().run_in_executor(None, read_file, file_path)
         logger.debug(f"File {file_name} retrieved successfully from flow {flow_id}.")