Fix atexit registration for JSON writing to ensure data is saved on exit

This commit is contained in:
fzzinchemical
2025-11-13 20:50:07 +01:00
parent a05b3a37f0
commit 7ecccda9d8

View File

@@ -33,6 +33,8 @@ def write_json(file_path, data):
with open(file_path, "w", encoding="utf-8") as f: with open(file_path, "w", encoding="utf-8") as f:
json.dump(data, f, ensure_ascii=False, indent=4) json.dump(data, f, ensure_ascii=False, indent=4)
atexit.register(lambda: write_json(OUTPUT_FILE, {"tweets": _results}))
async def main(): async def main():
global _results global _results
api = API() # or API("path-to.db") default is `accounts.db` api = API() # or API("path-to.db") default is `accounts.db`
@@ -63,7 +65,6 @@ async def main():
await asyncio.sleep(random.uniform(17, 31)) await asyncio.sleep(random.uniform(17, 31))
atexit.register(lambda: write_json(OUTPUT_FILE, {"tweets": _results}))
if __name__ == "__main__": if __name__ == "__main__":
asyncio.run(main()) asyncio.run(main())