def test_logger_log(caplog):
logger = custom_logging.get_logger(__name__)
logger.error("This is an error")
logger.warning("This is a warning")
logger.info("This is an informational message")
logger.debug("This is a debug message")
assert "This is an error" in caplog.text
assert "This is a warning" in caplog.text
assert "This is an informational message" not in caplog.text
assert "This is a debug message" not in caplog.text
# Licensed under the Apache License, Version 2.0 (the "License"). You may not use this file except in compliance #
# with the License. A copy of the License is located at #
# #
# http://www.apache.org/licenses/LICENSE-2.0 #
# #
# or in the 'license' file accompanying this file. This file is distributed on an 'AS IS' BASIS, WITHOUT WARRANTIES #
# OR CONDITIONS OF ANY KIND, express or implied. See the License for the specific language governing permissions #
# and limitations under the License. #
# #####################################################################################################################
from shared_util import custom_logging
from util.quicksight_application import QuicksightApplication
from util.template import TemplatePermissionType
logger = custom_logging.get_logger(__name__)
class QuicksightApi:
def __init__(self, resource_properties):
self.quicksight_application = QuicksightApplication(
resource_properties)
self.global_state = self.quicksight_application.get_global_state()
def create_all_resources(self):
responses = []
responses.append(self.create_data_source())
responses.append(self.create_data_sets())
responses.append(self.create_analysis())
responses.append(self.create_dashboard())
# #
# or in the 'license' file accompanying this file. This file is distributed on an 'AS IS' BASIS, WITHOUT WARRANTIES #
# OR CONDITIONS OF ANY KIND, express or implied. See the License for the specific language governing permissions #
# and limitations under the License. #
######################################################################################################################
import os
import tempfile
import botocore
from datetime import datetime, timezone
from shared_util.custom_logging import get_logger
from shared_util.service_helper import get_service_resource, get_service_client
from util.constants import TIMESTAMP_FORMAT
logger = get_logger(__name__)
local_folder_path = tempfile.gettempdir()
max_file_size = 10000000 # this number is in bytes that equals 10 MB
class FileSizeTooBigException(Exception):
pass
def download_file(bucket_name: str, key_prefix: str):
s3 = get_service_resource("s3")
""" Download the file locally to the /tmp folder provided by lambda runtime """
logger.debug(f"prefix is {key_prefix}")
local_file_name = os.path.basename(key_prefix) # get file name
def test_get_logger():
logger = custom_logging.get_logger(__name__)
assert logger.level == logging.WARNING
