tcsenpai/swingmusic
1
0
Fork 0
mirror of https://github.com/tcsenpai/swingmusic.git synced 2025-07-29 14:12:21 +00:00
Code Issues Packages Projects Releases Wiki Activity

import pymongo at db/mongobd/tracks.py

Browse Source
This commit is contained in:
geoffrey45 2022-06-04 11:00:12 +03:00
parent 4302fea0b7
commit 892a090ed5
1 changed files with 10 additions and 28 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

38
server/app/db/mongodb/tracks.py
View File

@ -1,6 +1,7 @@
""" """
This file contains the AllSongs class for interacting with track documents in MongoDB. This file contains the AllSongs class for interacting with track documents in MongoDB.
""" """
import pymongo
from app.db.mongodb import convert_many from app.db.mongodb import convert_many
from app.db.mongodb import convert_one from app.db.mongodb import convert_one
from app.db.mongodb import MongoTracks from app.db.mongodb import MongoTracks
@ -19,12 +20,9 @@ class Tracks(MongoTracks):
""" """
Inserts a new track object into the database. Inserts a new track object into the database.
""" """
return self.collection.update_one({ return self.collection.update_one(
"filepath": song_obj["filepath"] {"filepath": song_obj["filepath"]}, {"$set": song_obj}, upsert=True
}, { ).upserted_id
"$set": song_obj
},
upsert=True).upserted_id
def get_all_tracks(self) -> list: def get_all_tracks(self) -> list:
""" """
@ -50,33 +48,21 @@ class Tracks(MongoTracks):
""" """
Returns all the songs matching the albums in the query params (using regex). Returns all the songs matching the albums in the query params (using regex).
""" """
songs = self.collection.find( songs = self.collection.find({"album": {"$regex": query, "$options": "i"}})
{"album": {
"$regex": query,
"$options": "i"
}})
return convert_many(songs) return convert_many(songs)
def search_songs_by_artist(self, query: str) -> list: def search_songs_by_artist(self, query: str) -> list:
""" """
Returns all the songs matching the artists in the query params. Returns all the songs matching the artists in the query params.
""" """
songs = self.collection.find( songs = self.collection.find({"artists": {"$regex": query, "$options": "i"}})
{"artists": {
"$regex": query,
"$options": "i"
}})
return convert_many(songs) return convert_many(songs)
def find_song_by_title(self, query: str) -> list: def find_song_by_title(self, query: str) -> list:
""" """
Finds all the tracks matching the title in the query params. Finds all the tracks matching the title in the query params.
""" """
song = self.collection.find( song = self.collection.find({"title": {"$regex": query, "$options": "i"}})
{"title": {
"$regex": query,
"$options": "i"
}})
return convert_many(song) return convert_many(song)
def find_songs_by_album(self, name: str, artist: str) -> list: def find_songs_by_album(self, name: str, artist: str) -> list:
@ -90,9 +76,7 @@ class Tracks(MongoTracks):
""" """
Returns a sorted list of all the tracks exactly matching the folder in the query params Returns a sorted list of all the tracks exactly matching the folder in the query params
""" """
songs = self.collection.find({ songs = self.collection.find({"folder": query}).sort("title", pymongo.ASCENDING)
"folder": query
}).sort("title", db.pymongo.ASCENDING)
return convert_many(songs) return convert_many(songs)
def find_songs_by_folder_og(self, query: str) -> list: def find_songs_by_folder_og(self, query: str) -> list:
@ -114,10 +98,8 @@ class Tracks(MongoTracks):
Returns a list of all the tracks containing the albumartist in the query params. Returns a list of all the tracks containing the albumartist in the query params.
""" """
songs = self.collection.find( songs = self.collection.find(
{"albumartist": { {"albumartist": {"$regex": query, "$options": "i"}}
"$regex": query, )
"$options": "i"
}})
return convert_many(songs) return convert_many(songs)
def get_song_by_path(self, path: str) -> dict: def get_song_by_path(self, path: str) -> dict: