mirror of
https://github.com/autistic-symposium/blockchain-data-engineering-toolkit.git
synced 2025-05-17 22:20:29 -04:00
52 lines
1,023 B
Python
52 lines
1,023 B
Python
import motor.motor_asyncio
|
|
import json
|
|
MONGO_DETAILS = "mongodb://localhost:27017"
|
|
|
|
#client = motor.motor_asyncio.AsyncIOMotorClient(MONGO_DETAILS)
|
|
|
|
from pymongo import MongoClient
|
|
|
|
client = MongoClient(MONGO_DETAILS)
|
|
|
|
database = client.balances
|
|
|
|
collection = database.get_collection("balances_fuckkk")
|
|
|
|
|
|
print(database.list_collection_names())
|
|
|
|
|
|
def wallet_helper(item) -> dict:
|
|
return {
|
|
"wallet": item["wallet"],
|
|
}
|
|
|
|
|
|
from bson.objectid import ObjectId
|
|
|
|
async def retrieve_students():
|
|
bals = collection.find()
|
|
|
|
res = []
|
|
counter = 0
|
|
for i in bals:
|
|
res.append(wallet_helper(i))
|
|
if counter > 2:
|
|
break
|
|
counter += 1
|
|
|
|
return res
|
|
|
|
|
|
def balancer_helper(item) -> dict:
|
|
return {
|
|
"wallet": item["wallet"],
|
|
"balance": item["balance"],
|
|
}
|
|
|
|
|
|
# Retrieve a student with a matching ID
|
|
async def retrieve_student(wallet: str) -> dict:
|
|
student = collection.find_one({"wallet": wallet})
|
|
if student:
|
|
return balancer_helper(student)
|