support for .yaml policy file

permitio · Oct 29, 2024 · f19b265 · f19b265
1 parent c450108
commit f19b265
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/packages/opal-client/opal_client/policy/updater.py b/packages/opal-client/opal_client/policy/updater.py
@@ -1,6 +1,7 @@
 import asyncio
 from typing import List, Optional
 import json
+import yaml
 import pydantic
 from fastapi_websocket_pubsub import PubSubClient
 from fastapi_websocket_rpc.rpc_channel import RpcChannel
@@ -343,6 +344,12 @@ async def update_policy(
                                 await store_transaction.set_policy(policy_module.path, json.dumps(policy_data))
                             except json.JSONDecodeError:
                                 logger.error(f"Invalid JSON in OpenFGA policy file: {policy_module.path}")
+
+                        elif policy_module.path.endswith('.yaml'):
+                            policy_data = yaml.safe_load(policy_module.rego)
+                            await store_transaction.set_policy(policy_module.path, json.dumps(policy_data))
+                            #logger.error(f"yaml file")
+
                         else:
                             logger.warning(f"Skipping non-JSON file for OpenFGA: {policy_module.path}")
                 else: