Spaces:
Sleeping
Sleeping
File size: 1,240 Bytes
4620bae | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 | import os
import shutil
import sys
from pathlib import Path
ROOT = Path(__file__).resolve().parents[1]
DATA_DIR = ROOT / 'data'
OUTPUT_DIR = ROOT / 'talentsonar' / 'output'
FILES_TO_REMOVE = [
DATA_DIR / 'candidates_data.json',
DATA_DIR / 'candidates.csv',
DATA_DIR / 'discovered_candidates.json',
DATA_DIR / 'job_postings.json',
DATA_DIR / 'test_results.json',
DATA_DIR / 'invitations.json',
]
def remove_file(p: Path):
try:
if p.exists():
p.unlink()
print(f"Removed: {p}")
except Exception as e:
print(f"Could not remove {p}: {e}")
def clean_data():
print(f"Cleaning data directory: {DATA_DIR}")
DATA_DIR.mkdir(parents=True, exist_ok=True)
for f in FILES_TO_REMOVE:
remove_file(f)
# Optionally clean output artifacts
if OUTPUT_DIR.exists():
try:
shutil.rmtree(OUTPUT_DIR)
print(f"Removed directory: {OUTPUT_DIR}")
except Exception as e:
print(f"Could not remove {OUTPUT_DIR}: {e}")
OUTPUT_DIR.mkdir(parents=True, exist_ok=True)
print("\nCleanup complete. Fresh start ready.")
if __name__ == '__main__':
clean_data()
|