fix event loop for windows (#67)

* fix event loop for windows

Author: jssmith

src/postgres_mcp/__init__.py

@@ -1,11 +1,18 @@
 import asyncio
+import sys
 
 from . import server
 from . import top_queries
 
 
 def main():
     """Main entry point for the package."""
+    # As of version 3.3.0 Psycopg on Windows is not compatible with the default
+    # ProactorEventLoop.
+    # See: https://www.psycopg.org/psycopg3/docs/advanced/async.html#async
+    if sys.platform == "win32":
+        asyncio.set_event_loop_policy(asyncio.WindowsSelectorEventLoopPolicy())
+
     asyncio.run(server.main())
 
 

src/postgres_mcp/server.py

@@ -618,7 +618,3 @@ async def shutdown(sig=None):
 
     # Exit with appropriate status code
     sys.exit(128 + sig if sig is not None else 0)
-
-
-if __name__ == "__main__":
-    asyncio.run(main())