import other kontor repos into directories
This commit is contained in:
@@ -0,0 +1,53 @@
|
||||
# -*- coding: utf-8 -*-
|
||||
from bson.objectid import ObjectId
|
||||
import sys
|
||||
|
||||
class ComicDAO:
|
||||
|
||||
# constructor for the class
|
||||
def __init__(self, database):
|
||||
self.db = database
|
||||
self.comics = database.comics
|
||||
|
||||
def get_comics(self):
|
||||
cursor = self.comics.find()
|
||||
l = []
|
||||
for comic in cursor:
|
||||
l.append({'title':comic['title'],
|
||||
'_id':comic['_id'],
|
||||
'current_order':comic['current_order'],
|
||||
'completed':comic['completed']})
|
||||
return l
|
||||
|
||||
def get_comic(self, comic_id):
|
||||
comic = self.comics.find_one({"_id": ObjectId(comic_id)})
|
||||
return comic
|
||||
|
||||
def insert_entry(self, comic_title, comic_publisher=None, comic_order=False, comic_completed=False):
|
||||
print "inserting comic entry", comic_title
|
||||
|
||||
comic = {"title": comic_title, "current_order": comic_order, "completed": comic_completed}
|
||||
|
||||
# now insert the comic
|
||||
try:
|
||||
result = self.comics.insert_one(comic)
|
||||
print "Modified comic: ", result.modified_count
|
||||
except:
|
||||
print "Error inserting comic"
|
||||
print "Unexpected error:", sys.exc_info()
|
||||
|
||||
def update_entry(self, comic_id, comic_title, comic_publisher=None, comic_order=False, comic_completed=False):
|
||||
print "upserting comic entry", comic_title
|
||||
|
||||
filter_doc = {"_id": ObjectId(comic_id)}
|
||||
comic = { "$set": {"title": comic_title, 'current_order': comic_order, 'completed': comic_completed}}
|
||||
|
||||
# now insert the post
|
||||
try:
|
||||
result = self.comics.update_one(filter_doc, comic)
|
||||
print "Matching comic: ", result.matched_count
|
||||
print "Modified comic: ", result.modified_count
|
||||
except:
|
||||
print "Error inserting comic"
|
||||
print "Unexpected error:", sys.exc_info()
|
||||
|
||||
Reference in New Issue
Block a user