diff --git a/app/routers/animeguess.py b/app/routers/animeguess.py index 29744b6..c23ce29 100644 --- a/app/routers/animeguess.py +++ b/app/routers/animeguess.py @@ -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"):