File size: 2,881 Bytes
944e031 |
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 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 |
import os
import time
import zipfile
import shutil
import threading
from supabase import create_client, Client
# Configuration
SUPABASE_URL = os.environ.get("SUPABASE_URL")
SUPABASE_KEY = os.environ.get("SUPABASE_KEY")
BUCKET_NAME = "moltbot-memory"
MEMORY_DIR = "/root/.molt"
ZIP_FILENAME = "memory.zip"
ZIP_PATH = os.path.join("/app", ZIP_FILENAME) # Working in /app
SYNC_INTERVAL = 300 # 5 minutes
def get_supabase_client():
if not SUPABASE_URL or not SUPABASE_KEY:
print("Error: SUPABASE_URL or SUPABASE_KEY not set.")
return None
try:
return create_client(SUPABASE_URL, SUPABASE_KEY)
except Exception as e:
print(f"Error connecting to Supabase: {e}")
return None
def download_and_extract_memory():
client = get_supabase_client()
if not client: return
print(f"Attempting to download {ZIP_FILENAME} from bucket {BUCKET_NAME}...")
try:
# Check if bucket exists? We assume yes.
# Download file
with open(ZIP_PATH, "wb") as f:
res = client.storage.from_(BUCKET_NAME).download(ZIP_FILENAME)
f.write(res)
print("Download successful. Extracting...")
# Create destination dir if not exists
os.makedirs(MEMORY_DIR, exist_ok=True)
with zipfile.ZipFile(ZIP_PATH, 'r') as zip_ref:
zip_ref.extractall(MEMORY_DIR)
print(f"Extracted memory to {MEMORY_DIR}")
except Exception as e:
print(f"Download/Extraction failed (Normal on first run if no memory exists): {e}")
def zip_and_upload_memory():
client = get_supabase_client()
if not client: return
if not os.path.exists(MEMORY_DIR):
print(f"Memory directory {MEMORY_DIR} does not exist. Skipping upload.")
return
print(f"Zipping {MEMORY_DIR}...")
try:
# Create a zip file
# shutil.make_archive creates file with .zip extension, so we pass base name
base_name = os.path.join("/app", "memory_upload")
shutil.make_archive(base_name, 'zip', MEMORY_DIR)
upload_zip_path = base_name + ".zip"
print(f"Uploading {upload_zip_path} to {BUCKET_NAME}/{ZIP_FILENAME}...")
with open(upload_zip_path, "rb") as f:
# Upsert=true to overwrite
client.storage.from_(BUCKET_NAME).upload(ZIP_FILENAME, f, file_options={"upsert": "true"})
print("Upload successful.")
except Exception as e:
print(f"Upload failed: {e}")
def sync_loop():
while True:
time.sleep(SYNC_INTERVAL)
zip_and_upload_memory()
if __name__ == "__main__":
# 1. On Startup: Download
download_and_extract_memory()
# 2. Start Loop
# We run the loop in the main thread?
# The CMD will run this script in background (&).
# So this script should block (run loop forever).
sync_loop()
|