daniel/dtam-api
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

use new s3 bucket

Browse Source
This commit is contained in:
Daniel Tam
2023-08-09 00:42:00 -05:00
parent a06bd03f5f
commit 75576f96ce
1 changed files with 2 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
app/routers/animeguess.py
View File

@@ -149,34 +149,16 @@ async def ag_search(search: Search):
@router.get("/ag-list") @router.get("/ag-list")
async def ag_list(): async def ag_list():
current_name = os.getenv('CURRENT_BUCKET_NAME') days_name = os.getenv('DAYS_BUCKET_NAME')
future_name = os.getenv('FUTURE_BUCKET_NAME')
# Log into AWS # Log into AWS
session = boto3.Session( aws_access_key_id=os.getenv('ACCESS_KEY'), aws_secret_access_key=os.getenv('SECRET_ACCESS_KEY')) session = boto3.Session( aws_access_key_id=os.getenv('ACCESS_KEY'), aws_secret_access_key=os.getenv('SECRET_ACCESS_KEY'))
s3 = session.resource('s3') s3 = session.resource('s3')
current_bucket = s3.Bucket(current_name) current_bucket = s3.Bucket(days_name)
# list of anime titles # list of anime titles
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 # get titles through S3
for obj in current_bucket.objects.all(): for obj in current_bucket.objects.all():
if obj.key.endswith("metadata.json"): if obj.key.endswith("metadata.json"):