use new s3 bucket
This commit is contained in:
@@ -149,34 +149,16 @@ async def ag_search(search: Search):
|
||||
|
||||
@router.get("/ag-list")
|
||||
async def ag_list():
|
||||
current_name = os.getenv('CURRENT_BUCKET_NAME')
|
||||
future_name = os.getenv('FUTURE_BUCKET_NAME')
|
||||
days_name = os.getenv('DAYS_BUCKET_NAME')
|
||||
|
||||
# Log into AWS
|
||||
session = boto3.Session( aws_access_key_id=os.getenv('ACCESS_KEY'), aws_secret_access_key=os.getenv('SECRET_ACCESS_KEY'))
|
||||
s3 = session.resource('s3')
|
||||
current_bucket = s3.Bucket(current_name)
|
||||
current_bucket = s3.Bucket(days_name)
|
||||
|
||||
# list of anime titles
|
||||
anime_titles = []
|
||||
|
||||
# get titles through S3
|
||||
for obj in current_bucket.objects.filter(Prefix="days/"):
|
||||
if obj.key.endswith("metadata.json"):
|
||||
metadata_file = obj.get()['Body'].read()
|
||||
metadata = json.loads(metadata_file)
|
||||
|
||||
key = obj.key
|
||||
day_num_split = key.split('/')
|
||||
day_num = int(day_num_split[1])
|
||||
|
||||
if metadata['answer'].lower() == metadata['eng_title'].lower():
|
||||
anime_titles.append({'title': f'{metadata["answer"]} ({metadata["difficulty"]})', 'day': day_num})
|
||||
else:
|
||||
anime_titles.append({'title': f'{metadata["answer"]} [{metadata["eng_title"]}] ({metadata["difficulty"]})', 'day': day_num})
|
||||
|
||||
current_bucket = s3.Bucket(future_name)
|
||||
|
||||
# get titles through S3
|
||||
for obj in current_bucket.objects.all():
|
||||
if obj.key.endswith("metadata.json"):
|
||||
|
Reference in New Issue
Block a user