def add_notice(self, data):
notice = {
"content": data["content"],
"publisher": ModelBase.get_oid(data["publisher"]),
"create_date": datetime.utcnow(),
"invalid_date": datetime.utcnow() + timedelta(days = int(data["life"]))
}
return str(self.collection.insert(notice))
def add(self, user_id, song_id):
fav = self.collection.insert({
"user_id": ModelBase.get_oid(user_id),
"song_id": song_id
})
return str(fav)
def remove(self, user_id, record_id):
record = self.collection.find_one(ModelBase.get_oid(record_id))
if str(record["user_id"]) == user_id:
self.collection.remove(record)
return True
return False
def get_all(self, user_id):
favorites = self.collection.find({"user_id": ModelBase.get_oid(user_id)}, {"song_id": 1})
return ModelBase.cursor2list(favorites)
def get_one(self, notice_id): notice = self.collection.find_one(ModelBase.get_oid(notice_id)) notice = self.to_user(notice) return ModelBase.transform_id(notice)
def get_one(self, user_id): result = self.collection.find_one(ModelBase.get_oid(user_id)) return ModelBase.transform_id(result)
