from PyrogramBot import db

from typing import Dict, List, Union

notesdb = db.notes

async def get_notes_count() -> dict:

chats = notesdb.find({"chat_id": {"$lt": 0}})

if not chats:

return {}

chats_count = 0

notes_count = 0

for chat in await chats.to_list(length=1000000000):

notes_name = await get_note_names(chat['chat_id'])

notes_count += len(notes_name)

chats_count += 1

return {

"chats_count": chats_count,

"notes_count": notes_count

}

async def _get_notes(chat_id: int) -> Dict[str, int]:

_notes = await notesdb.find_one({"chat_id": chat_id})

if _notes:

_notes = _notes["notes"]

else:

_notes = {}

return _notes

async def get_note_names(chat_id: int) -> List[str]:

_notes = []

for note in await _get_notes(chat_id):

_notes.append(note)

return _notes

async def get_note(chat_id: int, name: str) -> Union[bool, dict]:

name = name.lower().strip()

_notes = await _get_notes(chat_id)

if name in _notes:

return _notes[name]

else:

return False

async def save_note(chat_id: int, name: str, note: dict):

name = name.lower().strip()

_notes = await _get_notes(chat_id)

_notes[name] = note

await notesdb.update_one(

{"chat_id": chat_id},

{

"$set": {

"notes": _notes

}

},

upsert=True

)

async def delete_note(chat_id: int, name: str) -> bool:

notesd = await _get_notes(chat_id)

name = name.lower().strip()

if name in notesd:

del notesd[name]

await notesdb.update_one(

{"chat_id": chat_id},

{

"$set": {

"notes": notesd

}

},

upsert=True

)

return True

return False