deposit.py

from ast import While
from urllib import response
import hikari
import requests
from constants import baseUrl
import os
import json
import time
from showcoins import ShowCoins
# deposits Cloudcoins from .bin /.png files sent as attachment by the user to the users wallet
async def Deposit(wallet, event: hikari.DMMessageCreateEvent):
    nftWalletName =  "NFTs." + wallet
    # encode # as special character to be used in url
    walletget = wallet.replace("#","%23")
    # check if wallet exists
    checkWalletUrl = baseUrl + 'wallets/' + walletget + '?contents=false'
    response = requests.get(checkWalletUrl)
    responsejson = response.json()
    fullpath = os.path.join(os.getcwd(), 'import')
    # create a new one is wallet does not exists
    if(responsejson['status'] != 'success'):
        print('Wallet does not exist.Creating new one')
        createWalletUrl = baseUrl + 'wallets'
        walletJson = { 'name': wallet}
        nftwalletJson = {'name': nftWalletName}
        createresponse = requests.post(createWalletUrl, json= walletJson)
        createresponsejson = createresponse.json()
        nftcreateresponse = requests.post(createWalletUrl, json= nftwalletJson)
        nftcreateresponsejson = nftcreateresponse.json()
        
        if(createresponsejson['status'] == 'success'):
            print('Wallet Created successfully')
            await event.message.respond("New Wallet created for you. your wallet name is:" + wallet)
    print('Depositing', len(event.message.attachments), ' files')
    await event.message.respond('Starting Coins deposit...')
    # check if there are any attachments, else response with error message
    if (len(event.message.attachments) == 0):
        await event.message.respond('No Coins found')
    else:
        for coin in event.message.attachments:
            try:
                fdata = await coin.read()
                filename = os.path.join(fullpath, coin.filename)
                await event.message.respond('Processing file: ' + coin.filename)
                # copy cloudcoins to user's respective import folder
                with open(filename, "wb") as binary_file:
                    binary_file.write(fdata)
                depositUrl = baseUrl + 'import'
                depositJson = {"name": wallet, "items":[{"type":"file", "data":filename}]}
                json_string = json.dumps(depositJson) 
                depositresponse = requests.post(depositUrl, json_string)
                depositresponsejson = depositresponse.json()
                depositstatus = depositresponsejson['payload']['status']
                TASK_URL = baseUrl + 'tasks/' + depositresponsejson['payload']['id']
                # poll for task status till status is changed to completed

                while depositstatus == 'running':
                    taskresponse = requests.get(TASK_URL)
                    taskresponsejson = taskresponse.json()
                    depositstatus = taskresponsejson['payload']['status']
                    time.sleep(2)
                if(depositstatus == 'completed'):
                    # calculate stats if deposit api call is successful and send a response to user
                    authentic = taskresponsejson['payload']['data']['pown_results']['authentic']
                    counterfeit = taskresponsejson['payload']['data']['pown_results']['counterfeit']
                    total = taskresponsejson['payload']['data']['pown_results']['total']
                    unknown = taskresponsejson['payload']['data']['pown_results']['unknown']
                    fracked = taskresponsejson['payload']['data']['pown_results']['fracked']
                    await event.message.respond('Coins imported..\nTotal: ' + str(total) + '\nAuthentic: '+ str(authentic + fracked) +  '\nCounterfeit:' + str(counterfeit) + '\nUnknown: ' + str(unknown))
            except:
                await event.message.respond('An Error occurred while depositing '+ coin.filename + '. Please check your file or try again')
            # show the updated balance of user's wallet
            await ShowCoins(wallet, event)

    return ''