mirror of
https://github.com/stashapp/CommunityScripts.git
synced 2026-02-05 04:45:09 -06:00
228c294ffa
* fix css files * create workflow * format code using prettier * black ignore 3rd party folder * format py code using black * black exclude option in workflow * Don't format workflow files * Add contributing instructions to readme --------- Co-authored-by: WithoutPants <53250216+WithoutPants@users.noreply.github.com>
75 lines
1.9 KiB
Python
75 lines
1.9 KiB
Python
import json
|
|
import os
|
|
import sys
|
|
import time
|
|
|
|
import config
|
|
|
|
# import log
|
|
import graphql
|
|
|
|
API_VERSION_BF_FILES = 31 # APP/DB Schema version prior to files refactor PR
|
|
MAX_RETRY_COUNT = 25
|
|
SLEEP_RETRY = 0.5
|
|
|
|
FRAGMENT = json.loads(sys.stdin.read())
|
|
# log.LogDebug(json.dumps(FRAGMENT))
|
|
FRAGMENT_SERVER = FRAGMENT["server_connection"]
|
|
FRAGMENT_SCENE_ID = FRAGMENT["args"].get("hookContext")
|
|
|
|
if FRAGMENT_SCENE_ID:
|
|
scene_id = FRAGMENT_SCENE_ID["id"]
|
|
else:
|
|
graphql.exit_plugin("No ID found")
|
|
|
|
graphql_port = FRAGMENT_SERVER["Port"]
|
|
graphql_scheme = FRAGMENT_SERVER["Scheme"]
|
|
graphql_session = FRAGMENT_SERVER.get("SessionCookie").get("Value")
|
|
|
|
system_status = graphql.get_api_version(
|
|
port=graphql_port, session=graphql_session, scheme=graphql_scheme
|
|
)
|
|
|
|
api_version = system_status.get("appSchema")
|
|
|
|
basename = None
|
|
|
|
if api_version > API_VERSION_BF_FILES: # only needed for versions after files refactor
|
|
files_base = graphql.get_scene_base(
|
|
scene_id=scene_id,
|
|
port=graphql_port,
|
|
session=graphql_session,
|
|
scheme=graphql_scheme,
|
|
)
|
|
if len(files_base["files"]) > 0:
|
|
basename = files_base["files"][0].get("basename")
|
|
else:
|
|
graphql.exit_plugin(
|
|
f"Stash with API version:{api_version} is not supported. You need at least {API_VERSION_BF_FILES}"
|
|
)
|
|
|
|
if basename is None:
|
|
graphql.exit_plugin("No basename found") # file-less scene
|
|
|
|
if config.STRIP_EXT:
|
|
basename = os.path.splitext(basename)[0]
|
|
|
|
i = MAX_RETRY_COUNT
|
|
while i >= 0:
|
|
# log.LogDebug(f"TitleFromFilename: Retry attempt {i}")
|
|
i -= 1
|
|
updated_scene = graphql.update_scene_title(
|
|
scene_id,
|
|
basename,
|
|
port=graphql_port,
|
|
session=graphql_session,
|
|
scheme=graphql_scheme,
|
|
)
|
|
if updated_scene:
|
|
graphql.exit_plugin(
|
|
f"Scene title updated after {MAX_RETRY_COUNT - i} tries. Title:{updated_scene.get('title')}"
|
|
)
|
|
time.sleep(SLEEP_RETRY)
|
|
|
|
graphql.exit_plugin("Error updating scene")
|