SHA256
67 lines
3.4 KiB
Python
67 lines
3.4 KiB
Python
#!/usr/bin/env python3
|
|
import argparse,json,hashlib,mimetypes,time
|
|
from pathlib import Path
|
|
import requests
|
|
try: import arweave
|
|
except ImportError: arweave=None
|
|
|
|
def atomic(path,obj):
|
|
p=Path(path); p.parent.mkdir(parents=True,exist_ok=True)
|
|
t=p.with_suffix(p.suffix+'.tmp'); t.write_text(json.dumps(obj,ensure_ascii=False,indent=2),encoding='utf8'); t.replace(p)
|
|
|
|
def exists(gateway,txid,timeout=15):
|
|
url=f"{gateway.rstrip('/')}/{txid}"
|
|
try:
|
|
r=requests.head(url,allow_redirects=True,timeout=timeout)
|
|
if r.status_code in (200,204): return True
|
|
if r.status_code in (405,501):
|
|
r=requests.get(url,headers={'Range':'bytes=0-0'},timeout=timeout)
|
|
return r.status_code in (200,206)
|
|
return False
|
|
except requests.RequestException: return False
|
|
|
|
def main():
|
|
ap=argparse.ArgumentParser(description='Prepare images for all SHiNE test channels')
|
|
ap.add_argument('--config',default='prepare.config.json'); a=ap.parse_args()
|
|
cfg=json.load(open(a.config,encoding='utf8')); base=Path(a.config).resolve().parent
|
|
rp=lambda s:(Path(s) if Path(s).is_absolute() else (base/s).resolve())
|
|
gateway=cfg['arweave'].get('gateway','https://arweave.net').rstrip('/')
|
|
statep=rp(cfg.get('upload_state_file','upload-state.json'))
|
|
state=json.loads(statep.read_text()) if statep.exists() else {'version':2,'files':{}}
|
|
wallet=None
|
|
def ensure_wallet():
|
|
nonlocal wallet
|
|
if wallet is None:
|
|
if arweave is None: raise SystemExit('pip install -r requirements.txt')
|
|
wallet=arweave.Wallet(str(rp(cfg['arweave']['wallet_file'])))
|
|
return wallet
|
|
|
|
summary={'channels':0,'posts':0,'images':0,'verified_existing':0,'uploaded':0}
|
|
for ch in cfg['channels']:
|
|
rawp=rp(ch['queue_file']); outp=rp(ch['prepared_queue_file'])
|
|
items=json.loads(rawp.read_text(encoding='utf8')); prepared=[]; summary['channels']+=1
|
|
for n,item in enumerate(items,1):
|
|
refs=[]
|
|
for rel in item.get('images',[]):
|
|
fp=(rawp.parent/rel).resolve(); data=fp.read_bytes(); sha=hashlib.sha256(data).hexdigest()
|
|
summary['images']+=1; rec=state['files'].get(sha); txid=None
|
|
if rec and exists(gateway,rec['id']):
|
|
txid=rec['id']; summary['verified_existing']+=1
|
|
print(f"[{ch['name']}:{n}] verified {fp.name} -> {txid}")
|
|
else:
|
|
ctype=mimetypes.guess_type(fp.name)[0] or 'application/octet-stream'
|
|
tx=arweave.Transaction(ensure_wallet(),data=data)
|
|
tx.add_tag('Content-Type',ctype); tx.add_tag('App-Name','SHiNE-Test-Publisher'); tx.add_tag('SHA-256',sha)
|
|
tx.sign(); tx.send(); txid=tx.id; summary['uploaded']+=1
|
|
state['files'][sha]={'id':txid,'file':str(fp),'uploaded_at':int(time.time())}; atomic(statep,state)
|
|
print(f"[{ch['name']}:{n}] uploaded {fp.name} -> {txid}")
|
|
refs.append({'arweave_id':txid,'url':f'{gateway}/{txid}','sha256':sha})
|
|
text=item['text'].rstrip()
|
|
if refs: text += '\n\n' + '\n'.join(x['url'] for x in refs)
|
|
prepared.append({'id':item.get('id',str(n)),'text':text,'images':refs})
|
|
summary['posts']+=1
|
|
atomic(outp,prepared)
|
|
print(f"Ready {ch['name']}: {outp} ({len(prepared)} posts)")
|
|
print(json.dumps(summary,ensure_ascii=False,indent=2))
|
|
if __name__=='__main__': main()
|