Merge pull request #261 from Haidra-Org/main

fix: set_event_loop_policy for win32
Haidra-Org · Sep 23, 2024 · ac38514 · ac38514
2 parents d3efead + 8efc611
commit ac38514
Show file tree

Hide file tree

Showing 5 changed files with 18 additions and 0 deletions.
diff --git a/examples/ai_horde_client/image/async_manual_client_example.py b/examples/ai_horde_client/image/async_manual_client_example.py
@@ -1,5 +1,6 @@
 import argparse
 import asyncio
+import sys
 import time
 from pathlib import Path
 
@@ -119,4 +120,7 @@ async def main(apikey: str = ANON_API_KEY) -> None:
     )
     args = parser.parse_args()
 
+    if sys.platform == "win32":
+        asyncio.set_event_loop_policy(asyncio.WindowsSelectorEventLoopPolicy())
+
     asyncio.run(main(args.apikey))
diff --git a/examples/ai_horde_client/image/async_simple_client_example.py b/examples/ai_horde_client/image/async_simple_client_example.py
@@ -1,5 +1,6 @@
 import argparse
 import asyncio
+import sys
 from pathlib import Path
 
 import aiohttp
@@ -161,5 +162,8 @@ async def async_simple_generate_example(apikey: str = ANON_API_KEY) -> None:
     )
     args = parser.parse_args()
 
+    if sys.platform == "win32":
+        asyncio.set_event_loop_policy(asyncio.WindowsSelectorEventLoopPolicy())
+
     # Run the example.
     asyncio.run(async_simple_generate_example(args.apikey))
diff --git a/examples/ai_horde_client/text/async_text_generate.py b/examples/ai_horde_client/text/async_text_generate.py
@@ -1,5 +1,6 @@
 import argparse
 import asyncio
+import sys
 from pathlib import Path
 
 import aiofiles
@@ -115,4 +116,7 @@ async def main(apikey: str) -> None:
     )
     args = parser.parse_args()
 
+    if sys.platform == "win32":
+        asyncio.set_event_loop_policy(asyncio.WindowsSelectorEventLoopPolicy())
+
     asyncio.run(main(args.api_key))
diff --git a/horde_sdk/__init__.py b/horde_sdk/__init__.py
@@ -60,6 +60,7 @@ def _dev_env_var_warnings() -> None:  # pragma: no cover
 
 _dev_env_var_warnings()
 
+
 from horde_sdk.consts import (
     PAYLOAD_HTTP_METHODS,
     HTTPMethod,

diff --git a/tests/conftest.py b/tests/conftest.py
@@ -1,12 +1,17 @@
+import asyncio
 import base64
 import os
 import pathlib
+import sys
 
 import pytest
 from loguru import logger
 
 os.environ["TESTS_ONGOING"] = "1"
 
+if sys.platform == "win32":
+    asyncio.set_event_loop_policy(asyncio.WindowsSelectorEventLoopPolicy())
+
 from horde_sdk.ai_horde_api.apimodels import ImageGenerateAsyncRequest, ImageGenerationInputPayload
 from horde_sdk.generic_api.consts import ANON_API_KEY