forked from Haidra-Org/AI-Horde
-
Notifications
You must be signed in to change notification settings - Fork 0
/
prune_stale_images.py
38 lines (35 loc) · 1.35 KB
/
prune_stale_images.py
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
import logging
import boto3
import time
from datetime import datetime, timedelta, timezone
from tqdm import tqdm
from concurrent.futures import ThreadPoolExecutor
from botocore.exceptions import ClientError
from dotenv import load_dotenv
import patreon
import os
load_dotenv()
sr3 = boto3.resource('s3', endpoint_url="https://a223539ccf6caa2d76459c9727d276e6.r2.cloudflarestorage.com")
while True:
try:
cutoff_time = datetime.now(timezone.utc) - timedelta(minutes=120)
print("Starting Next Cleanup Iteration...")
for bucket in [
sr3.Bucket('stable-horde'),
sr3.Bucket('stable-horde-source-images')
]:
with ThreadPoolExecutor(max_workers=10) as executor:
futures = []
for obj in bucket.objects.all():
last_modified = obj.last_modified.replace(tzinfo=timezone.utc)
if last_modified < cutoff_time:
futures.append(executor.submit(obj.delete))
if len(futures) >= 1000:
for future in tqdm(futures):
future.result()
futures = []
for future in tqdm(futures):
future.result()
time.sleep(30)
except:
time.sleep(30)