SMVaria: Allow raise_for_status to be false for debugging purposes

tcprescott · Sep 15, 2022 · c7f09c7 · c7f09c7
1 parent 4afc9f6
commit c7f09c7
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/pyz3r/smvaria.py b/pyz3r/smvaria.py
@@ -130,7 +130,7 @@ def __init__(
         self.auth = aiohttp.BasicAuth(login=username, password=password) if username and password else None
         self.settings_dict = settings_dict
 
-    async def generate_game(self):
+    async def generate_game(self, raise_for_status=True):
         try:
             async for attempt in AsyncRetrying(
                     stop=stop_after_attempt(3),
@@ -141,7 +141,7 @@ async def generate_game(self):
                             url=f'{self.baseurl}/randomizerWebService',
                             data=self.settings,
                             auth=self.auth,
-                            raise_for_status=True) as resp:
+                            raise_for_status=raise_for_status) as resp:
                         req = await resp.json(content_type='text/html')
                     return req
         except RetryError as e:
@@ -157,6 +157,7 @@ async def create(
         username=None,
         password=None,
         settings_dict=None,
+        raise_for_status=True,
     ):
         seed = cls(
             skills_preset=skills_preset,
@@ -170,7 +171,7 @@ async def create(
 
         seed.settings = await seed.get_settings()
 
-        seed.data = await seed.generate_game()
+        seed.data = await seed.generate_game(raise_for_status)
         seed.guid = uuid.UUID(hex=seed.data['seedKey'])
 
         return seed