102 lines
2.9 KiB
Python
102 lines
2.9 KiB
Python
from __future__ import annotations
|
|
|
|
from typing import Any
|
|
|
|
try:
|
|
from .krea_action_context import (
|
|
axis_values_text,
|
|
is_climax_text,
|
|
is_foreplay_text,
|
|
is_oral_text,
|
|
is_outercourse_text,
|
|
is_toy_assisted_double_text,
|
|
is_vaginal_penetration_text,
|
|
)
|
|
except ImportError: # Allows local smoke tests with `python -c`.
|
|
from krea_action_context import (
|
|
axis_values_text,
|
|
is_climax_text,
|
|
is_foreplay_text,
|
|
is_oral_text,
|
|
is_outercourse_text,
|
|
is_toy_assisted_double_text,
|
|
is_vaginal_penetration_text,
|
|
)
|
|
|
|
|
|
ACTION_CLIMAX = "climax"
|
|
ACTION_FOREPLAY = "foreplay"
|
|
ACTION_OUTERCOURSE = "outercourse"
|
|
ACTION_ORAL = "oral"
|
|
ACTION_PENETRATION = "penetration"
|
|
ACTION_TOY_DOUBLE = "toy_double"
|
|
ACTION_DEFAULT = "default"
|
|
|
|
HARDCORE_ACTION_FAMILY_CHOICES = {
|
|
ACTION_CLIMAX,
|
|
ACTION_FOREPLAY,
|
|
ACTION_OUTERCOURSE,
|
|
ACTION_ORAL,
|
|
ACTION_PENETRATION,
|
|
ACTION_TOY_DOUBLE,
|
|
ACTION_DEFAULT,
|
|
}
|
|
|
|
|
|
def normalize_hardcore_action_family(value: Any, default: str = "") -> str:
|
|
text = str(value or "").strip().lower()
|
|
if text == "penetrative":
|
|
text = ACTION_PENETRATION
|
|
return text if text in HARDCORE_ACTION_FAMILY_CHOICES else default
|
|
|
|
|
|
def infer_hardcore_action_family(
|
|
role_graph: str,
|
|
hard_item: str,
|
|
composition: str = "",
|
|
axis_values: Any = None,
|
|
*,
|
|
is_climax: bool | None = None,
|
|
) -> str:
|
|
axis_text = axis_values_text(axis_values)
|
|
if is_climax is None:
|
|
is_climax = is_climax_text(role_graph, hard_item, composition, axis_text)
|
|
if is_climax:
|
|
return ACTION_CLIMAX
|
|
if is_foreplay_text(role_graph, hard_item, composition, axis_text):
|
|
return ACTION_FOREPLAY
|
|
if is_outercourse_text(role_graph, hard_item, composition, axis_text):
|
|
return ACTION_OUTERCOURSE
|
|
if is_oral_text(role_graph, hard_item, composition, axis_text):
|
|
return ACTION_ORAL
|
|
if is_vaginal_penetration_text(role_graph, hard_item, composition, axis_text):
|
|
return ACTION_PENETRATION
|
|
if is_toy_assisted_double_text(role_graph, hard_item, composition, axis_text):
|
|
return ACTION_TOY_DOUBLE
|
|
return ACTION_DEFAULT
|
|
|
|
|
|
def source_hardcore_action_family(
|
|
source_family: Any,
|
|
role_graph: str,
|
|
hard_item: str,
|
|
composition: str = "",
|
|
axis_values: Any = None,
|
|
) -> str:
|
|
inferred = infer_hardcore_action_family(role_graph, hard_item, composition, axis_values)
|
|
if inferred in (ACTION_CLIMAX, ACTION_TOY_DOUBLE):
|
|
return inferred
|
|
family = str(source_family or "").strip().lower()
|
|
source_mapping = {
|
|
"penetrative": ACTION_PENETRATION,
|
|
"foreplay": ACTION_FOREPLAY,
|
|
"interaction": ACTION_FOREPLAY,
|
|
"manual": ACTION_FOREPLAY,
|
|
"oral": ACTION_ORAL,
|
|
"outercourse": ACTION_OUTERCOURSE,
|
|
"climax": ACTION_CLIMAX,
|
|
}
|
|
if family == "anal":
|
|
return ACTION_DEFAULT
|
|
return source_mapping.get(family, inferred)
|