From 1429f1259c93ba55bbda0675baa13c386dc5b2ea Mon Sep 17 00:00:00 2001
From: Tristan Gosselin-Hane <starcraft66@gmail.com>
Date: Mon, 15 Oct 2018 23:31:18 -0400
Subject: [PATCH] Fix bug where the database session sometimes wasn't closed

---
 auth_server.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/auth_server.py b/auth_server.py
index ee1d93a..163c185 100644
--- a/auth_server.py
+++ b/auth_server.py
@@ -38,10 +38,12 @@ class AuthProtocol(ServerProtocol):
             token = session.query(AccountLinkToken).filter_by(token=connection_token).first()
             if not token:
                 self.close("You have connected with an invalid token!")
+                session.close()
                 return
             discord_account = session.query(DiscordAccount).filter_by(link_token_id=token.id).first()
             if not discord_account:
                 self.close("You have connected with an invalid token!")
+                session.close()
                 return
             if datetime.utcnow() < token.expiry:
                 # Check if they already have a linked account and are re-linking
@@ -66,6 +68,7 @@ class AuthProtocol(ServerProtocol):
 
         except Exception as e:
             self.logger.error(e)
+            session.close()
 
 
         # Kick the player.