Django template tag for encoding images in base64 and rendering with server with Python 3.x

static_encoder.py

from django import template
from django.contrib.staticfiles.finders import find as find_static_file
import base64

register = template.Library()


def get_file_data(file_path: str) -> bytes:
    """
    @param file_path: Path of the file to get the data
    """
    with open(file_path, 'rb') as f:
        data = f.read()
        f.close()
        return data


@register.simple_tag
def encode_static_base_64(path: str) -> str:
    """
    A template tag that returns an encoded string representation of a static file
    Usage::
        {% encode_static_base_64 path [encoding] %}
    Examples::
        <img src="{% encode_static_base_64 'path/to/img.png' %}">
    """
    try:
        file_path = find_static_file(path)
        if file_path is None:
            return ''
        
        ext = file_path.split('.')[-1]
        file_str = base64.b64encode(get_file_data(file_path)).decode('utf-8')
        return f"data:image/{ext};base64,{file_str}"
    except IOError:
        return ''

If the file doesn't exist, this will crash and prevent the template from rendering. I fixed that by catching it here:

        try:
            ext = file_path.split(".")[-1]
        except AttributeError:
            return ""

You're right @AllenEllis! I've updated the script to check if it's None. Thank you!