Use predefined wbsockets enum for exception code comparison

Author: steventux

src/relay_listener.py

@@ -16,6 +16,7 @@
 
 from dotenv import load_dotenv
 from websockets.asyncio.client import connect
+from websockets.frames import CloseCode
 from websockets.exceptions import ConnectionClosedError
 
 from services.mwl.create_worklist_item import CreateWorklistItem
@@ -139,7 +140,7 @@ async def main():
             logger.warning("\nShutting down...")
             break
         except ConnectionClosedError as e:
-            if "ExpiredToken" in str(e) and e.code == 1011:
+            if "ExpiredToken" in str(e) and e.code == CloseCode.INTERNAL_ERROR.value:
                 logger.info("SAS token expired, refreshing...")
             else:
                 logger.warning(f"Connection closed with code {e.code}: {e.reason}")