lookyloo/tools/change_captures_dir.py

30 lines
926 B
Python
Raw Permalink Normal View History

#!/usr/bin/env python3
from datetime import datetime
from pathlib import Path
from redis import Redis
2021-10-18 13:06:43 +02:00
from lookyloo.default import safe_create_dir, get_socket_path
from lookyloo.helpers import get_captures_dir
2024-01-12 17:15:41 +01:00
def rename_captures() -> None:
r = Redis(unix_socket_path=get_socket_path('cache'))
capture_dir: Path = get_captures_dir()
for uuid_path in capture_dir.glob('*/uuid'):
with uuid_path.open() as f:
uuid = f.read()
dir_key = r.hget('lookup_dirs', uuid)
if dir_key:
2021-08-27 14:27:38 +02:00
r.hdel('lookup_dirs', uuid)
r.delete(dir_key)
timestamp = datetime.strptime(uuid_path.parent.name, '%Y-%m-%dT%H:%M:%S.%f')
dest_dir = capture_dir / str(timestamp.year) / f'{timestamp.month:02}'
safe_create_dir(dest_dir)
uuid_path.parent.rename(dest_dir / uuid_path.parent.name)
if __name__ == '__main__':
rename_captures()