storage.py

__author__ = 'couldtt'

import pymongo
import time


class MongoStorage():
    def __init__(self):
        try:
            client = pymongo.MongoClient('localhost', 27017)
            self.db = client.limitfree
            self.collection = self.db.books
        except:
            print("MongoDB数据库链接失败")

        localtime = time.localtime(time.time())
        self.now_time = time.strftime('%Y%m%d%H', localtime)

    def save(self, type, platform):
        today = {
            'time': self.now_time,
            'type': type,
            'platform': platform
        }
        save_id = self.collection.save(today)
        if save_id:
            return True

    def get_today(self, type):
        return self.collection.find_one({'time': self.now_time, 'type': type})