purge_cache.py 1.3 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647
  1. #!/usr/bin/env python3
  2. import argparse
  3. import glob
  4. import os
  5. if __name__ != "__main__":
  6. raise ImportError(f"{__name__} should not be used as a module.")
  7. def main():
  8. parser = argparse.ArgumentParser(description="Cleanup old cache files")
  9. parser.add_argument("timestamp", type=int, help="Unix timestamp cutoff")
  10. parser.add_argument("directory", help="Path to cache directory")
  11. args = parser.parse_args()
  12. ret = 0
  13. # TODO: Convert to non-hardcoded path
  14. if os.path.exists("redundant.txt"):
  15. with open("redundant.txt") as redundant:
  16. for item in map(str.strip, redundant):
  17. if os.path.isfile(item):
  18. try:
  19. os.remove(item)
  20. except OSError:
  21. print(f'Failed to handle "{item}"; skipping.')
  22. ret += 1
  23. for file in glob.glob(os.path.join(args.directory, "*", "*")):
  24. try:
  25. if os.path.getatime(file) < args.timestamp:
  26. os.remove(file)
  27. except OSError:
  28. print(f'Failed to handle "{file}"; skipping.')
  29. ret += 1
  30. return ret
  31. try:
  32. raise SystemExit(main())
  33. except KeyboardInterrupt:
  34. import signal
  35. signal.signal(signal.SIGINT, signal.SIG_DFL)
  36. os.kill(os.getpid(), signal.SIGINT)