21 lines
1.0 KiB
Python
21 lines
1.0 KiB
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 os
|
|
from .models import Picture
|
|
def update_pictures(container='tianyi-random', remotedir='', secretdir='/home/tianyi/repos/TianyiMoe/backend/secret.txt'):
|
|
# print(os.getcwd())
|
|
try:
|
|
with open(secretdir, 'r') as f:
|
|
connect_str = f.read()
|
|
blob_service_client = BlobServiceClient.from_connection_string(connect_str)
|
|
container_client = blob_service_client.get_container_client(container)
|
|
existing_files=[blob.name.split("/")[-1] for blob in container_client.list_blobs(name_starts_with=remotedir)]
|
|
for file in existing_files:
|
|
if not Picture.objects.filter(name=file.split('.')[0]).exists():
|
|
Picture.objects.create(name=file.split('.')[0])
|
|
print("File number: " + str(len(existing_files)))
|
|
except Exception as e:
|
|
print(e)
|
|
|
|
if __name__ == '__main__':
|
|
update_pictures() |