File Extension Checker

Write a Python function that takes a filename and raises a custom exception if the file’s extension is not supported. The function should only allow .txt or .csv files.

Example 1:

Input: "data.json" 
Output: "Error: File extension not supported."

Example 2:

Input: "data.csv" 
Output: "Valid file extension."

Use Python’s string methods to extract the file extension.

def check_file_extension(filename):
    if not (filename.endswith('.txt') or filename.endswith('.csv')):
        raise Exception("Error: File extension not supported.")
        return "Valid file extension."

print(check_file_extension("data.json"))  # Output: "Error: File extension not supported."
print(check_file_extension("data.csv"))  # Output: "Valid file extension."


© Let’s Data Science


Unlock AI & Data Science treasures. Log in!