add future days to be queried

This commit is contained in:
2023-07-25 18:30:09 -05:00
parent 05e658543a
commit 95bb2a6055

View File

@@ -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)