feat: update license flag handling and default permissions

- Update DEFAULT_LICENSE_FLAGS from 57 to 127 to enable all commercial modes by default
- Replace CommercialUseLevel enum with bitwise commercial permission handling
- Simplify commercial value normalization and validation using allowed values set
- Adjust bit shifting in license flag construction to accommodate new commercial bits structure
- Remove CommercialUseLevel from exports and update tests accordingly
- Improve handling of empty commercial use values with proper type checking

The changes streamline commercial permission processing and align with CivitAI's default license configuration while maintaining backward compatibility.

py/utils/civitai_utils.py

@@ -2,21 +2,10 @@
 
 from __future__ import annotations
 
-from enum import IntEnum
 from typing import Any, Dict, Iterable, Mapping, Sequence
 from urllib.parse import urlparse, urlunparse
 
 
-class CommercialUseLevel(IntEnum):
-    """Enumerate supported commercial use permission levels."""
-
-    NONE = 0
-    IMAGE = 1
-    RENT_CIVIT = 2
-    RENT = 3
-    SELL = 4
-
-
 _DEFAULT_ALLOW_COMMERCIAL_USE: Sequence[str] = ("Sell",)
 _LICENSE_DEFAULTS: Dict[str, Any] = {
     "allowNoCredit": True,
@@ -24,13 +13,8 @@ _LICENSE_DEFAULTS: Dict[str, Any] = {
     "allowDerivatives": True,
     "allowDifferentLicense": True,
 }
-_COMMERCIAL_VALUE_TO_LEVEL = {
-    "none": CommercialUseLevel.NONE,
-    "image": CommercialUseLevel.IMAGE,
-    "rentcivit": CommercialUseLevel.RENT_CIVIT,
-    "rent": CommercialUseLevel.RENT,
-    "sell": CommercialUseLevel.SELL,
-}
+_COMMERCIAL_ALLOWED_VALUES = {"sell", "rent", "rentcivit", "image"}
+_COMMERCIAL_SHIFT = 1
 
 
 def _normalize_commercial_values(value: Any) -> Sequence[str]:
@@ -53,6 +37,11 @@ def _normalize_commercial_values(value: Any) -> Sequence[str]:
             result.append(str(item))
         if result:
             return result
+        try:
+            if len(value) == 0:  # type: ignore[arg-type]
+                return []
+        except TypeError:
+            pass
 
     return list(_DEFAULT_ALLOW_COMMERCIAL_USE)
 
@@ -96,17 +85,25 @@ def resolve_license_payload(model_data: Mapping[str, Any] | None) -> Dict[str, A
     return payload
 
 
-def _resolve_commercial_level(values: Sequence[str]) -> CommercialUseLevel:
-    level = CommercialUseLevel.NONE
+def _resolve_commercial_bits(values: Sequence[str]) -> int:
+    normalized_values = set()
     for value in values:
-        normalized = str(value).strip().lower().replace("_", "")
-        normalized = normalized.replace("-", "")
-        candidate = _COMMERCIAL_VALUE_TO_LEVEL.get(normalized)
-        if candidate is None:
-            continue
-        if candidate > level:
-            level = candidate
-    return level
+        normalized = str(value).strip().lower().replace("_", "").replace("-", "")
+        if normalized in _COMMERCIAL_ALLOWED_VALUES:
+            normalized_values.add(normalized)
+
+    has_sell = "sell" in normalized_values
+    has_rent = has_sell or "rent" in normalized_values
+    has_rentcivit = has_rent or "rentcivit" in normalized_values
+    has_image = has_sell or "image" in normalized_values
+
+    commercial_bits = (
+        (1 if has_sell else 0) << 3
+        | (1 if has_rent else 0) << 2
+        | (1 if has_rentcivit else 0) << 1
+        | (1 if has_image else 0)
+    )
+    return commercial_bits << _COMMERCIAL_SHIFT
 
 
 def build_license_flags(payload: Mapping[str, Any] | None) -> int:
@@ -118,14 +115,14 @@ def build_license_flags(payload: Mapping[str, Any] | None) -> int:
     if resolved.get("allowNoCredit", True):
         flags |= 1 << 0
 
-    commercial_level = _resolve_commercial_level(resolved.get("allowCommercialUse", ()))
-    flags |= (int(commercial_level) & 0b111) << 1
+    commercial_bits = _resolve_commercial_bits(resolved.get("allowCommercialUse", ()))
+    flags |= commercial_bits
 
     if resolved.get("allowDerivatives", True):
-        flags |= 1 << 4
+        flags |= 1 << 5
 
     if resolved.get("allowDifferentLicense", True):
-        flags |= 1 << 5
+        flags |= 1 << 6
 
     return flags
 
@@ -176,7 +173,6 @@ def rewrite_preview_url(source_url: str | None, media_type: str | None = None) -
 
 
 __all__ = [
-    "CommercialUseLevel",
     "build_license_flags",
     "resolve_license_payload",
     "resolve_license_info",