mirror of
https://github.com/lifegpc/pythonscript.git
synced 2026-06-06 11:28:58 +08:00
114 lines
3.3 KiB
Python
114 lines
3.3 KiB
Python
from argparse import ArgumentParser
|
|
from ehapi import EHC
|
|
from sys import exit
|
|
from re import compile
|
|
|
|
|
|
BGI = compile(r'^(tmb_|ex_)?([ehm]v\d+[a-z_]*|scene\d+)\.(jpg|png)$')
|
|
KRKR = compile(r'^(E[VC]H?\d+H?(_\d+[a-z_]*)?|img_.*)\.(jpg|png)$')
|
|
p = ArgumentParser()
|
|
p.add_argument('-b', '--base', default='http://localhost:8080', help='API Host') # noqa: E501
|
|
p.add_argument('-t', '--token', help='Token')
|
|
p.add_argument('-p', '--print', action='store_true', default=False, help='Print page') # noqa: E501
|
|
p.add_argument('-d', '--dryrun', action='store_true', default=False, help='Dry run') # noqa: E501
|
|
p.add_argument('gid', help='Gallery id', type=int)
|
|
p.add_argument('typ', nargs='?', default='auto', choices=['auto', 'bgi', 'krkr'], help='Engine') # noqa: E501
|
|
arg = p.parse_intermixed_args()
|
|
|
|
|
|
def mark(page, changes, is_nsfw=None, is_ad=None):
|
|
token = page['token']
|
|
index = page['index']
|
|
name = page['name']
|
|
width = page['width']
|
|
height = page['height']
|
|
t = []
|
|
if is_nsfw is not None and page['is_nsfw'] != is_nsfw:
|
|
t.append('NSFW' if is_nsfw else 'SFW')
|
|
if is_ad is not None and page['is_ad'] != is_ad:
|
|
t.append('AD' if is_ad else 'non-AD')
|
|
if not len(t):
|
|
return
|
|
changes.append({"token": token, 'is_ad': is_ad, 'is_nsfw': is_nsfw})
|
|
print(f'Mark {index}.{name}({width}x{height}) as {", ".join(t)}')
|
|
|
|
|
|
def bgi_check_pages(pages, changes):
|
|
for p in pages:
|
|
name: str = p['name']
|
|
if name.startswith("tmb_"):
|
|
name = name[4:]
|
|
is_nsfw = name.startswith("hv") or name.startswith("scene")
|
|
mark(p, changes, is_nsfw, True)
|
|
elif name.startswith("hv"):
|
|
mark(p, changes, True, False)
|
|
else:
|
|
mark(p, changes, False, False)
|
|
|
|
|
|
def krkr_check_pages(pages, changes):
|
|
for p in pages:
|
|
name: str = p['name']
|
|
ns = name.split('_')
|
|
if ns[0].startswith("EVH"):
|
|
mark(p, changes, True, False)
|
|
elif ns[0].startswith("EV") and ns[0].endswith("H"):
|
|
mark(p, changes, True, False)
|
|
else:
|
|
mark(p, changes, False, False)
|
|
|
|
|
|
def guess_as_bgi(pages):
|
|
c = 0
|
|
for p in pages:
|
|
name: str = p['name']
|
|
if name.startswith('_') or BGI.match(name):
|
|
c += 1
|
|
return c / len(pages)
|
|
|
|
|
|
def guess_as_krkr(pages):
|
|
c = 0
|
|
for p in pages:
|
|
name: str = p['name']
|
|
if name.startswith('_') or KRKR.match(name):
|
|
c += 1
|
|
return c / len(pages)
|
|
|
|
|
|
def guess_engine(pages):
|
|
bgi = guess_as_bgi(pages)
|
|
print('BGI coverage:', bgi)
|
|
if bgi >= 0.9:
|
|
return 'bgi'
|
|
krkr = guess_as_krkr(pages)
|
|
print('KRKR coverage:', krkr)
|
|
if krkr >= 0.9:
|
|
return krkr
|
|
return None
|
|
|
|
|
|
with EHC() as c:
|
|
c.base = f"{arg.base}/api"
|
|
if arg.token:
|
|
c.ses.headers.update({'X-TOKEN': arg.token})
|
|
gallery = c.get_gallery(arg.gid)
|
|
pages = gallery['pages']
|
|
if arg.print:
|
|
for p in pages:
|
|
print(p)
|
|
exit(0)
|
|
typ = arg.typ
|
|
if typ == 'auto':
|
|
typ = guess_engine(pages)
|
|
if not typ:
|
|
raise ValueError('Unknown engine.')
|
|
changes = []
|
|
if typ == 'bgi':
|
|
bgi_check_pages(pages, changes)
|
|
elif typ == 'krkr':
|
|
krkr_check_pages(pages, changes)
|
|
if arg.dryrun:
|
|
exit(0)
|
|
c.update_file_meta_json(changes)
|