Как создать функцию которая отвечала бы за создание пользователя в MongoDB PyThon?

Question

[Nemo Dev]

У меня есть код который функционирует:

import pymongo
from pymongo import MongoClient
import asyncio

from loguru import logger
logger.add("./utils/logs/file.log", format="{time:YYYY-MM-DD at HH:mm:ss} | {level} | {message}")

logger.info("Подключаю базу данных...")
client = MongoClient(host="localhost", port=27017)
logger.debug("База данных подключена!")
    
db = client.tgbot
    
user = db["users"]

uid = user.count_documents({})
x = { "uid": uid, "id": 1, "tag": "test"}
res = user.insert_one(x)
print(res)

Как из этого можно сделать функцию? (create_user, в последних 4 строчках генерируется пользователь)

Comments

[Алан Гибизов]

Покажите свою попытку создать функцию.

[Nemo Dev]

Алан Гибизов, т.к я не нашел данной темы на хабре, я сделал простое решение для тех кто не знает, внизу мой ответ

Answer 1

[Nemo Dev]

Вместо:

uid = user.count_documents({})
x = { "uid": uid, "id": 1, "tag": "test"}
res = user.insert_one(x)
print(res)

Функция:

def create_user(id, tag):
    uid = user.count_documents({})

    x = { "uid": uid, "id": id, "tag": tag}
    res = user.insert_one(x)

Comments

[Алан Гибизов]

Тут как минимум хорошо бы сделать чистую функцию, которая не меняет ничего вне своей области видимости, а вместо этого возвращает необходимый объект, который после вызова функции можно «присвоить» нужной переменной в нужной области видимости.

def create_user(id, tag):
    uid = user.count_documents({})
    x = { "uid": uid, "id": id, "tag": tag}
    return user.insert_one(x)

result = create_user(id=my_id, tag=my_tag)
print(result)

[Алан Гибизов]

Также неплохо бы проверять, что приходит в параметры функции, и обрабатывать соответствующие ошибки.

[Nemo Dev]

Алан Гибизов, тут важна ситуация, т.к я беру данные с апи телеграма а там данные всегда будут таковы, айди предоставляется, тэг тоже (имя). Что на счет того что вызывать нужный объект, для этого будет сделана другая функция. get_user, если судить по моей ситуации то тут в данный момент все хорошо, все зависит от гибкости данных, если нужна обработка ошибок то можно добавить.

[Алан Гибизов]

Nemo Dev, да делайте как хотите, кто я такой, чтобы вам мешать? :) Потом придете с любимым тут вопросом «почему у None нет метода “send”» или что-то в таком роде.
Или тоже традиционное «а как сделать, чтобы работало с разными пользователями одновременно и не путалось»