oskarastrom commited on
Commit
2027cf1
1 Parent(s): 7a4b92f

Update aws_handler.py

Browse files
Files changed (1) hide show
  1. aws_handler.py +9 -1
aws_handler.py CHANGED
@@ -17,8 +17,16 @@ def upload_file(file_name, bucket, object_name=None):
17
  if object_name is None:
18
  object_name = os.path.basename(file_name)
19
 
 
 
 
 
20
  # Upload the file
21
- s3_client = boto3.client('s3')
 
 
 
 
22
  try:
23
  response = s3_client.upload_file(file_name, bucket, object_name)
24
  except ClientError as e:
 
17
  if object_name is None:
18
  object_name = os.path.basename(file_name)
19
 
20
+ if (not 'AAK_ID' in os.environ) or (not 'ASAK' in os.environ):
21
+ print('AWS keys not specified. Cancelling sync')
22
+ return False
23
+
24
  # Upload the file
25
+ s3_client = boto3.client(
26
+ 's3',
27
+ aws_access_key_id=os.environ['AAK_ID'],
28
+ aws_secret_access_key=os.environ['ASAK']
29
+ )
30
  try:
31
  response = s3_client.upload_file(file_name, bucket, object_name)
32
  except ClientError as e: