Spaces:

MaxSainz2000
/

moltbot-cloud-agent

Paused

MaxSainz2000

Switch to Supabase Storage

944e031 verified 9 months ago

2.88 kB

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()

	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()