Add complete history lib
This commit is contained in:
parent
b0333ab908
commit
1dd680e7ae
|
@ -0,0 +1,29 @@
|
||||||
|
#!/usr/bin/env python3
|
||||||
|
|
||||||
|
import sys, argparse, os
|
||||||
|
from os.path import join, dirname
|
||||||
|
from shutil import copyfile
|
||||||
|
from dotenv import load_dotenv
|
||||||
|
from libs.historylib import History
|
||||||
|
|
||||||
|
# Get variables environment
|
||||||
|
if not os.path.isfile('.env'):
|
||||||
|
copyfile(".env.template", ".env")
|
||||||
|
dotenv_path = join(dirname(__file__), '.env')
|
||||||
|
load_dotenv(dotenv_path)
|
||||||
|
|
||||||
|
dunikey = os.getenv('DUNIKEY')
|
||||||
|
node = os.getenv('NODE')
|
||||||
|
if not node:
|
||||||
|
sys.stderr.write("Please fill a Duniter node in .env file\n")
|
||||||
|
sys.exit(1)
|
||||||
|
|
||||||
|
# Parse arguments
|
||||||
|
parser = argparse.ArgumentParser()
|
||||||
|
parser.add_argument('-p', '--pubkey', help="Clé publique du compte visé")
|
||||||
|
parser.add_argument('--mempool', action='store_true', help="Utilise les sources en Mempool")
|
||||||
|
args = parser.parse_args()
|
||||||
|
|
||||||
|
# Create transaction and send it
|
||||||
|
hist = History(dunikey, node, args.pubkey, args.mempool)
|
||||||
|
result = hist.history()
|
|
@ -0,0 +1,108 @@
|
||||||
|
#!/usr/bin/env python3
|
||||||
|
|
||||||
|
import sys, re, os.path, json, ast
|
||||||
|
from datetime import datetime
|
||||||
|
from termcolor import colored
|
||||||
|
from 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 History:
|
||||||
|
|
||||||
|
def __init__(self, dunikey, node, pubkey, useMempool=False):
|
||||||
|
self.dunikey = dunikey
|
||||||
|
self.pubkey = pubkey if pubkey else get_privkey(dunikey, "pubsec").pubkey
|
||||||
|
self.useMempool = useMempool
|
||||||
|
if not re.match(PUBKEY_REGEX, self.pubkey) or len(self.pubkey) > 45:
|
||||||
|
sys.stderr.write("La clé publique n'est pas au bon format.\n")
|
||||||
|
sys.exit(1)
|
||||||
|
|
||||||
|
# Define Duniter GVA node
|
||||||
|
transport = AIOHTTPTransport(url=node)
|
||||||
|
self.client = Client(transport=transport, fetch_schema_from_transport=True)
|
||||||
|
|
||||||
|
def sendDoc(self):
|
||||||
|
# Build TX generation document
|
||||||
|
queryBuild = gql(
|
||||||
|
"""
|
||||||
|
query ($pubkey: String!){
|
||||||
|
transactionsHistory(pubkey: $pubkey) {
|
||||||
|
received {
|
||||||
|
writtenTime
|
||||||
|
issuers
|
||||||
|
outputs
|
||||||
|
comment
|
||||||
|
}
|
||||||
|
sent {
|
||||||
|
writtenTime
|
||||||
|
issuers
|
||||||
|
outputs
|
||||||
|
comment
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
"""
|
||||||
|
)
|
||||||
|
paramsBuild = {
|
||||||
|
"pubkey": self.pubkey
|
||||||
|
}
|
||||||
|
|
||||||
|
# Send TX document
|
||||||
|
try:
|
||||||
|
self.historyDoc = self.client.execute(queryBuild, variable_values=paramsBuild)
|
||||||
|
except Exception as e:
|
||||||
|
message = ast.literal_eval(str(e))["message"]
|
||||||
|
sys.stderr.write("Echec de récupération de l'historique:\n" + message + "\n")
|
||||||
|
sys.exit(1)
|
||||||
|
|
||||||
|
|
||||||
|
def parseHistory(self):
|
||||||
|
|
||||||
|
res = self.historyDoc['transactionsHistory']['received']
|
||||||
|
transIn=[[0 for x in range(0)] for y in range(len(res))]
|
||||||
|
for i, bloc in enumerate(res):
|
||||||
|
for output in bloc['outputs']:
|
||||||
|
if re.search(self.pubkey, output):
|
||||||
|
transIn[i].append("IN")
|
||||||
|
transIn[i].append(bloc['writtenTime'])
|
||||||
|
transIn[i].append(bloc['issuers'][0])
|
||||||
|
transIn[i].append(int(output.split(':')[0])/100)
|
||||||
|
transIn[i].append(bloc['comment'])
|
||||||
|
|
||||||
|
|
||||||
|
res = self.historyDoc['transactionsHistory']['sent']
|
||||||
|
transOut=[[0 for x in range(0)] for y in range(len(res))]
|
||||||
|
i = 0
|
||||||
|
for bloc in res:
|
||||||
|
for output in bloc['outputs']:
|
||||||
|
if not re.search(self.pubkey, output):
|
||||||
|
transOut[i].append("OUT")
|
||||||
|
transOut[i].append(bloc['writtenTime'])
|
||||||
|
transOut[i].append(output.split("SIG(")[1].replace(')',''))
|
||||||
|
transOut[i].append(int(output.split(':')[0])/100)
|
||||||
|
transOut[i].append(bloc['comment'])
|
||||||
|
i += 1
|
||||||
|
|
||||||
|
trans = transIn + transOut
|
||||||
|
trans = list(filter(None, trans))
|
||||||
|
trans.sort(key=lambda x: x[1])
|
||||||
|
|
||||||
|
# Get terminal size
|
||||||
|
rows = int(os.popen('stty size', 'r').read().split()[1])
|
||||||
|
|
||||||
|
print('-'.center(rows, '-'))
|
||||||
|
print("{: <20} | {: <45} | {: <7} | {: <30}".format(" Date"," De la part de (clé publique)","Montant","Commentaire"))
|
||||||
|
for i in trans:
|
||||||
|
color = "green" if i[0] == "IN" else "blue"
|
||||||
|
date = datetime.fromtimestamp(i[1]).strftime("%d/%m/%Y à %H:%M")
|
||||||
|
print('-'.center(rows, '-'))
|
||||||
|
print(colored("{: <20} | {: <45} | {: <7} | {: <30}".format(date, *i[2:]), color))
|
||||||
|
# print(*i)
|
||||||
|
|
||||||
|
|
||||||
|
def history(self):
|
||||||
|
self.sendDoc()
|
||||||
|
self.parseHistory()
|
||||||
|
|
Reference in New Issue