Spaces:
Running
Running
File size: 2,054 Bytes
775411b | 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 | import os
import shutil
from pathlib import Path
from huggingface_hub import hf_hub_download, HfApi
DB_FILE = Path(__file__).parent.parent / "db" / "music_dashboard.db"
def download_db():
"""Download the SQLite database from the Hugging Face Dataset."""
token = os.environ.get("HF_TOKEN")
repo = os.environ.get("HF_DATASET_REPO")
if not token or not repo:
print("SYNC: HF_TOKEN or HF_DATASET_REPO not set. Skipping DB download.")
return
print(f"SYNC: Downloading database from dataset {repo}...")
try:
downloaded_path = hf_hub_download(
repo_id=repo,
repo_type="dataset",
filename="music_dashboard.db",
token=token,
force_download=True
)
DB_FILE.parent.mkdir(parents=True, exist_ok=True)
shutil.copy2(downloaded_path, DB_FILE)
print("SYNC: Database downloaded and replaced successfully.")
except Exception as e:
print(f"SYNC: Failed to download DB from dataset (this is normal if the dataset is newly created and empty): {e}")
def upload_db():
"""Upload the local SQLite database to the Hugging Face Dataset."""
token = os.environ.get("HF_TOKEN")
repo = os.environ.get("HF_DATASET_REPO")
if not token or not repo:
print("SYNC: HF_TOKEN or HF_DATASET_REPO not set. Skipping DB upload.")
return
if not DB_FILE.exists():
print(f"SYNC: Database file {DB_FILE} does not exist. Skipping upload.")
return
print(f"SYNC: Uploading database to dataset {repo}...")
try:
api = HfApi()
api.upload_file(
path_or_fileobj=str(DB_FILE),
path_in_repo="music_dashboard.db",
repo_id=repo,
repo_type="dataset",
token=token,
commit_message="Auto-sync daily dashboard database update"
)
print("SYNC: Database uploaded successfully.")
except Exception as e:
print(f"SYNC: Failed to upload DB to dataset: {e}")
|