create a utility for pinning sicker packs
Signed-off-by: Jakub Sokołowski <jakub@status.im>
This commit is contained in:
parent
bdb06a90bd
commit
3ff020d23d
|
@ -0,0 +1,53 @@
|
|||
#!/usr/bin/env python3
|
||||
|
||||
import re
|
||||
import json
|
||||
import requests
|
||||
import content_hash # https://pypi.org/project/content-hash
|
||||
import ipfscluster # https://pypi.org/project/ipfscluster
|
||||
from multiaddr import Multiaddr
|
||||
|
||||
IPFS_CLUSTER_ADDR = 'http://localhost:9094'
|
||||
STICKER_PACKS_META_URLS = [
|
||||
"https://cloudflare-ipfs.com/ipfs/QmWVVLwVKCwkVNjYJrRzQWREVvEk917PhbHYAUhA1gECTM",
|
||||
"https://cloudflare-ipfs.com/ipfs/QmWpG2Q5NB472KLgFysdCjB8D1Qf9hxR2KNJvtCJQJufDj",
|
||||
]
|
||||
|
||||
# https://github.com/ethereum/EIPs/blob/master/EIPS/eip-1577.md
|
||||
content_hash_rgx = r'e30101701220\w+'
|
||||
|
||||
content_hashes = []
|
||||
|
||||
for url in STICKER_PACKS_META_URLS:
|
||||
resp = requests.get(url)
|
||||
resp.raise_for_status()
|
||||
matches = re.findall(content_hash_rgx, resp.text)
|
||||
content_hashes.extend(matches)
|
||||
|
||||
# IPFS can't handle EIP-1577 content hashes
|
||||
decoded_hashes = [content_hash.decode(ch) for ch in content_hashes]
|
||||
|
||||
class IpfsPinner:
|
||||
|
||||
def __init__(self, addr=ipfscluster.DEFAULT_ADDR):
|
||||
self.client = ipfscluster.connect(addr)
|
||||
|
||||
def is_pinned(self, chash):
|
||||
resp = self.client.pins.ls(chash)
|
||||
statuses = [peer['status'] for peer in resp['peer_map'].values()]
|
||||
return all(s == 'pinned' for s in statuses), statuses
|
||||
|
||||
def pin(self, chash):
|
||||
return self.client.pins.add(chash)
|
||||
|
||||
ip = IpfsPinner()
|
||||
|
||||
for chash in decoded_hashes:
|
||||
pinned, statuses = ip.is_pinned(chash)
|
||||
print('{} - {}'.format(chash, statuses))
|
||||
|
||||
if pinned:
|
||||
continue
|
||||
|
||||
ip.pin(chash)
|
||||
print('{} - PINNED'.format(chash))
|
Loading…
Reference in New Issue