Fix(mgmt_api) : add default country canada in companies resource [CPC…

…N-385]
superdesk · Oct 25, 2023 · e1485f5 · e1485f5
1 parent d74cf83
commit e1485f5
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/server/cp/mgmt_api/companies.py b/server/cp/mgmt_api/companies.py
@@ -13,10 +13,16 @@
 
 def init_app(app):
     CompaniesResource.internal_resource = False
-    superdesk.register_resource('companies', CompaniesResource, CPCompaniesService, _app=app)
+    superdesk.register_resource('companies', CPCompaniesResource, CPCompaniesService, _app=app)
     superdesk.register_resource('company_products', CompanyProductsResource, CompanyProductsService, _app=app)
 
 
+class CPCompaniesResource(CompaniesResource):
+    """
+    CP Companies Schema
+    """
+    schema = {**CompaniesResource.schema, "country":{"type": "string", "default" : "CAN"}}
+
 class CPCompaniesService(CompaniesService):
     def on_create(self, docs):
         super().on_create(docs)