# coding=utf-8
# __author__ : "shyorange"
# __date__ :  2018/12/3

import pymongo;
# 导入生成日期的类
import time;
from datetime import datetime;
from hashlib import md5;

"""
该应用的数据库文件
"""

class TODO_mongo:

    def __init__(self):
        self.__conn = pymongo.MongoClient("127.0.0.1",27017);
        self.__db = self.__conn["TODO"];
        self.__collection = self.__db["first_todo"];

    def __do_data(self, data):
        """
        对数据进行包装，将数据包装成字典格式
        :return: data
        """
        now_time = datetime.now();
        # 根据当前时间为每一条数据生成一个唯一标示
        sign = md5(str(now_time).encode("utf-8")).hexdigest();
        # 其中的is_over表示是否完成该任务。0表示未完成，1表示完成。
        new_data = {
            "text": data,
            "create_time": now_time,
            "is_finish" : 0,
            "finish" : None,
            "only_sign" : sign
        }
        return new_data;

    def add(self, data):
        """
        添加数据
        :return:None
        """
        self.__collection.insert_one(self.__do_data(data));

    def delete(self,sign):
        """
        删除数据
        :param : 要删除的内容的sign
        :return:None
        """
        self.__collection.delete_one({"only_sign":sign});

    def view_all(self):
        """
        查询全部数据
        :return: all_data
        """
        return self.__collection.find();

    def update(self,data):
        """
        更新一条数据
        :return： None
        """
        self.__collection.update_one({"only_sign":data},{"$set":{"is_finish":1}})

if __name__ == '__main__':
    db = TODO_mongo();
    date = db.view_all();
    print(date)
    for d in date:
        print(d)