CHANGE: Adding quotes now return their IDs

utils/commands.py

@@ -29,13 +29,13 @@ async def quote(bot: object, *quote: str) -> str:
 
     try:
         user = bot.author.name
-        set_quote(user, quote)
+        qtid = set_quote(user, quote)
     except Exception as ex:
         if ex.args[0].find("Duplicate") != -1:
             return await bot.send("There's already a quote from that same person, with that exact match!")
         return await bot.send(f'{ex.args}\n_What the fuck are you doing?_')
     else:
-        return await bot.send('Done:\n`%s`' % quote)
+        return await bot.send("Done: `" + quote + "` ID: `" + str(qtid) + "`")
 
 
 @client.command(aliases=['rq'])

utils/database.py

@@ -58,16 +58,20 @@ def migrate() -> None:
         logger.error(ex.args)
 
 
-def set_quote(user: str, quote: str) -> None:
+def set_quote(user: str, quote: str) -> int:
     """
     Set a quote into database.
     """
+
+    qt = Quotes(quote=quote, user=user,)
+    qtid = 0
     with Session(SQLACHEMY) as session:
-        session.add(Quotes(
-            quote=quote,
-            user=user,
-        ))
+        session.add(qt)
+        session.flush()
+        session.refresh(qt)
+        qtid = qt.id
         session.commit()
+    return qtid
 
 
 def get_quotes(ids: list) -> tuple: