datas = [
      {
          "_id": 1,
          "insert_time": time.time(),
          "update_time": time.time(),
          "name": "zs"
      },
      {
          "_id": 2,
          "insert_time": time.time(),
          "update_time": time.time(),
          "name": "ls"
      },
      {
          "_id": 3,
          "insert_time": time.time(),
          "update_time": time.time(),
          "name": "ww"
      },
  ]

# -*- coding: utf-8 -*-

# @Author: Толстый, очень тонкий
# @Date: 2024-03-03 12:58:42
# @LastEditors: Толстый, очень тонкий
# @LastEditTime: 2024-03-03 13:57:49

# Импортировать пакет
import time
from pymongo import MongoClient as MC
from pymongo import UpdateOne
client = MC()["test"]


def exists_update_and_insert(data, bulk=False):
    """
    Обновить, если оно существует
    Вставьте, если нет
    :param data: данные
    :param bulk: Использовать ли пакетную вставку
    # ordered
    # упорядоченное исполнение, Сообщение об ошибке, Не будет выполнено позже
    # внеочередное исполнение, Сообщение об ошибке, Остальные не затронуты
    """

    if bulk:
        bulk_docs = []
        for rd in data:
            bulk_docs.append(
                UpdateOne(
                    {"_id": rd["_id"]},
                    {"$set": rd},
                    upsert=True
                ))
        result = client['goods'].bulk_write(bulk_docs, ordered=False).bulk_api_result
        print(result)
    else:
        for rd in data:
            client['goods'].update_one(
                {'_id': rd['_id']}, 
                {'$set': rd}, 
                upsert=True
            )


def exists_do_noting_and_insert(data, bulk=False):
    """
    Если он существует, ничего не делайте.
    Вставьте, если нет
    :param data: данные
    :param bulk: Использовать ли пакетную вставку
    """

    if bulk:
        bulk_docs = []
        for rd in data:
            bulk_docs.append(
                UpdateOne(
                    {"_id": rd["_id"]},
                    {"$setOnInsert": rd},
                    upsert=True
                ))
        result = client['goods'].bulk_write(bulk_docs, ordered=False).bulk_api_result
        print(result)
    else:
        for rd in data:
            client['goods'].update_one(
                {'_id': rd['_id']}, 
                {'$setOnInsert': rd}, 
                upsert=True
            )


def exists_update_any_field_and_insert(data, bulk=False):
    """
    Обновить, если оно существует部分字段
    Вставьте, если нет
    :param data: данные
    :param bulk: Использовать ли пакетную вставку
    """

    if bulk:
        bulk_docs = []
        for rd in data:
            update_time = rd.pop("update_time", time.time())
            bulk_docs.append(
                UpdateOne(
                    {"_id": rd["_id"]},
                    {
                        "$setOnInsert": rd,
                        "$set": {"update_time": update_time}
                    },
                    upsert=True
                ))
        result = client['goods'].bulk_write(bulk_docs, ordered=False).bulk_api_result
        print(result)
    else:
        for rd in data:
            update_time = rd.pop("update_time", time.time())
            client['goods'].update_one(
                {'_id': rd['_id']},
                {
                    '$setOnInsert': rd,
                    "$set": {"update_time": update_time}
                },
                upsert=True
            )


def find(query=None):
    """
    查询данные    :query: Условия запроса
    """
    return list(client['goods'].find(query))


def remove(query=None):
    """
    удалитьданные    :query: удалить условие
    """
    return client["goods"].remove(query)


if __name__ == '__main__':
    # тестданные    datas = [
        {
            "_id": 1,
            "insert_time": time.time(),
            "update_time": time.time(),
            "name": "zs"
        },
        {
            "_id": 2,
            "insert_time": time.time(),
            "update_time": time.time(),
            "name": "ls"
        },
        {
            "_id": 3,
            "insert_time": time.time(),
            "update_time": time.time(),
            "name": "ww"
        },
    ]

    """
    тестовый пример
    """
    # exists_update_and_insert(datas)
    # exists_update_and_insert(datas, True)
    # print(find())
    # remove()

    # exists_do_noting_and_insert(datas)
    # exists_do_noting_and_insert(datas, True)
    # print(find())
    # remove()


    # exists_update_any_field_and_insert(datas)
    # exists_update_any_field_and_insert(datas, True)
    # print(find())
    # remove()