From 95bb2a605502c41bc19a8ac3c0121be546739b24 Mon Sep 17 00:00:00 2001 From: Daniel Tam Date: Tue, 25 Jul 2023 18:30:09 -0500 Subject: [PATCH] add future days to be queried --- app/routers/animeguess.py | 16 +++++++++++++--- 1 file changed, 13 insertions(+), 3 deletions(-) diff --git a/app/routers/animeguess.py b/app/routers/animeguess.py index 71b2e41..bac5c5e 100644 --- a/app/routers/animeguess.py +++ b/app/routers/animeguess.py @@ -149,18 +149,28 @@ async def ag_search(search: Search): @router.get("/ag-list") async def ag_list(): - source_name = os.getenv('SOURCE_BUCKET_NAME') + current_name = os.getenv('CURRENT_BUCKET_NAME') + future_name = os.getenv('FUTURE_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') - source_bucket = s3.Bucket(source_name) + current_bucket = s3.Bucket(current_name) # list of anime titles anime_titles = [] # get titles through S3 - for obj in source_bucket.objects.filter(Prefix="days/"): + 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) + anime_titles.append({'title': f'{metadata["answer"]} [{metadata["eng_title"]}]'}) + + current_bucket = s3.Bucket(future_name) + + # get titles through S3 + for obj in current_bucket.objects.all(): if obj.key.endswith("metadata.json"): metadata_file = obj.get()['Body'].read() metadata = json.loads(metadata_file)