mirror of
https://github.com/th30d4y/OpenLearnX.git
synced 2026-05-26 19:26:33 +00:00
5t4l1n
31 lines
1007 B
Python
31 lines
1007 B
Python
from bson import ObjectId
|
|
from datetime import datetime
|
|
from pymongo.collection import Collection
|
|
|
|
class UserModel:
|
|
def __init__(self, collection: Collection):
|
|
self.collection = collection
|
|
|
|
async def get_by_wallet(self, wallet_address: str):
|
|
return await self.collection.find_one({"wallet_address": wallet_address.lower()})
|
|
|
|
async def create_user(self, wallet_address: str):
|
|
now = datetime.utcnow()
|
|
user = {
|
|
"wallet_address": wallet_address.lower(),
|
|
"created_at": now,
|
|
"last_login": now,
|
|
"total_tests": 0,
|
|
"certificates": []
|
|
}
|
|
result = await self.collection.insert_one(user)
|
|
user["_id"] = result.inserted_id
|
|
return user
|
|
|
|
async def update_last_login(self, wallet_address: str):
|
|
now = datetime.utcnow()
|
|
await self.collection.update_one(
|
|
{"wallet_address": wallet_address.lower()},
|
|
{"$set": {"last_login": now}}
|
|
)
|