malojaserver 3.2.2__py3-none-any.whl → 3.2.3__py3-none-any.whl

maloja/database/exceptions.py

@@ -1,37 +1,57 @@
 from bottle import HTTPError
 
+
 class EntityExists(Exception):
-	def __init__(self,entitydict):
+	def __init__(self, entitydict):
 		self.entitydict = entitydict
 
 
 class TrackExists(EntityExists):
 	pass
 
+
 class ArtistExists(EntityExists):
 	pass
 
+
 class AlbumExists(EntityExists):
 	pass
 
+
+# if the scrobbles dont match
+class DuplicateTimestamp(Exception):
+	def __init__(self, existing_scrobble, rejected_scrobble):
+		self.existing_scrobble = existing_scrobble
+		self.rejected_scrobble = rejected_scrobble
+
+
+# if it's the same scrobble
+class DuplicateScrobble(Exception):
+	def __init__(self, scrobble):
+		self.scrobble = scrobble
+
+
 class DatabaseNotBuilt(HTTPError):
 	def __init__(self):
 		super().__init__(
 			status=503,
 			body="The Maloja Database is being upgraded to support new Maloja features. This could take a while.",
-			headers={"Retry-After":120}
+			headers={"Retry-After": 120}
 		)
 
 
 class MissingScrobbleParameters(Exception):
-	def __init__(self,params=[]):
+	def __init__(self, params=[]):
 		self.params = params
 
+
 class MissingEntityParameter(Exception):
 	pass
 
+
 class EntityDoesNotExist(HTTPError):
 	entitytype = 'Entity'
+
 	def __init__(self,entitydict):
 		self.entitydict = entitydict
 		super().__init__(
@@ -39,9 +59,14 @@ class EntityDoesNotExist(HTTPError):
 			body=f"The {self.entitytype} '{self.entitydict}' does not exist in the database."
 		)
 
+
 class ArtistDoesNotExist(EntityDoesNotExist):
 	entitytype = 'Artist'
+
+
 class AlbumDoesNotExist(EntityDoesNotExist):
 	entitytype = 'Album'
+
+
 class TrackDoesNotExist(EntityDoesNotExist):
 	entitytype = 'Track'