# lint-amnesty, pylint: disable=missing-module-docstring

import re

from django.conf import settings

DEFAULT_PYTHON_LIB_FILENAME = 'python_lib.zip'


def can_execute_unsafe_code(course_id):
    """
    Determine if this course is allowed to run unsafe code.

    For use from the ModuleStore.  Checks the `course_id` against a list of whitelisted
    regexes.

    Returns a boolean, true if the course can run outside the sandbox.

    """
    # To decide if we can run unsafe code, we check the course id against
    # a list of regexes configured on the server.
    # If this is not defined in the environment variables then default to the most restrictive, which
    # is 'no unsafe courses'
    # TODO: This should be a database configuration, where we can mark individual courses as being
    # safe/unsafe. Someone in the future should switch us over to that rather than using regexes
    # in a settings file
    # To others using this: the code as-is is brittle and likely to be changed in the future,
    # as per the TODO, so please consider carefully before adding more values to COURSES_WITH_UNSAFE_CODE
    for regex in getattr(settings, 'COURSES_WITH_UNSAFE_CODE', []):
        if re.match(regex, str(course_id)):
            return True
    return False


def get_python_lib_zip(contentstore, course_id):
    """Return the bytes of the course code library file, if it exists."""
    python_lib_filename = getattr(settings, 'PYTHON_LIB_FILENAME', DEFAULT_PYTHON_LIB_FILENAME)
    asset_key = course_id.make_asset_key("asset", python_lib_filename)
    zip_lib = contentstore().find(asset_key, throw_on_not_found=False)
    if zip_lib is not None:
        return zip_lib.data
    else:
        return None


class SandboxService:
    """
    A service which provides utilities for executing sandboxed Python code, for example, inside custom Python questions.

    Args:
        contentstore(function): function which creates an instance of xmodule.content.ContentStore
        course_id(string or CourseLocator): identifier for the course
    """
    def __init__(self, contentstore, course_id, **kwargs):
        super().__init__(**kwargs)
        self.contentstore = contentstore
        self.course_id = course_id

    def can_execute_unsafe_code(self):
        """
        Returns a boolean, true if the course can run outside the sandbox.
        """
        return can_execute_unsafe_code(self.course_id)

    def get_python_lib_zip(self):
        """
        Return the bytes of the course code library file, if it exists.
        """
        return get_python_lib_zip(self.contentstore, self.course_id)