mirror of
https://github.com/more-tech4-magnum-opus/backend.git
synced 2024-12-01 07:13:43 +03:00
131 lines
3.5 KiB
Python
131 lines
3.5 KiB
Python
from pydantic import BaseModel
|
|
import requests as r
|
|
import json
|
|
from time import sleep
|
|
from typing import List
|
|
|
|
URL = 'https://hackathon.lsp.team/hk'
|
|
|
|
base_headers = {
|
|
'Content-Type': 'application/json',
|
|
'Accept': 'application/json'
|
|
}
|
|
|
|
|
|
class Wallet(BaseModel):
|
|
publicKey: str
|
|
privateKey: str
|
|
|
|
|
|
class WalletBalance(BaseModel):
|
|
matic: float
|
|
coins: float
|
|
|
|
|
|
class TransHash(BaseModel):
|
|
transaction_hash: str
|
|
|
|
|
|
class NftInfo(BaseModel):
|
|
uri: str
|
|
tokens: List[int]
|
|
|
|
|
|
class TransactionHistory(BaseModel):
|
|
hash: str
|
|
timestamp: int
|
|
token_name: str
|
|
from_wallet: str
|
|
to_wallet: str
|
|
|
|
|
|
def create_wallet() -> Wallet:
|
|
response = r.post(URL+'/v1/wallets/new')
|
|
data = response.json()
|
|
return Wallet(publicKey=data['publicKey'], privateKey=data['privateKey'])
|
|
|
|
|
|
def check_transaction(trans_hash: str) -> str:
|
|
res = r.get(URL + '/v1/transfers/status/'+trans_hash, data=json.dumps({
|
|
'transactionHash': trans_hash
|
|
}),
|
|
headers=base_headers
|
|
)
|
|
return res.json()['status']
|
|
|
|
|
|
def transfer_rubbles(my_wallet_private_key, transfer_publick_key: str, amount: float) -> TransHash:
|
|
response = r.post(URL+'/v1/transfers/ruble', data=json.dumps({
|
|
"fromPrivateKey": my_wallet_private_key,
|
|
"toPublicKey": transfer_publick_key,
|
|
"amount": amount
|
|
}),
|
|
headers={
|
|
'Content-Type': 'application/json',
|
|
'Accept': 'application/json'
|
|
}
|
|
)
|
|
return TransHash(transaction_hash=response.json()['transaction'])
|
|
|
|
|
|
def create_nft(wallet_public_key: str, string_url: str) -> TransHash:
|
|
response = r.post(URL + '/v1/nft/generate', data=json.dumps(
|
|
{
|
|
"toPublicKey": wallet_public_key,
|
|
"uri": string_url,
|
|
"nftCount": 1
|
|
}
|
|
),
|
|
headers=base_headers
|
|
)
|
|
return TransHash(transaction_hash=response.json()['transaction_hash'])
|
|
|
|
|
|
def get_nfts(wallet_public_key: str) -> List[NftInfo]:
|
|
res = r.get(URL+f'/v1/wallets/{wallet_public_key}/nft/balance')
|
|
print(res.json())
|
|
return list(
|
|
map(
|
|
lambda res: NftInfo(
|
|
uri=res['uri'],
|
|
tokens=res['tokens']
|
|
),
|
|
res.json()['balance']
|
|
),
|
|
)
|
|
|
|
def get_history(wallet_public_key: str) -> List[TransactionHistory]:
|
|
res = r.post(URL+f'/v1/wallets/{wallet_public_key}/history', data=json.dumps({
|
|
"page": 0,
|
|
"offset": 100,
|
|
"sort": "asc"
|
|
|
|
}), headers=base_headers)
|
|
return list(
|
|
map(
|
|
lambda res: TransactionHistory(
|
|
hash=res['hash'],
|
|
timestamp=res['timeStamp'],
|
|
token_name=res['tokenName'],
|
|
from_wallet=res['from'],
|
|
to_wallet=res['to']
|
|
),
|
|
res.json()['history']
|
|
),
|
|
)
|
|
|
|
|
|
def transfer_nft(my_private_wallet_key: str, transfer_publick_key: str, token_id: int) -> TransHash:
|
|
res = r.post(URL+'/v1/transfers/nft', data=json.dumps(
|
|
{
|
|
"fromPrivateKey": my_private_wallet_key,
|
|
"toPublicKey": transfer_publick_key,
|
|
"tokenId": token_id
|
|
}
|
|
), headers=base_headers)
|
|
return TransHash(transaction_hash=res.json()['transaction_hash'])
|
|
|
|
def get_balance(my_public_wallet_key: str) -> WalletBalance:
|
|
res = r.get(URL+f'/v1/wallets/{my_public_wallet_key}/balance')
|
|
return WalletBalance(matic=res.json()['maticAmount'], coins=res.json()['coinsAmount'])
|