mirror of
https://github.com/more-tech4-magnum-opus/backend.git
synced 2024-09-21 10:08:47 +03:00
135 lines
3.5 KiB
Python
135 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"]
|
|
)
|