mirror of
https://github.com/stashapp/CommunityScripts.git
synced 2026-02-10 20:52:23 -06:00
73 lines
2.1 KiB
Python
73 lines
2.1 KiB
Python
import sys, json
|
|
from PythonDepManager import ensure_import
|
|
ensure_import("dateparser>=1.2.1")
|
|
ensure_import("stashapi:stashapp-tools>=0.2.58")
|
|
import stashapi.log as log
|
|
from stashapi.stashapp import StashInterface
|
|
import re
|
|
from dateparser import parse
|
|
|
|
|
|
def main():
|
|
global stash
|
|
global pattern
|
|
|
|
pattern = re.compile(
|
|
r"\D(\d{4}|\d{1,2})[\._\- /\\](\d{1,2}|[a-zA-Z]{3,}\.*)[\._\- /\\](\d{4}|\d{1,2})\D"
|
|
)
|
|
json_input = json.loads(sys.stdin.read())
|
|
mode_arg = json_input["args"]["mode"]
|
|
|
|
stash = StashInterface(json_input["server_connection"])
|
|
|
|
if mode_arg == "gallery":
|
|
find_date_for_galleries()
|
|
|
|
|
|
def parse_date_candidate(string):
|
|
result = None
|
|
for match in pattern.finditer(string):
|
|
g1 = match.group(1)
|
|
g2 = match.group(2)
|
|
g3 = match.group(3)
|
|
temp = parse(g1 + " " + g2 + " " + g3)
|
|
if temp:
|
|
result = temp.strftime("%Y-%m-%d")
|
|
return result
|
|
|
|
|
|
def find_date_for_galleries():
|
|
|
|
galleries = stash.find_galleries(f={"is_missing": "date"})
|
|
|
|
total = len(galleries)
|
|
|
|
log.info(f"Found {total} galleries")
|
|
|
|
for i, gallery in enumerate(galleries):
|
|
log.progress(i / total)
|
|
acceptableDate = None
|
|
|
|
for file in gallery.get("files", []):
|
|
if candidate := parse_date_candidate(file["path"]):
|
|
acceptableDate = candidate
|
|
|
|
if "folder" in gallery and gallery["folder"]:
|
|
if "path" in gallery["folder"] and gallery["folder"]["path"]:
|
|
if candidate := parse_date_candidate(gallery["folder"]["path"]):
|
|
acceptableDate = candidate
|
|
|
|
if acceptableDate:
|
|
log.info(
|
|
"Gallery ID ("
|
|
+ gallery.get("id")
|
|
+ ") has matched the date : "
|
|
+ acceptableDate
|
|
)
|
|
updateObject = {"id": gallery.get("id"), "date": acceptableDate}
|
|
stash.update_gallery(updateObject)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|