mirror of
https://github.com/stashapp/CommunityScripts.git
synced 2025-12-17 03:30:51 -06:00
49e4156bb0
* Parse All Galleries for date - Break out the date parsing to a separate function. - Expand gallery search to more than single file zip galleries. - Also scan the gallery's folder path if available. The final valid parsed date of the files, then the gallery folder will be taken as the gallery's date. This will preserve the previous behavior of single file zip galleries, as there will be only one path to ever consider. * Revert to previous logic Previous logic returned the last possible valid date string in the path string, which is the more correct approach when a parent directory might also have a date string. * formatting * bump version * Cleanup lingering prototype line
71 lines
1.9 KiB
Python
71 lines
1.9 KiB
Python
import sys, json
|
|
|
|
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()
|