models.py

# Standard library
import os
import posixpath

# Third-party
import polib
from django.conf import settings
from django.db import models
from django.db.models import Q
from django.utils import translation

# First-party/Local
from i18n import LANGMAP_DJANGO_TO_PCRE
from i18n.utils import (
    get_default_language_for_jurisdiction,
    get_jurisdiction_name,
    get_pofile_path,
    get_translation_object,
)
from legal_tools.constants import EXCLUDED_LANGUAGE_IDENTIFIERS

# For context of "freedom levels" see:
# https://creativecommons.org/share-your-work/public-domain/freeworks/
FREEDOM_LEVEL_MAX = 1
FREEDOM_LEVEL_MID = 2
FREEDOM_LEVEL_MIN = 3
MAX_LANGUAGE_CODE_LENGTH = 15
UNITS_LICENSES = [
    # Units are in all versions, unless otherwise noted:
    "by",
    "by-nc",
    "by-nc-nd",  # ........ in versions: > 1.0
    "by-nc-sa",
    "by-nd",
    "by-nd-nc",  # ........ in versions: 1.0 unported, 1.0 ported
    "by-sa",
    "devnations",  # ...... in versions: 2.0
    "nc",  # .............. in versions: 2.0-jp, 1.0 unported, 1.0 ported
    "nc-sa",  # ........... in versions: 2.0-jp, 1.0 unported, 1.0 ported
    "nc-sampling+",  # .... in versions: 1.0 unported, 1.0 ported
    "nd",  # .............. in versions: 2.0-jp, 1.0 unported, 1.0 ported
    "nd-nc",  # ........... in versions: 2.0-jp, 1.0 unported, 1.0 ported
    "sa",  # .............. in versions: 2.0-jp, 1.0 unported, 1.0 ported
    "sampling",  # ........ in versions: 1.0 unported, 1.0 ported
    "sampling+",  # ....... in versions: 1.0 unported, 1.0 ported
]
TOOLS_VERSIONS = ["1.0", "2.0", "2.1", "2.5", "3.0", "4.0"]
UNITS_PUBLIC_DOMAIN = [
    "certification",
    "mark",
    "zero",
]
UNITS_DEPRECATED = {
    # Sorted by date, ascending:
    "nc": "2004-05-25",
    "nc-sa": "2004-05-25",
    "nc-sampling": "2004-05-25",
    "nd": "2004-05-25",
    "nd-nc": "2004-05-25",
    "sa": "2004-05-25",
    "devnations": "2007-06-04",
    "sampling": "2007-06-04",
    "certification": "2010-10-11",
    "nc-sampling+": "2011-09-12",
    "sampling+": "2011-09-12",
}
UNITS_DEED_ONLY = [
    "certification",
    "mark",
]


class LegalCodeQuerySet(models.QuerySet):
    # We'll create LegalCode and Tool objects for all the by licenses,
    # and the zero_1.0 ones.
    # We're just doing these units and versions for now:
    # by* 4.0
    # by* 3.0 - including ported
    # cc 1.0

    # Queries for LegalCode objects
    LICENSES_ALL_QUERY = Q(
        tool__unit__in=UNITS_LICENSES,
    )
    LICENSES_40_QUERY = Q(
        tool__unit__in=UNITS_LICENSES,
        tool__version="4.0",
    )
    LICENSES_30_QUERY = Q(
        tool__unit__in=UNITS_LICENSES,
        tool__version="3.0",
    )
    LICENSES_25_QUERY = Q(
        tool__unit__in=UNITS_LICENSES,
        tool__version="2.5",
    )
    LICENSES_21_QUERY = Q(
        tool__unit__in=UNITS_LICENSES,
        tool__version="2.1",
    )
    LICENSES_20_QUERY = Q(
        tool__unit__in=UNITS_LICENSES,
        tool__version="2.0",
    )
    LICENSES_10_QUERY = Q(
        tool__unit__in=UNITS_LICENSES,
        tool__version="1.0",
    )

    # All of the Public Domain tools are at version 1.0
    PUBLIC_DOMAIN_ALL_QUERY = Q(tool__unit__in=UNITS_PUBLIC_DOMAIN)

    PUBLIC_DOMAIN_ZERO_QUERY = Q(tool__unit="zero")

    def translated(self):
        """
        Return a queryset of the LegalCode objects that we are doing the
        translation process on.
        """
        # Only the 4.0 Licenses 4.0 and CC0 1.0 currently have translation
        # support. TODO: add Licenses 3.0 IGO
        return self.filter(
            self.LICENSES_40_QUERY | self.PUBLIC_DOMAIN_ZERO_QUERY
        )

    def valid(self):
        """
        Return a queryset of the LegalCode objects that exist and are valid
        ones that we expect to work. This will change over time as we add
        support for more tools.
        """

        return self.filter(
            self.LICENSES_ALL_QUERY | self.PUBLIC_DOMAIN_ALL_QUERY
        ).exclude(language_code__in=EXCLUDED_LANGUAGE_IDENTIFIERS)

    def validgroups(self):
        """
        Return a queryset of the LegalCode objects that exist and are valid
        ones that we expect to work. This will change over time as we add
        support for more tools.
        """

        return {
            "Licenses 4.0": self.filter(self.LICENSES_40_QUERY).exclude(
                language_code__in=EXCLUDED_LANGUAGE_IDENTIFIERS
            ),
            "Licenses 3.0": self.filter(self.LICENSES_30_QUERY).exclude(
                language_code__in=EXCLUDED_LANGUAGE_IDENTIFIERS
            ),
            "Licenses 2.5": self.filter(self.LICENSES_25_QUERY).exclude(
                language_code__in=EXCLUDED_LANGUAGE_IDENTIFIERS
            ),
            "Licenses 2.1": self.filter(self.LICENSES_21_QUERY).exclude(
                language_code__in=EXCLUDED_LANGUAGE_IDENTIFIERS
            ),
            "Licenses 2.0": self.filter(self.LICENSES_20_QUERY).exclude(
                language_code__in=EXCLUDED_LANGUAGE_IDENTIFIERS
            ),
            "Licenses 1.0": self.filter(self.LICENSES_10_QUERY).exclude(
                language_code__in=EXCLUDED_LANGUAGE_IDENTIFIERS
            ),
            "Public Domain all": self.filter(
                self.PUBLIC_DOMAIN_ALL_QUERY
            ).exclude(language_code__in=EXCLUDED_LANGUAGE_IDENTIFIERS),
        }


class LegalCode(models.Model):
    tool = models.ForeignKey(
        "legal_tools.Tool",
        on_delete=models.CASCADE,
        related_name="legal_codes",
    )
    language_code = models.CharField(
        max_length=MAX_LANGUAGE_CODE_LENGTH,
        help_text="Django langauge code (lowercase IETF language tag)",
    )
    html_file = models.CharField(
        "HTML file",
        max_length=300,
        help_text="HTML file we got this from",
        blank=True,
        default="",
    )
    translation_last_update = models.DateTimeField(
        help_text="The last_updated field from Transifex for this translation",
        blank=True,
        null=True,
        default=None,
    )
    title = models.CharField(
        max_length=112,
        help_text="Tool title in this language, ex."
        " 'Atribución/Reconocimiento 4.0 Internacional'",
        blank=True,
        default="",
    )
    html = models.TextField("HTML", blank=True, default="")
    legal_code_url = models.URLField("Legal Code URL", blank=True, default="")
    deed_url = models.URLField("Deed URL", unique=True)
    plain_text_url = models.URLField(
        "Plain text URL",
        blank=True,
        default="",
    )

    objects = LegalCodeQuerySet.as_manager()

    class Meta:
        ordering = ["tool", "language_code"]

    def __str__(self):
        return f"LegalCode<{self.language_code}, {self.tool}>"

    def save(self, *args, **kwargs):
        self.deed_url = build_path(
            self.tool.base_url,
            "deed",
            self.language_code,
        )
        self.legal_code_url = build_path(
            self.tool.base_url,
            "legalcode",
            self.language_code,
        )
        # NOTE: plaintext functionality disabled
        # unit = self.tool.unit
        # if (
        #     (unit in UNITS_LICENSES and float(self.tool.version) > 2.5)
        #     or unit == "zero"
        # ) and self.language_code == "en":
        #     self.plain_text_url = build_path(
        #         self.tool.base_url,
        #         "legalcode.txt",
        #         self.language_code,
        #     )
        super().save(*args, **kwargs)

    def get_publish_files(self):
        """
        1. Generate relative path for legal code
        2. Generate list of symlinks to ensure expected legal code paths
           function correctly
        2. Generate list of redirects to ensure deed-only legal code leads to
           deed
        """
        language_code = self.language_code
        tool = self.tool
        juris_code = tool.jurisdiction_code
        language_default = get_default_language_for_jurisdiction(juris_code)
        filename = f"legalcode.{self.language_code}.html"

        # Relative path
        if self.tool.deed_only:
            relpath = None
        else:
            relpath = os.path.join(tool._get_save_path(), filename)

        # Symlinks
        symlinks = []
        if language_code == language_default:
            if not self.tool.deed_only:
                # Symlink default languages
                symlinks.append("legalcode.html")

        # Redirects
        redirects_data = []
        if self.tool.deed_only:
            redirects_data.append(
                {
                    "redirect_file": os.path.join(
                        tool._get_save_path(),
                        f"legalcode.{language_code}.html",
                    ),
                    "title": self.title,
                    "destination": f"deed.{language_code}",
                    "language_code": language_code,
                }
            )
            redirects_data.append(
                {
                    "redirect_file": os.path.join(
                        tool._get_save_path(), "legalcode.html"
                    ),
                    "title": self.title,
                    "destination": f"deed.{language_code}",
                    "language_code": language_code,
                }
            )

        return [relpath, symlinks, redirects_data]

    def get_redirect_pairs(self):
        language_code = self.language_code
        tool = self.tool
        filename = f"legalcode.{language_code}"
        dest_path = os.path.join("/", tool._get_save_path(), filename)
        pairs = []
        for pcre in LANGMAP_DJANGO_TO_PCRE.get(language_code, []):
            pcre_match = os.path.join(
                "/",
                tool._get_save_path().replace(".", "[.]"),
                f"legalcode[.]{pcre}(?:[.]html)?",
            )
            pairs.append([pcre_match, dest_path])
        return pairs

    def has_english(self):
        """
        Return True if there's an English translation for the same tool.
        """
        return (
            self.language_code == "en"
            or self.tool.legal_codes.filter(language_code="en").exists()
        )

    def branch_name(self):
        """
        If this translation is modified, what is the name of the GitHub branch
        we'll use to manage the modifications?  Basically its
        "{unit}-{version}-{language}-{jurisdiction code}", except that all the
        "by* 4.0" licenses use "cc4" for the unit part. This has to be a valid
        DNS domain, so we also change any _ to - and remove any periods.
        """
        tool = self.tool
        parts = []
        if tool.unit.startswith("by") and tool.version == "4.0":
            parts.append("cc4")
        else:
            parts.extend([tool.unit, tool.version])
        parts.append(self.language_code)
        if tool.jurisdiction_code:
            parts.append(tool.jurisdiction_code)
        return "-".join(parts).replace("_", "-").replace(".", "").lower()

    def identifier(self):
        """
        Returns ex. 'CC BY-SA 4.0' - all upper case etc. No language.
        """
        return self.tool.identifier()

    @property
    def translation_domain(self):
        return self.tool.resource_slug

    def get_translation_object(self):
        language_default = get_default_language_for_jurisdiction(
            self.tool.jurisdiction_code
        )
        return get_translation_object(
            domain=self.tool.resource_slug,
            language_code=self.language_code,
            language_default=language_default,
        )

    def get_pofile(self) -> polib.POFile:
        with open(self.translation_filename(), "rb") as f:
            content = f.read()
        return polib.pofile(content.decode(), encoding="utf-8")

    def get_english_pofile_path(self) -> str:
        if self.language_code != settings.LANGUAGE_CODE:
            # Same tool, just in English translation:
            english_legal_code = self.tool.get_legal_code_for_language_code(
                settings.LANGUAGE_CODE
            )
            return english_legal_code.translation_filename()
        return self.translation_filename()

    def translation_filename(self):
        """
        Return absolute path to the .po file with this translation.
        """
        pofile_path = get_pofile_path(
            locale_or_legalcode="legalcode",
            language_code=self.language_code,
            translation_domain=self.tool.resource_slug,
        )
        return pofile_path


class Tool(models.Model):
    base_url = models.URLField(
        "Base URL",
        max_length=200,
        help_text="The tool's unique identifier, ex."
        " 'https://creativecommons.org/licenses/by-nd/2.0/br/'",
        unique=True,
        default="",
    )
    unit = models.CharField(
        max_length=20,
        help_text="shorthand representation for which class of tools this"
        " falls into. Ex. 'by-nc-sa', or 'MIT', 'nc-sampling+',"
        " 'devnations', ...",
    )
    version = models.CharField(
        max_length=3,
        help_text="Ex. '4.0'. Not required.",
        blank=True,
        default="",
    )
    jurisdiction_code = models.CharField(
        max_length=9,
        blank=True,
        default="",
        help_text="legal jurisdiction (usually a ISO 3166-1 alpha-2 country"
        " code)",
    )
    creator_url = models.URLField(
        "Creator URL",
        max_length=200,
        blank=True,
        default="",
        help_text="Ex. https://creativecommons.org",
    )
    category = models.CharField(
        max_length=13,
        help_text="'licenses' or 'publicdomain'",
        blank=True,
        default="",
    )
    source = models.ForeignKey(
        "self",
        blank=True,
        null=True,
        default=None,
        on_delete=models.CASCADE,
        related_name="source_of",
        help_text="A related resource from which the described resource is"
        " derived (dcterms:source)",
    )
    is_replaced_by = models.ForeignKey(
        "self",
        blank=True,
        null=True,
        default=None,
        on_delete=models.CASCADE,
        related_name="+",
        help_text="A related resource that supplants, displaces, or supersedes"
        " the described resource (dcterms:isReplacedBy)",
    )
    deprecated_on = models.DateField(
        blank=True,
        null=True,
        default=None,
        help_text="if set, the date on which this tool was deprecated",
    )

    deed_only = models.BooleanField(default=False)

    permits_derivative_works = models.BooleanField(default=None)
    permits_reproduction = models.BooleanField(default=None)
    permits_distribution = models.BooleanField(default=None)
    permits_sharing = models.BooleanField(default=None)

    requires_share_alike = models.BooleanField(default=None)
    requires_notice = models.BooleanField(default=None)
    requires_attribution = models.BooleanField(default=None)

    prohibits_commercial_use = models.BooleanField(default=None)
    prohibits_high_income_nation_use = models.BooleanField(default=None)

    def __lt__(self, other):
        """Magic method to support sorting"""
        return (
            self.category,
            self.unit,
            self.version,
            self.jurisdiction_code,
        ) < (
            other.category,
            other.unit,
            other.version,
            other.jurisdiction_code,
        )

    def __str__(self):
        return f"Tool<{self.unit},{self.version}," f"{self.jurisdiction_code}>"

    def _get_save_path(self):
        """
        If saving the deed or legal code as a static file, this returns the
        relative path where the saved file should be, not including the actual
        filename.

        ported Licenses 3.0 and earlier
            Formula
                CATEGORY/UNIT/VERSION/JURISDICTION
            Examples
                licenses/by/3.0/am
                licenses/by-nc/3.0/pl
                licenses/by-nc-nd/2.5/au
                licenses/by-nc-sa/2.5/ch
                licenses/by/2.1/es
                licenses/by-nc/2.1/jp
                licenses/by/2.0/kr
                licenses/nd-nc/1.0/fi

        unported Licenses 3.0, Licenses 4.0, and Public Domain:
            Formula
                CATEGORY/UNIT/VERSION
            Examples
                publicdomain/zero/1.0
                licenses/by-nc-nd/4.0/
                licenses/by-nc-sa/4.0/
                licenses/by-nc/4.0/
                licenses/by-nd/4.0/
                licenses/by-sa/4.0/
                licenses/by/4.0/
        """

        unit = self.unit.lower()
        if self.jurisdiction_code:
            # ported Licenses 3.0 and earlier
            return os.path.join(
                self.category,  # licenses or publicdomain
                unit,  # ex. by, by-nc-nd
                self.version,  # ex. 1.0, 2.0
                self.jurisdiction_code,  # ex. ca, tw
            )
        else:
            # unported Licenses 3.0, Licenses 4.0, and Public Domain:
            return os.path.join(
                self.category,  # licenses or publicdomain
                unit,  # ex. by, by-nc-nd, zero
                self.version,  # ex. 1.0, 4.0
            )

    class Meta:
        ordering = ["-version", "unit", "jurisdiction_code"]

    def save(self, *args, **kwargs):
        self.creator_url = self.creator_url.rstrip("/")
        super().save(*args, **kwargs)

    def get_legal_code_for_language_code(self, language_code):
        """
        Return the LegalCode object for this tool and language.
        """
        if not language_code:
            language_code = translation.get_language()
        try:
            return self.legal_codes.get(language_code=language_code)
        except LegalCode.DoesNotExist as e:
            e.args = (f"{e.args[0]} language_code={language_code}",)
            raise

    def get_metadata(self):
        """
        Return a dictionary with the metadata for this tool.
        """
        language_default = get_default_language_for_jurisdiction(
            self.jurisdiction_code
        )
        data = {}
        default_lc = self.legal_codes.filter(language_code=language_default)[0]
        data["base_url"] = self.base_url
        data["deed_only"] = self.deed_only
        if self.deprecated_on:
            data["deprecated_on"] = self.deprecated_on
        if self.jurisdiction_code:
            data["jurisdiction_code"] = self.jurisdiction_code
        data["jurisdiction_name"] = get_jurisdiction_name(
            self.category,
            self.unit,
            self.version,
            self.jurisdiction_code,
        )
        data["identifier"] = self.identifier()
        if not self.deed_only:
            data["legal_code_languages"] = {}
            for lc in self.legal_codes.order_by("language_code"):
                lang_code = lc.language_code
                language_info = translation.get_language_info(lang_code)
                data["legal_code_languages"][lang_code] = language_info["name"]
        data["permits_derivative_works"] = self.permits_derivative_works
        data["permits_distribution"] = self.permits_distribution
        data["permits_reproduction"] = self.permits_reproduction
        data["permits_sharing"] = self.permits_sharing
        data["prohibits_commercial_use"] = self.prohibits_commercial_use
        data["prohibits_high_income_nation_use"] = (
            self.prohibits_high_income_nation_use
        )
        data["requires_attribution"] = self.requires_attribution
        data["requires_notice"] = self.requires_notice
        data["requires_share_alike"] = self.requires_share_alike
        data["title"] = default_lc.title
        data["unit"] = self.unit
        data["version"] = self.version
        return data

    def get_publish_files(self, language_code):
        """
        1. Generate relative path for deed
        2. Generate list of symlinks to ensure expected deed paths function
           correctly
        """
        juris_code = self.jurisdiction_code
        language_default = get_default_language_for_jurisdiction(juris_code)
        filename = f"deed.{language_code}.html"

        # Relative path
        relpath = os.path.join(self._get_save_path(), filename)

        # Symlinks
        symlinks = []
        if language_code == language_default:
            # Symlink default languages
            symlinks.append("deed.html")
            symlinks.append("index.html")

        return [relpath, symlinks]

    def get_redirect_pairs(self, language_code):
        filename = f"deed.{language_code}"
        dest_path = os.path.join("/", self._get_save_path(), filename)
        pairs = []
        for pcre in LANGMAP_DJANGO_TO_PCRE.get(language_code, []):
            pcre_match = os.path.join(
                "/",
                self._get_save_path().replace(".", "[.]"),
                f"deed[.]{pcre}(?:[.]html)?",
            )
            pairs.append([pcre_match, dest_path])
        return pairs

    def logos(self):
        """
        Return an iterable of the codes for the logos that should be
        displayed with this tool. Ex.:
        ["cc-logo", "cc-zero", "cc-by"]
        """
        result = ["cc-logo"]  # Everybody gets this
        if self.unit == "mark":
            result.append("cc-pdm")
        elif self.unit == "sampling":
            result.append("cc-sampling")
        elif self.unit == "sampling+":
            result.append("cc-sampling-plus")
        elif self.unit == "nc-sampling+":
            result.append("cc-nc")
            result.append("cc-sampling-plus")
        elif self.unit == "zero":
            result.append("cc-zero")
        else:
            for unit_part in self.unit.split("-"):
                if unit_part in ["by", "nc", "nd", "sa"]:
                    result.append(f"cc-{unit_part}")
        return result

    @property
    def resource_name(self):
        """Human-readable name for the translation resource for this tool"""
        return self.identifier()

    @property
    def resource_slug(self):
        # Transifex translation resource slug for this tool.
        # letters, numbers, underscores or hyphens.
        # No periods.
        # All lowercase.
        if self.jurisdiction_code:
            slug = f"{self.unit}_{self.version}_{self.jurisdiction_code}"
        else:
            slug = f"{self.unit}_{self.version}"
        slug = slug.replace(".", "")
        slug = slug.lower()
        return slug

    def identifier(self):
        """
        Returns ex. 'CC BY-SA 4.0' - all upper case etc. No language.
        """
        tool = self
        identifier = f"{tool.unit} {tool.version}"

        if tool.unit == "mark":
            identifier = f"PDM {tool.version}"
        elif tool.unit == "zero":
            identifier = f"CC0 {tool.version}"
        elif tool.unit in UNITS_LICENSES:
            identifier = f"CC {identifier}"

        if tool.jurisdiction_code:
            identifier = f"{identifier} {tool.jurisdiction_code}"
        identifier = identifier.upper()
        return identifier

    @property
    def level_of_freedom(self):
        if self.unit in ("devnations", "sampling"):
            return FREEDOM_LEVEL_MIN
        elif (
            self.unit.find("sampling") > -1
            or self.unit.find("nc") > -1
            or self.unit.find("nd") > -1
        ):
            return FREEDOM_LEVEL_MID
        else:
            return FREEDOM_LEVEL_MAX

    @property
    def superseded(self):
        return self.is_replaced_by is not None

    @property
    def sampling_plus(self):
        return self.unit in ("nc-sampling+", "sampling+")

    @property
    def include_share_adapted_material_clause(self):
        return self.unit in ["by", "by-nc"]

    @property
    def nc(self):
        return "nc" in self.unit

    @property
    def nd(self):
        return "nd" in self.unit

    @property
    def sa(self):
        return "sa" in self.unit


class TranslationBranch(models.Model):
    branch_name = models.CharField(max_length=40)
    legal_codes = models.ManyToManyField("LegalCode")
    version = models.CharField(
        max_length=3,
        help_text="Ex. '4.0'. Not required.",
        blank=True,
        default="",
    )
    language_code = models.CharField(
        max_length=MAX_LANGUAGE_CODE_LENGTH,
        help_text="Django langauge code (lowercase IETF language tag)",
    )
    last_transifex_update = models.DateTimeField(
        "Time when last updated on Transifex.",
        blank=True,
        null=True,
        default=None,
    )
    complete = models.BooleanField(default=False)

    class Meta:
        verbose_name_plural = "translation branches"

    def __str__(self):
        return (
            f"Translation branch {self.branch_name}."
            f" {'Complete' if self.complete else 'In progress'}."
        )


def build_path(base_url, document, language_code=None):
    path = base_url.replace(settings.CANONICAL_SITE, "")
    if document == "legalcode.txt" or not language_code:
        path = posixpath.join(path, document)
    else:
        path = posixpath.join(path, f"{document}.{language_code}")
    return path