use_cache.py

use_cache.py

def get_bucket(): 
  return 'your bucket key'

def get_key():
  return 'get cache key'

def use_cache(f):
  @wraps(f)
  def decorated_function(*args, **kwargs):
    client = boto3.client('s3')
    bucket = get_bucket()
    key = get_key()

    try:
      result = client.get_object(
        Bucket=bucket,
        Key=key,
      )

      return json.loads(result['Body'].read())
    except:
      print('Could not fetch ' + key + ' from S3: ' + bucket)

    results = f(*args, **kwargs)

    if 'status' in results.keys() and results['status'] != 200:
      return results

    try:
      results_bytes = json.dumps(results).encode('utf-8')
      client.put_object(
        ACL='private',
        Body=results_bytes,
        Bucket=bucket,
        Key=key,
      )
    except:
      print('Could not put ' + key + ' to S3: ' + bucket)

    return results

  return decorated_function