TianyiMoe/backend/update_pics.py

19 lines
875 B
Python

# Description: This script is used to update the list of pictures in the Azure Blob Storage container.
from azure.storage.blob import BlobServiceClient
import argparse
import json
parser = argparse.ArgumentParser()
parser.add_argument('--container', type=str, default="tianyi-random")
parser.add_argument('--remotedir', type=str, default="")
args=parser.parse_args()
try:
with open('secret.txt', 'r') as f:
connect_str = f.read()
blob_service_client = BlobServiceClient.from_connection_string(connect_str)
container_client = blob_service_client.get_container_client(args.container)
existing_files=[blob.name.split("/")[-1] for blob in container_client.list_blobs(name_starts_with=args.remotedir)]
with open('jsons/pics.json', 'w') as f:
json.dump(existing_files, f, indent=4, sort_keys=True)
except Exception as e:
print(e)
exit(1)