Spaces:
Build error
Build error
File size: 1,225 Bytes
6276843 |
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 |
import os
from huggingface_hub import SpaceStorage
# Initialize Spaces storage
storage = SpaceStorage()
# Custom storage adapter for Mastodon (simplified)
class MastodonStorage:
def __init__(self):
self.storage = storage
def set(self, key, value):
self.storage.set(key, value)
def get(self, key):
return self.storage.get(key, default=None)
def delete(self, key):
self.storage.delete(key)
# Example functions to hook into Mastodon
def save_user(username, data):
storage.set(f"user:{username}", data)
def get_user(username):
return storage.get(f"user:{username}")
def save_post(post_id, content):
storage.set(f"post:{post_id}", content)
def get_post(post_id):
return storage.get(f"post:{post_id}")
# Called from Ruby via system call (simplified integration)
if __name__ == "__main__":
import sys
action = sys.argv[1] if len(sys.argv) > 1 else None
if action == "save_user":
save_user(sys.argv[2], sys.argv[3])
elif action == "get_user":
print(get_user(sys.argv[2]) or "")
elif action == "save_post":
save_post(sys.argv[2], sys.argv[3])
elif action == "get_post":
print(get_post(sys.argv[2]) or "") |