horny jail
This commit is contained in:
198
pick-most-skin
Executable file
198
pick-most-skin
Executable file
@@ -0,0 +1,198 @@
|
||||
#!/usr/bin/env python3
|
||||
#
|
||||
# Scan ~/webGoggles (or given path) for all session screenshots.
|
||||
# For each user (site/username), pick the one with the highest
|
||||
# skin-to-total pixel ratio and save it as most_skin.png
|
||||
# next to their bio_screenshot.png.
|
||||
#
|
||||
# Usage:
|
||||
# ./pick-most-skin [--force] [path ...]
|
||||
#
|
||||
# If no path given, defaults to ~/webGoggles.
|
||||
# Multiple paths can be specified.
|
||||
#
|
||||
# Requires: opencv-python (pip install opencv-python)
|
||||
|
||||
import os
|
||||
import sys
|
||||
import glob as _glob
|
||||
import shutil
|
||||
|
||||
# If cv2 isn't on the normal path, check pipx-installed opencv-python
|
||||
try:
|
||||
import cv2
|
||||
import numpy as np
|
||||
except ImportError:
|
||||
_pipx_venv = os.path.expanduser(
|
||||
"~/.local/pipx/venvs/opencv-python/lib/python*/site-packages"
|
||||
)
|
||||
_matches = sorted(_glob.glob(_pipx_venv))
|
||||
if _matches:
|
||||
sys.path.insert(0, _matches[-1])
|
||||
import cv2
|
||||
import numpy as np
|
||||
else:
|
||||
print("Error: opencv-python not installed.")
|
||||
print(" pip install opencv-python")
|
||||
sys.exit(1)
|
||||
|
||||
|
||||
# HSV ranges for skin tones (tunable)
|
||||
SKIN_RANGES = [
|
||||
((0, 20, 70), (20, 255, 255)), # warm/peach
|
||||
((170, 20, 70), (180, 255, 255)), # reddish wrap-around
|
||||
]
|
||||
|
||||
|
||||
def skin_ratio(path: str) -> float:
|
||||
"""Return fraction of pixels that look like skin (0.0 – 1.0)."""
|
||||
img = cv2.imread(path)
|
||||
if img is None:
|
||||
return 0.0
|
||||
hsv = cv2.cvtColor(img, cv2.COLOR_BGR2HSV)
|
||||
mask = np.zeros(img.shape[:2], dtype=np.uint8)
|
||||
for lower, upper in SKIN_RANGES:
|
||||
mask |= cv2.inRange(hsv, np.array(lower, dtype=np.uint8),
|
||||
np.array(upper, dtype=np.uint8))
|
||||
return float(cv2.countNonZero(mask)) / (img.shape[0] * img.shape[1])
|
||||
|
||||
|
||||
def find_most_skin(site_user: str) -> tuple[float, str | None]:
|
||||
"""Walk sessions under site/user/, return (best_ratio, best_path)."""
|
||||
sessions_dir = os.path.join(site_user, "sessions")
|
||||
if not os.path.isdir(sessions_dir):
|
||||
return 0.0, None
|
||||
|
||||
best_ratio = 0.0
|
||||
best_path = None
|
||||
|
||||
for root, dirs, files in os.walk(sessions_dir):
|
||||
if "screenshot.png" not in files:
|
||||
continue
|
||||
path = os.path.join(root, "screenshot.png")
|
||||
ratio = skin_ratio(path)
|
||||
if ratio > best_ratio:
|
||||
best_ratio = ratio
|
||||
best_path = path
|
||||
|
||||
return best_ratio, best_path
|
||||
|
||||
|
||||
def main() -> None:
|
||||
args = [a for a in sys.argv[1:] if not a.startswith("-")]
|
||||
force = "--force" in sys.argv[1:]
|
||||
|
||||
roots = args if args else [os.path.expanduser("~/webGoggles")]
|
||||
|
||||
for root in roots:
|
||||
root = os.path.abspath(root)
|
||||
if not os.path.isdir(root):
|
||||
print(f"[pick-most-skin] Skipping {root} – not a directory")
|
||||
continue
|
||||
|
||||
print(f"[pick-most-skin] Scanning {root}")
|
||||
|
||||
# Walk site/user directories
|
||||
for site in sorted(os.listdir(root)):
|
||||
site_path = os.path.join(root, site)
|
||||
if not os.path.isdir(site_path):
|
||||
continue
|
||||
for user in sorted(os.listdir(site_path)):
|
||||
user_path = os.path.join(site_path, user)
|
||||
if not os.path.isdir(user_path):
|
||||
continue
|
||||
dest = os.path.join(user_path, "most_skin.png")
|
||||
|
||||
if os.path.exists(dest) and not force:
|
||||
print(f" {site}/{user}: most_skin.png exists (use --force to re-scan)")
|
||||
continue
|
||||
|
||||
ratio, best = find_most_skin(user_path)
|
||||
if best is None:
|
||||
continue
|
||||
|
||||
shutil.copy2(best, dest)
|
||||
print(f" {site}/{user}: {ratio:.1%} skin -> {dest}")
|
||||
|
||||
# Build/refresh GALLERY symlinks
|
||||
gallery = os.path.join(root, "GALLERY")
|
||||
os.makedirs(gallery, exist_ok=True)
|
||||
|
||||
# Remove stale symlinks
|
||||
for entry in os.listdir(gallery):
|
||||
if entry == "index.html":
|
||||
continue
|
||||
entry_path = os.path.join(gallery, entry)
|
||||
if os.path.islink(entry_path) or entry.endswith("_ms.png"):
|
||||
os.unlink(entry_path)
|
||||
|
||||
entries = []
|
||||
for site in sorted(os.listdir(root)):
|
||||
site_path = os.path.join(root, site)
|
||||
if not os.path.isdir(site_path):
|
||||
continue
|
||||
for user in sorted(os.listdir(site_path)):
|
||||
user_path = os.path.join(site_path, user)
|
||||
most_skin = os.path.join(user_path, "most_skin.png")
|
||||
if not os.path.isfile(most_skin):
|
||||
continue
|
||||
link_name = f"{user}_ms.png"
|
||||
full_link = os.path.join(gallery, link_name)
|
||||
os.symlink(os.path.relpath(most_skin, gallery), full_link)
|
||||
entries.append((user, link_name))
|
||||
|
||||
# Generate gallery HTML
|
||||
html_path = os.path.join(gallery, "index.html")
|
||||
count = len(entries)
|
||||
rows = "".join(
|
||||
f' <a href="#img-{i}" class="thumb"><img src="{fn}" loading="lazy"><span>{u}</span></a>\n'
|
||||
for i, (u, fn) in enumerate(entries)
|
||||
)
|
||||
lbs = "".join(
|
||||
f' <div id="img-{i}" class="lightbox"><a href="#"><img src="{fn}"></a></div>\n'
|
||||
for i, (u, fn) in enumerate(entries)
|
||||
)
|
||||
html = f"""<!DOCTYPE html>
|
||||
<html lang="en">
|
||||
<head>
|
||||
<meta charset="UTF-8">
|
||||
<meta name="viewport" content="width=device-width, initial-scale=1">
|
||||
<title>webGoggles Gallery</title>
|
||||
<style>
|
||||
* {{ margin: 0; padding: 0; box-sizing: border-box; }}
|
||||
body {{ background: #111; color: #ccc; font: 14px/1.4 system-ui, sans-serif; padding: 20px; }}
|
||||
h1 {{ font-size: 1.2rem; margin-bottom: 16px; color: #888; font-weight: 400; }}
|
||||
.grid {{ display: grid; grid-template-columns: repeat(auto-fill, minmax(200px, 1fr)); gap: 12px; }}
|
||||
.thumb {{ display: block; position: relative; border-radius: 6px; overflow: hidden; background: #1a1a1a; text-decoration: none; }}
|
||||
.thumb img {{ width: 100%; height: 260px; object-fit: cover; display: block; }}
|
||||
.thumb span {{ display: block; padding: 6px 10px; font-size: .8rem; color: #999; white-space: nowrap; overflow: hidden; text-overflow: ellipsis; }}
|
||||
.thumb:hover img {{ opacity: .85; }}
|
||||
.lightbox {{ display: none; position: fixed; inset: 0; z-index: 100; background: rgba(0,0,0,.94); align-items: center; justify-content: center; }}
|
||||
.lightbox:target {{ display: flex; }}
|
||||
.lightbox img {{ max-width: 95vw; max-height: 95vh; object-fit: contain; border-radius: 4px; cursor: zoom-out; }}
|
||||
</style>
|
||||
</head>
|
||||
<body>
|
||||
<h1>GALLERY · {count} users</h1>
|
||||
<div class="grid">
|
||||
{rows}</div>
|
||||
{lbs}<script>
|
||||
document.addEventListener('keydown', e => {{
|
||||
const m = location.hash.match(/^#img-(\\d+)$/);
|
||||
if (!m) return;
|
||||
const i = +m[1], n = {count};
|
||||
if (e.key === 'ArrowRight') location.hash = '#img-' + (i + 1 < n ? i + 1 : 0);
|
||||
if (e.key === 'ArrowLeft') location.hash = '#img-' + (i > 0 ? i - 1 : n - 1);
|
||||
if (e.key === 'Escape') location.hash = '#';
|
||||
}});
|
||||
</script></body>
|
||||
</html>"""
|
||||
with open(html_path, "w") as f:
|
||||
f.write(html)
|
||||
print(f" GALLERY/index.html — {len(entries)} entries")
|
||||
|
||||
print("[pick-most-skin] Done")
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
main()
|
||||
Reference in New Issue
Block a user