import other kontor repos into directories
This commit is contained in:
@@ -0,0 +1,50 @@
|
||||
# -*- coding: utf-8 -*-
|
||||
from bson.objectid import ObjectId
|
||||
import sys
|
||||
|
||||
class ArtistDAO:
|
||||
|
||||
# constructor for the class
|
||||
def __init__(self, database):
|
||||
self.db = database
|
||||
self.artists = database.artists
|
||||
|
||||
def get_artists(self):
|
||||
cursor = self.artists.find()
|
||||
l = []
|
||||
for artist in cursor:
|
||||
l.append({'name':artist['name'], '_id':artist['_id']})
|
||||
return l
|
||||
|
||||
def get_artist(self, artist_id):
|
||||
artist = self.artists.find_one({"_id": ObjectId(artist_id)})
|
||||
return artist
|
||||
|
||||
def insert_entry(self, artist_name):
|
||||
print "inserting artist entry", artist_name
|
||||
|
||||
artist = {"name": artist_name}
|
||||
|
||||
# now insert the post
|
||||
try:
|
||||
result = self.artists.insert_one(artist)
|
||||
print "Matching artist: ", result.matched_count
|
||||
print "Modified artist: ", result.modified_count
|
||||
except:
|
||||
print "Error inserting artist"
|
||||
print "Unexpected error:", sys.exc_info()
|
||||
|
||||
def update_entry(self, artist_id, artist_name):
|
||||
print "upserting artist entry", artist_name
|
||||
|
||||
filter_doc = {"_id": ObjectId(artist_id)}
|
||||
artist = { "$set": {"name": artist_name}}
|
||||
|
||||
# now insert the post
|
||||
try:
|
||||
result = self.artists.update_one(filter_doc, artist)
|
||||
print "Modified artist: ", result.modified_count
|
||||
except:
|
||||
print "Error inserting artist"
|
||||
print "Unexpected error:", sys.exc_info()
|
||||
|
||||
Reference in New Issue
Block a user