| import os | |
| target_dir = r"D:\Desktop_March_26\LYZR\graph-RAG\src\graph_rag_service" | |
| for root, dirs, files in os.walk(target_dir): | |
| for f in files: | |
| if f.endswith(".py"): | |
| p = os.path.join(root, f) | |
| with open(p, "r", encoding="utf-8") as file: | |
| content = file.read() | |
| if "datetime.utcnow()" in content: | |
| content = content.replace("datetime.utcnow()", "datetime.now(timezone.utc).replace(tzinfo=None)") | |
| if "from datetime import datetime" in content and "timezone" not in content: | |
| content = content.replace("from datetime import datetime", "from datetime import datetime, timezone") | |
| elif "import datetime" in content and "timezone" not in content: | |
| content = "from datetime import timezone\n" + content | |
| with open(p, "w", encoding="utf-8") as file: | |
| file.write(content) | |
| print(f"Updated {p}") | |