Extract hardcore position config policy

prompt_builder.py

@@ -28,6 +28,7 @@ try:
     from . import filter_config as filter_policy
     from . import generate_prompt_batches as g
     from . import generation_profile_config as generation_profile_policy
+    from . import hardcore_position_config as hardcore_position_policy
     from . import location_config as location_policy
     from . import pair_clothing
     from . import pair_camera
@@ -69,6 +70,7 @@ except ImportError:  # Allows local smoke tests with `python -c`.
     import filter_config as filter_policy
     import generate_prompt_batches as g
     import generation_profile_config as generation_profile_policy
+    import hardcore_position_config as hardcore_position_policy
     import location_config as location_policy
     import pair_clothing
     import pair_camera
@@ -292,221 +294,21 @@ CHARACTER_EYE_COLOR_CHOICES = [
 
 CAMERA_DETAIL_CHOICES = camera_policy.CAMERA_DETAIL_CHOICES
 HARDCORE_DETAIL_DENSITY_CHOICES = ["compact", "balanced", "dense"]
-HARDCORE_POSITION_FAMILY_CHOICES = [
-    "any",
-    "penetrative",
-    "foreplay",
-    "interaction",
-    "manual",
-    "oral",
-    "outercourse",
-    "anal",
-    "climax",
-    "threesome",
-    "group",
-]
-HARDCORE_POSITION_FOCUS_CHOICES = [
-    "keep_pool",
-    "penetration_only",
-    "foreplay_only",
-    "interaction_only",
-    "manual_only",
-    "oral_only",
-    "outercourse_only",
-    "anal_only",
-    "climax_only",
-    "threesome_only",
-    "group_only",
-]
-HARDCORE_POSITION_KEY_CHOICES = [
-    "missionary",
-    "cowgirl",
-    "reverse_cowgirl",
-    "doggy",
-    "bent_over",
-    "face_down_ass_up",
-    "standing",
-    "side_lying",
-    "edge_supported",
-    "kneeling",
-    "lotus_lap",
-    "face_sitting",
-    "sixty_nine",
-    "reclining_oral",
-    "straddled_oral",
-    "spread_leg_oral",
-    "chair_oral",
-    "kissing",
-    "caressing",
-    "breast_touch",
-    "face_touch",
-    "undressing",
-    "body_worship",
-    "nipple_play",
-    "ass_grab",
-    "thigh_kissing",
-    "hair_holding",
-    "wrist_pinning",
-    "dirty_talk",
-    "position_transition",
-    "guided_positioning",
-    "camera_showing",
-    "watching",
-    "aftercare",
-    "cleanup",
-    "fingering",
-    "clit_rubbing",
-    "mutual_masturbation",
-    "boobjob",
-    "testicle_sucking",
-    "penis_licking",
-    "handjob",
-    "footjob",
-    "open_thighs",
-    "front_back",
-]
-HARDCORE_POSITION_FAMILY_SUBCATEGORIES = {
-    "any": [
-        "penetrative_sex",
-        "foreplay_teasing",
-        "body_worship_touching",
-        "clothing_position_transitions",
-        "dominant_guidance",
-        "camera_performance",
-        "manual_stimulation",
-        "oral_sex",
-        "outercourse_sex",
-        "anal_double_penetration",
-        "threesomes",
-        "group_coordination",
-        "group_sex_orgy",
-        "cumshot_climax",
-        "aftercare_cleanup",
-    ],
-    "penetrative": ["penetrative_sex"],
-    "foreplay": ["foreplay_teasing"],
-    "interaction": [
-        "foreplay_teasing",
-        "body_worship_touching",
-        "clothing_position_transitions",
-        "dominant_guidance",
-        "camera_performance",
-        "group_coordination",
-        "aftercare_cleanup",
-    ],
-    "manual": ["manual_stimulation"],
-    "oral": ["oral_sex"],
-    "outercourse": ["outercourse_sex", "manual_stimulation"],
-    "anal": ["anal_double_penetration"],
-    "climax": ["cumshot_climax"],
-    "threesome": ["threesomes"],
-    "group": ["group_sex_orgy"],
-}
-HARDCORE_POSITION_KEY_MATCHES = {
-    "missionary": ("missionary", "above her", "under her"),
-    "cowgirl": ("cowgirl", "straddling", "straddles", "on top", "squatting on top"),
-    "reverse_cowgirl": ("reverse cowgirl", "facing away"),
-    "doggy": ("doggy", "all fours", "rear-entry", "from behind"),
-    "bent_over": ("bent-over", "bent over", "hips raised"),
-    "face_down_ass_up": ("face-down", "ass-up"),
-    "standing": ("standing", "stands", "braced standing"),
-    "side_lying": ("side-lying", "side lying", "spooning", "on the side", "on her side"),
-    "edge_supported": ("edge-of-bed", "edge of bed", "bed edge", "raised edge", "edge-supported"),
-    "kneeling": ("kneeling", "kneels", "kneeling center"),
-    "lotus_lap": ("lotus", "lap", "seated in a partner's lap"),
-    "face_sitting": ("face-sitting", "face sitting"),
-    "sixty_nine": ("sixty-nine", "69"),
-    "reclining_oral": ("reclining cunnilingus",),
-    "straddled_oral": ("straddled oral",),
-    "spread_leg_oral": ("spread-leg", "spread leg", "reclining cunnilingus"),
-    "chair_oral": ("chair oral",),
-    "kissing": ("kiss", "kissing", "mouth-to-mouth", "mouth to mouth", "lips pressed"),
-    "caressing": ("caress", "caressing", "hands roaming", "stroking skin", "hands sliding"),
-    "breast_touch": ("breast", "breasts", "nipple", "cupping breasts", "touching breasts"),
-    "face_touch": ("face", "cheek", "jaw", "chin", "hand on the cheek", "fingers under the chin"),
-    "undressing": ("undressing", "removing clothing", "removing clothes", "pulling clothing", "sliding straps", "unbuttoning"),
-    "body_worship": ("body worship", "worship", "kissing down", "mouth on skin", "kissing the body"),
-    "nipple_play": ("nipple", "nipples", "licking nipples", "sucking nipples", "nipple play"),
-    "ass_grab": ("ass grab", "ass-grab", "ass grabbing", "hand on the ass", "squeezing the ass"),
-    "thigh_kissing": ("thigh kiss", "thigh kissing", "kissing thighs", "mouth on inner thighs"),
-    "hair_holding": ("hair holding", "hair held", "holding hair", "hair pulled back"),
-    "wrist_pinning": ("wrist", "wrists", "pinning wrists", "wrists pinned", "hands pinned"),
-    "dirty_talk": ("dirty talk", "whispering", "mouth near the ear", "telling", "verbal teasing"),
-    "position_transition": ("transition", "turning around", "pulling onto the bed", "moving into position", "position change"),
-    "guided_positioning": ("guiding", "guided", "guides", "lifting legs", "spreading thighs", "pulling hips", "turning the body"),
-    "camera_showing": ("camera", "showing to camera", "presenting to camera", "spread open for camera", "creator-shot"),
-    "watching": ("watching", "voyeur", "waiting turn", "partner watches", "onlooker"),
-    "aftercare": ("aftercare", "cuddling", "kissing after", "holding close", "post-sex"),
-    "cleanup": ("cleanup", "wiping", "cleaning", "towel", "wet cloth"),
-    "fingering": ("fingering", "fingers inside", "fingers in pussy", "finger stimulation"),
-    "clit_rubbing": ("clit", "clitoris", "clit rubbing", "rubbing the clit", "fingers on clit"),
-    "mutual_masturbation": ("mutual masturbation", "both touching themselves", "masturbating together", "hands on their own bodies"),
-    "boobjob": ("boobjob", "titjob", "breast-sex", "breast sex"),
-    "testicle_sucking": ("testicle", "balls-licking", "balls licking", "balls and mouth"),
-    "penis_licking": ("penis-licking", "penis licking", "tongue along", "tongue licking"),
-    "handjob": ("handjob", "hand job", "stroking the penis", "hand stroking", "manual stimulation"),
-    "footjob": ("footjob", "soles", "toes curled", "feet stroking"),
-    "open_thighs": ("thighs open", "legs spread", "open thighs", "legs open", "reclining with thighs open"),
-    "front_back": ("front-and-back", "front and back", "one behind and one in front", "between two partners"),
-}
-HARDCORE_POSITION_AXIS_KEYS = {
-    "position",
-    "body_position",
-    "body_arrangement",
-    "arrangement",
-    "tease_act",
-    "touch_detail",
-    "manual_act",
-    "manual_detail",
-    "worship_act",
-    "transition_act",
-    "control_act",
-    "performance_act",
-    "coordination_act",
-    "aftercare_act",
-    "cleanup_detail",
-}
-
-HARDCORE_SOURCE_FAMILY_BY_SUBCATEGORY = {
-    "penetrative_sex": "penetrative",
-    "foreplay_teasing": "foreplay",
-    "body_worship_touching": "interaction",
-    "clothing_position_transitions": "interaction",
-    "dominant_guidance": "interaction",
-    "camera_performance": "interaction",
-    "manual_stimulation": "manual",
-    "oral_sex": "oral",
-    "outercourse_sex": "outercourse",
-    "anal_double_penetration": "anal",
-    "threesomes": "threesome",
-    "group_coordination": "interaction",
-    "group_sex_orgy": "group",
-    "cumshot_climax": "climax",
-    "aftercare_cleanup": "interaction",
-}
+HARDCORE_POSITION_FAMILY_CHOICES = hardcore_position_policy.HARDCORE_POSITION_FAMILY_CHOICES
+HARDCORE_POSITION_FOCUS_CHOICES = hardcore_position_policy.HARDCORE_POSITION_FOCUS_CHOICES
+HARDCORE_POSITION_KEY_CHOICES = hardcore_position_policy.HARDCORE_POSITION_KEY_CHOICES
+HARDCORE_POSITION_FAMILY_SUBCATEGORIES = hardcore_position_policy.HARDCORE_POSITION_FAMILY_SUBCATEGORIES
+HARDCORE_POSITION_KEY_MATCHES = hardcore_position_policy.HARDCORE_POSITION_KEY_MATCHES
+HARDCORE_POSITION_AXIS_KEYS = hardcore_position_policy.HARDCORE_POSITION_AXIS_KEYS
+HARDCORE_SOURCE_FAMILY_BY_SUBCATEGORY = hardcore_position_policy.HARDCORE_SOURCE_FAMILY_BY_SUBCATEGORY
 
 
 def _hardcore_source_position_family(subcategory: dict[str, Any], config: dict[str, Any] | None = None) -> str:
-    slug = str(subcategory.get("slug") or subcategory.get("name") or "").strip().lower()
-    family = HARDCORE_SOURCE_FAMILY_BY_SUBCATEGORY.get(slug, "")
-    if family:
-        return family
-    config_family = _normalize_hardcore_position_family((config or {}).get("family"), "")
-    return "" if config_family == "any" else config_family
+    return hardcore_position_policy.hardcore_source_position_family(subcategory, config)
 
 
 def _hardcore_position_keys(*parts: Any, axis_values: dict[str, Any] | None = None) -> list[str]:
-    text_parts = [str(part or "") for part in parts if str(part or "").strip()]
-    if isinstance(axis_values, dict):
-        text_parts.extend(str(value or "") for value in axis_values.values() if str(value or "").strip())
-    text = " ".join(text_parts).lower()
-    if not text:
-        return []
-    keys: list[str] = []
-    for key, tokens in HARDCORE_POSITION_KEY_MATCHES.items():
-        if any(token in text for token in tokens):
-            keys.append(key)
-    return keys
+    return hardcore_position_policy.hardcore_position_keys(*parts, axis_values=axis_values)
 
 
 CAMERA_ORBIT_FRAMING_CHOICES = camera_policy.CAMERA_ORBIT_FRAMING_CHOICES
@@ -1266,104 +1068,23 @@ def _parse_filter_config(filter_config: str | dict[str, Any] | None) -> dict[str
 
 
 def _normalize_hardcore_position_family(value: Any, default: str = "any") -> str:
-    text = str(value or default).strip()
-    return text if text in HARDCORE_POSITION_FAMILY_CHOICES else default
+    return hardcore_position_policy.normalize_hardcore_position_family(value, default)
 
 
 def _normalize_hardcore_position_values(values: Any) -> list[str]:
-    raw_values = _list_from(values)
-    selected: list[str] = []
-    for value in raw_values:
-        text = str(value or "").strip()
-        if not text or text == "any":
-            continue
-        normalized = re.sub(r"[^a-z0-9]+", "_", text.lower()).strip("_")
-        if normalized in HARDCORE_POSITION_KEY_CHOICES and normalized not in selected:
-            selected.append(normalized)
-    return selected
+    return hardcore_position_policy.normalize_hardcore_position_values(values)
 
 
 def _empty_hardcore_position_config() -> dict[str, Any]:
-    return {
-        "config_type": "hardcore_position",
-        "enabled": False,
-        "family": "any",
-        "positions": [],
-        "require_position": False,
-        "allow_toys": True,
-        "allow_double": True,
-        "allow_penetration": True,
-        "allow_foreplay": True,
-        "allow_interaction": True,
-        "allow_manual": True,
-        "allow_oral": True,
-        "allow_outercourse": True,
-        "allow_anal": True,
-        "allow_climax": True,
-    }
+    return hardcore_position_policy.empty_hardcore_position_config()
 
 
 def _parse_hardcore_position_config(value: str | dict[str, Any] | None) -> dict[str, Any]:
-    if not value:
-        return _empty_hardcore_position_config()
-    if isinstance(value, dict):
-        raw = value
-    else:
-        try:
-            raw = json.loads(str(value))
-        except json.JSONDecodeError:
-            return _empty_hardcore_position_config()
-    if not isinstance(raw, dict):
-        return _empty_hardcore_position_config()
-    parsed = {**_empty_hardcore_position_config(), **raw}
-    parsed["enabled"] = bool(parsed.get("enabled", True))
-    parsed["family"] = _normalize_hardcore_position_family(parsed.get("family"))
-    parsed["positions"] = _normalize_hardcore_position_values(parsed.get("positions"))
-    parsed["require_position"] = not _is_false(parsed.get("require_position", False))
-    for key in (
-        "allow_toys",
-        "allow_double",
-        "allow_penetration",
-        "allow_foreplay",
-        "allow_interaction",
-        "allow_manual",
-        "allow_oral",
-        "allow_outercourse",
-        "allow_anal",
-        "allow_climax",
-    ):
-        parsed[key] = not _is_false(parsed.get(key, True))
-    return parsed
+    return hardcore_position_policy.parse_hardcore_position_config(value)
 
 
 def _hardcore_position_summary(config: dict[str, Any]) -> str:
-    if not config.get("enabled"):
-        return "hardcore position unrestricted"
-    parts = [f"family={config.get('family', 'any')}"]
-    positions = config.get("positions") or []
-    if positions:
-        parts.append("positions=" + ",".join(positions))
-    elif config.get("require_position"):
-        parts.append("position_templates=required")
-    disabled = [
-        label
-        for key, label in (
-            ("allow_toys", "toys"),
-            ("allow_double", "double"),
-            ("allow_penetration", "penetration"),
-            ("allow_foreplay", "foreplay"),
-            ("allow_interaction", "interaction"),
-            ("allow_manual", "manual"),
-            ("allow_oral", "oral"),
-            ("allow_outercourse", "outercourse"),
-            ("allow_anal", "anal"),
-            ("allow_climax", "climax"),
-        )
-        if not config.get(key, True)
-    ]
-    if disabled:
-        parts.append("blocked=" + ",".join(disabled))
-    return "; ".join(parts)
+    return hardcore_position_policy.hardcore_position_summary(config)
 
 
 def build_hardcore_position_pool_json(
@@ -1372,24 +1093,12 @@ def build_hardcore_position_pool_json(
     family: str = "any",
     selected_positions: list[str] | tuple[str, ...] | str | None = None,
 ) -> str:
-    base = _parse_hardcore_position_config(hardcore_position_config)
-    if combine_mode == "replace":
-        base = {**_empty_hardcore_position_config(), "enabled": True}
-    else:
-        base["enabled"] = True
-    base["family"] = _normalize_hardcore_position_family(family, base.get("family", "any"))
-    selected = _normalize_hardcore_position_values(selected_positions)
-    if combine_mode == "add":
-        existing = list(base.get("positions") or [])
-        for value in selected:
-            if value not in existing:
-                existing.append(value)
-        base["positions"] = existing
-    else:
-        base["positions"] = selected
-    base["require_position"] = bool(base.get("require_position")) or bool(base["positions"]) or base["family"] != "any"
-    base["summary"] = _hardcore_position_summary(base)
-    return json.dumps(base, ensure_ascii=True, sort_keys=True)
+    return hardcore_position_policy.build_hardcore_position_pool_json(
+        hardcore_position_config=hardcore_position_config,
+        combine_mode=combine_mode,
+        family=family,
+        selected_positions=selected_positions,
+    )
 
 
 def build_hardcore_action_filter_json(
@@ -1406,84 +1115,28 @@ def build_hardcore_action_filter_json(
     allow_anal: bool = True,
     allow_climax: bool = True,
 ) -> str:
-    config = _parse_hardcore_position_config(hardcore_position_config)
-    config["enabled"] = True
-    focus = str(focus or "keep_pool").strip()
-    focus_family = {
-        "penetration_only": "penetrative",
-        "foreplay_only": "foreplay",
-        "interaction_only": "interaction",
-        "manual_only": "manual",
-        "oral_only": "oral",
-        "outercourse_only": "outercourse",
-        "anal_only": "anal",
-        "climax_only": "climax",
-        "threesome_only": "threesome",
-        "group_only": "group",
-    }.get(focus)
-    if focus_family:
-        config["family"] = focus_family
-    config["allow_toys"] = bool(allow_toys)
-    config["allow_double"] = bool(allow_double)
-    config["allow_penetration"] = bool(allow_penetration)
-    config["allow_foreplay"] = bool(allow_foreplay)
-    config["allow_interaction"] = bool(allow_interaction)
-    config["allow_manual"] = bool(allow_manual)
-    config["allow_oral"] = bool(allow_oral)
-    config["allow_outercourse"] = bool(allow_outercourse)
-    config["allow_anal"] = bool(allow_anal)
-    config["allow_climax"] = bool(allow_climax)
-
-    if not focus_family and config["family"] != "any":
-        enabled_action_families = {
-            family
-            for enabled, family in (
-                (config["allow_penetration"], "penetrative"),
-                (config["allow_foreplay"], "foreplay"),
-                (config["allow_interaction"], "interaction"),
-                (config["allow_manual"], "manual"),
-                (config["allow_oral"], "oral"),
-                (config["allow_outercourse"], "outercourse"),
-                (config["allow_anal"], "anal"),
-                (config["allow_climax"], "climax"),
-            )
-            if enabled
-        }
-        if config["family"] in enabled_action_families and len(enabled_action_families) > 1:
-            config["family"] = "any"
-
-    if focus == "foreplay_only":
-        config["allow_foreplay"] = True
-        config["allow_interaction"] = True
-    elif focus == "interaction_only":
-        config["allow_interaction"] = True
-        config["allow_foreplay"] = True
-    elif focus == "manual_only":
-        config["allow_manual"] = True
-    elif focus == "oral_only":
-        config["allow_oral"] = True
-        config["allow_penetration"] = False
-    elif focus == "outercourse_only":
-        config["allow_outercourse"] = True
-        config["allow_oral"] = False
-        config["allow_penetration"] = False
-    elif focus == "anal_only":
-        config["allow_anal"] = True
-        config["allow_penetration"] = True
-    elif focus == "climax_only":
-        config["allow_climax"] = True
-    config["summary"] = _hardcore_position_summary(config)
-    return json.dumps(config, ensure_ascii=True, sort_keys=True)
+    return hardcore_position_policy.build_hardcore_action_filter_json(
+        hardcore_position_config=hardcore_position_config,
+        focus=focus,
+        allow_toys=allow_toys,
+        allow_double=allow_double,
+        allow_penetration=allow_penetration,
+        allow_foreplay=allow_foreplay,
+        allow_interaction=allow_interaction,
+        allow_manual=allow_manual,
+        allow_oral=allow_oral,
+        allow_outercourse=allow_outercourse,
+        allow_anal=allow_anal,
+        allow_climax=allow_climax,
+    )
 
 
 def _hardcore_position_config_active(config: dict[str, Any]) -> bool:
-    return bool(config.get("enabled"))
+    return hardcore_position_policy.hardcore_position_config_active(config)
 
 
 def _hardcore_position_template_required(config: dict[str, Any]) -> bool:
-    if not _hardcore_position_config_active(config):
-        return False
-    return bool(config.get("require_position")) or bool(config.get("positions")) or _normalize_hardcore_position_family(config.get("family")) != "any"
+    return hardcore_position_policy.hardcore_position_template_required(config)
 
 
 def _is_hardcore_sexual_category(category: dict[str, Any]) -> bool:
@@ -1491,37 +1144,7 @@ def _is_hardcore_sexual_category(category: dict[str, Any]) -> bool:
 
 
 def _hardcore_allowed_subcategory_slugs(config: dict[str, Any]) -> set[str]:
-    family = _normalize_hardcore_position_family(config.get("family"))
-    allowed = set(HARDCORE_POSITION_FAMILY_SUBCATEGORIES.get(family, HARDCORE_POSITION_FAMILY_SUBCATEGORIES["any"]))
-    if not config.get("allow_penetration", True):
-        allowed.difference_update({"penetrative_sex", "anal_double_penetration", "threesomes", "group_sex_orgy"})
-    if not config.get("allow_foreplay", True):
-        allowed.discard("foreplay_teasing")
-    if not config.get("allow_interaction", True):
-        allowed.difference_update(
-            {
-                "foreplay_teasing",
-                "body_worship_touching",
-                "clothing_position_transitions",
-                "dominant_guidance",
-                "camera_performance",
-                "group_coordination",
-                "aftercare_cleanup",
-            }
-        )
-    if not config.get("allow_manual", True):
-        allowed.discard("manual_stimulation")
-    if not config.get("allow_oral", True):
-        allowed.discard("oral_sex")
-    if not config.get("allow_outercourse", True):
-        allowed.discard("outercourse_sex")
-    if not config.get("allow_anal", True):
-        allowed.discard("anal_double_penetration")
-    if not config.get("allow_climax", True):
-        allowed.discard("cumshot_climax")
-    if not config.get("allow_double", True) and family == "anal":
-        allowed.add("anal_double_penetration")
-    return allowed or set(HARDCORE_POSITION_FAMILY_SUBCATEGORIES["any"])
+    return hardcore_position_policy.hardcore_allowed_subcategory_slugs(config)
 
 
 def _filter_hardcore_categories_for_position(
@@ -1993,15 +1616,15 @@ def hardcore_detail_density_choices() -> list[str]:
 
 
 def hardcore_position_family_choices() -> list[str]:
-    return list(HARDCORE_POSITION_FAMILY_CHOICES)
+    return hardcore_position_policy.hardcore_position_family_choices()
 
 
 def hardcore_position_focus_choices() -> list[str]:
-    return list(HARDCORE_POSITION_FOCUS_CHOICES)
+    return hardcore_position_policy.hardcore_position_focus_choices()
 
 
 def hardcore_position_key_choices() -> list[str]:
-    return list(HARDCORE_POSITION_KEY_CHOICES)
+    return hardcore_position_policy.hardcore_position_key_choices()
 
 
 def character_softcore_outfit_source_choices() -> list[str]: