104 lines
3.2 KiB
Python
Executable File
104 lines
3.2 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
|
|
import sys, re, os.path, json, ast
|
|
from termcolor import colored
|
|
from lib.natools import fmt, sign, get_privkey
|
|
from gql import gql, Client
|
|
from gql.transport.aiohttp import AIOHTTPTransport
|
|
|
|
PUBKEY_REGEX = "(?![OIl])[1-9A-Za-z]{42,45}"
|
|
|
|
|
|
class ListWallets:
|
|
def __init__(self, node, brut, mbr, nonMbr, larf, map=False):
|
|
self.mbr = mbr
|
|
self.larf = larf
|
|
self.nonMbr = nonMbr
|
|
self.brut = brut
|
|
self.map = map
|
|
# Define Duniter GVA node
|
|
transport = AIOHTTPTransport(url=node)
|
|
self.client = Client(transport=transport, fetch_schema_from_transport=True)
|
|
|
|
def sendDoc(self):
|
|
# Build wallets generation document
|
|
|
|
queryBuild = gql(
|
|
"""
|
|
{
|
|
wallets(pagination: { cursor: null, ord: ASC, pageSize: 0 }) {
|
|
pageInfo {
|
|
hasNextPage
|
|
endCursor
|
|
}
|
|
edges {
|
|
node {
|
|
script
|
|
balance {
|
|
amount
|
|
base
|
|
}
|
|
idty {
|
|
isMember
|
|
username
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
"""
|
|
)
|
|
|
|
# Send wallets document
|
|
try:
|
|
queryResult = self.client.execute(queryBuild)
|
|
except Exception as e:
|
|
sys.stderr.write("Echec de récupération de la liste:\n" + str(e) + "\n")
|
|
sys.exit(1)
|
|
|
|
jsonBrut = queryResult["wallets"]["edges"]
|
|
|
|
walletList = []
|
|
walletMap = {}
|
|
for i, trans in enumerate(jsonBrut):
|
|
dataWork = trans["node"]
|
|
if self.mbr and (
|
|
dataWork["idty"] == None or dataWork["idty"]["isMember"] == False
|
|
):
|
|
continue
|
|
if self.nonMbr and (
|
|
dataWork["idty"] == None or dataWork["idty"]["isMember"] == True
|
|
):
|
|
continue
|
|
if self.larf and (dataWork["idty"] != None):
|
|
continue
|
|
|
|
if self.map:
|
|
walletMap[dataWork["script"]] = {
|
|
"balance": dataWork["balance"]["amount"] / 100,
|
|
"id": dataWork["idty"],
|
|
}
|
|
else:
|
|
walletList.append(
|
|
{
|
|
"pubkey": dataWork["script"],
|
|
"balance": dataWork["balance"]["amount"] / 100,
|
|
"id": dataWork["idty"],
|
|
}
|
|
)
|
|
|
|
if self.brut:
|
|
names = []
|
|
for dataWork in walletList:
|
|
if self.mbr or self.nonMbr:
|
|
names.append(dataWork["pubkey"] + " " + dataWork["id"]["username"])
|
|
else:
|
|
names.append(dataWork["pubkey"])
|
|
|
|
return "\n".join(names)
|
|
else:
|
|
if self.map:
|
|
return json.dumps(walletMap, indent=2)
|
|
else:
|
|
return json.dumps(walletList, indent=2)
|