rethinks commited on
Commit
5986bee
·
verified ·
1 Parent(s): c0a1bbe

Upload supabase_storage.py

Browse files
Files changed (1) hide show
  1. supabase_storage.py +12 -5
supabase_storage.py CHANGED
@@ -240,21 +240,28 @@ def list_datasets_from_supabase() -> List[Dict[str, Any]]:
240
  try:
241
  storage = client.storage.from_(BUCKET_NAME)
242
 
243
- # List all items in the bucket root
244
- items = storage.list()
245
  print(f"[Supabase] Bucket contains {len(items)} items")
246
 
 
 
 
 
247
  # Filter to get only folders (datasets), excluding files like _registry.json
248
- # Folders have metadata.id = None or have no 'metadata' key with 'mimetype'
249
  dataset_names = []
250
  for item in items:
251
  name = item.get('name', '')
252
  # Skip hidden files and registry
253
  if name.startswith('_') or name.startswith('.'):
254
  continue
255
- # Check if it's a folder (no metadata.mimetype means it's a folder)
 
256
  item_metadata = item.get('metadata')
257
- if item_metadata is None or item_metadata.get('mimetype') is None:
 
 
258
  dataset_names.append(name)
259
  print(f"[Supabase] Found dataset folder: {name}")
260
 
 
240
  try:
241
  storage = client.storage.from_(BUCKET_NAME)
242
 
243
+ # List all items in the bucket root (pass empty string for root path)
244
+ items = storage.list(path="")
245
  print(f"[Supabase] Bucket contains {len(items)} items")
246
 
247
+ # Debug: print what we got
248
+ for item in items:
249
+ print(f"[Supabase] Item: {item.get('name')} - metadata: {item.get('metadata')}")
250
+
251
  # Filter to get only folders (datasets), excluding files like _registry.json
252
+ # In Supabase Storage, folders typically have id=None or no mimetype
253
  dataset_names = []
254
  for item in items:
255
  name = item.get('name', '')
256
  # Skip hidden files and registry
257
  if name.startswith('_') or name.startswith('.'):
258
  continue
259
+ # Skip if it's a file (has mimetype in metadata)
260
+ # Folders in Supabase typically don't have mimetype
261
  item_metadata = item.get('metadata')
262
+ is_file = item_metadata and item_metadata.get('mimetype')
263
+
264
+ if not is_file:
265
  dataset_names.append(name)
266
  print(f"[Supabase] Found dataset folder: {name}")
267