Kapitan
5a97d610a7
Some checks failed
Gitea/kapitanbooru-uploader/pipeline/head There was a failure building this commit
Now with tagger Miejsce na zdjęcie Linki do wiki Zapis ustawień Tagger działa w tle Kolorujemy pliki po ratingu Tagger cache Tagi w bazie Pobranie implikacji tagów Autocomplete Podział na pliki i skrypty + nowe API Structure for package Version 0.1.0
125 lines
4.0 KiB
Python
125 lines
4.0 KiB
Python
import subprocess
|
|
|
|
from bs4 import BeautifulSoup
|
|
import requests
|
|
|
|
from .settings import Settings
|
|
|
|
|
|
def open_tag_wiki_url(tag, settings: Settings):
|
|
"""Otwiera w przeglądarce URL strony wiki dla podanego tagu."""
|
|
# Usuń prefiksy
|
|
for prefix in [
|
|
"character:",
|
|
"artist:",
|
|
"meta:",
|
|
"copyright:",
|
|
"general:",
|
|
]:
|
|
if tag.startswith(prefix):
|
|
tag = tag[len(prefix) :]
|
|
break
|
|
|
|
url = "https://danbooru.donmai.us/wiki_pages/" + tag
|
|
open_webbrowser(url, settings)
|
|
|
|
|
|
def open_webbrowser(url, settings: Settings):
|
|
"""Otwiera URL w wybranej przeglądarce (lub domyślnej)."""
|
|
if settings.browser:
|
|
try:
|
|
subprocess.run([settings.browser, url], check=True)
|
|
return
|
|
except Exception as e:
|
|
print("Błąd przy otwieraniu przeglądarki:", e)
|
|
import webbrowser
|
|
|
|
webbrowser.open(url)
|
|
|
|
|
|
def login(settings: Settings):
|
|
"""
|
|
Log in to the server using settings and return a session with cookies.
|
|
|
|
settings should have:
|
|
- base_url (e.g., "https://example.com")
|
|
- username
|
|
- password
|
|
"""
|
|
# Construct the full URL for login
|
|
url = settings.base_url.rstrip("/") + "/user_admin/login"
|
|
|
|
# Prepare the payload for URL-encoded form data
|
|
payload = {"user": settings.username, "pass": settings.password}
|
|
|
|
# Set the proper header
|
|
headers = {"Content-Type": "application/x-www-form-urlencoded"}
|
|
|
|
# Use a session so that cookies are automatically stored for future requests.
|
|
session = requests.Session()
|
|
|
|
# Send the POST request and prevent automatic redirects,
|
|
# so we can capture the 302 response with Set-Cookie headers.
|
|
response = session.post(url, data=payload, headers=headers, allow_redirects=False)
|
|
|
|
if response.status_code == 302:
|
|
# The session's cookie jar should now contain the cookies set by the server.
|
|
shm_user = session.cookies.get("shm_user")
|
|
shm_session = session.cookies.get("shm_session")
|
|
|
|
if not (shm_user and shm_session):
|
|
raise Exception("Login succeeded, but expected cookies were not set.")
|
|
|
|
print("Login successful. Cookies stored in session:")
|
|
print(f"shm_user: {shm_user}")
|
|
print(f"shm_session: {shm_session}")
|
|
|
|
return session
|
|
else:
|
|
raise Exception(f"Login failed: {response.status_code} - {response.text}")
|
|
|
|
|
|
def get_auth_token(session, settings):
|
|
"""
|
|
Given a logged-in session and settings, fetch the user page
|
|
and extract the auth_token from the hidden input field.
|
|
|
|
settings should have:
|
|
- base_url (e.g., "https://example.com")
|
|
The session should contain the 'shm_user' cookie.
|
|
"""
|
|
# Retrieve the user identifier from cookies
|
|
shm_user = session.cookies.get("shm_user")
|
|
if not shm_user:
|
|
raise Exception("shm_user cookie not found; login might have failed.")
|
|
|
|
# Build the URL to fetch, e.g., /user/<shm_user>
|
|
user_url = f"{settings.base_url.rstrip('/')}/user/{shm_user}"
|
|
|
|
# Option 1: Simply allow redirects (if your server sends 302 and eventually a 200)
|
|
# response = session.get(user_url) # redirects allowed by default
|
|
|
|
# Option 2: If you want to control redirection manually, disable them:
|
|
# response = session.get(user_url, allow_redirects=False)
|
|
# Then you might have to follow the redirects manually.
|
|
|
|
# For simplicity, we'll allow redirects:
|
|
response = session.get(user_url)
|
|
|
|
if response.status_code != 200:
|
|
raise Exception(
|
|
f"Failed to load {user_url}, status code: {response.status_code}"
|
|
)
|
|
|
|
# Parse the returned HTML with BeautifulSoup
|
|
soup = BeautifulSoup(response.text, "html.parser")
|
|
|
|
# Look for the hidden input with name "auth_token"
|
|
auth_input = soup.find("input", {"name": "auth_token"})
|
|
if auth_input and auth_input.has_attr("value"):
|
|
auth_token = auth_input["value"]
|
|
print(f"Found auth_token: {auth_token}")
|
|
return auth_token
|
|
else:
|
|
raise Exception("auth_token not found in the HTML page.")
|