From db207f4995cf4b1c7f5f34154cfc84e0e4ebd369 Mon Sep 17 00:00:00 2001
From: laurentbarontini <l.barontini@open-squared.ch>
Date: Mon, 26 Jan 2026 22:05:58 +0100
Subject: [PATCH] 26.01.26

---
 modules/automation/automation.py |  2 +-
 modules/purchase_trade/party.py  | 14 ++++++++++----
 2 files changed, 11 insertions(+), 5 deletions(-)

diff --git a/modules/automation/automation.py b/modules/automation/automation.py
index d9c35a0..47ec9dd 100644
--- a/modules/automation/automation.py
+++ b/modules/automation/automation.py
@@ -270,7 +270,7 @@ class AutomationDocument(ModelSQL, ModelView, Workflow):
                             else:
                                 sale = Sale()
                                 sale_line = SaleLine()
-                                sale.party = Party.getPartyByName(customer)
+                                sale.party = Party.getPartyByName(customer,'CLIENT')
                                 logger.info("SALE_PARTY:%s",sale.party)
                                 sale.reference = reference
                                 sale.company = 6
diff --git a/modules/purchase_trade/party.py b/modules/purchase_trade/party.py
index fde79b8..d612244 100755
--- a/modules/purchase_trade/party.py
+++ b/modules/purchase_trade/party.py
@@ -1,12 +1,9 @@
 from trytond.model import ModelSQL, ModelView, fields
-from trytond.pool import PoolMeta
+from trytond.pool import PoolMeta, Pool
 from trytond.exceptions import UserError
 from trytond.modules.purchase_trade.purchase import (TRIGGERS)
 from trytond.transaction import Transaction
 
-__all__ = ['Party']
-__metaclass__ = PoolMeta
-
 class PartyExecution(ModelSQL,ModelView):
     "Party Execution"
     __name__ = 'party.execution'
@@ -34,5 +31,14 @@ class Party(metaclass=PoolMeta):
             p = cls()
             p.name = party
             cls.save([p])
+            if category:
+                Category = Pool().get('party.category')
+                cat = Category.search(['name','=',category])
+                if cat:
+                    PartyCategory = Pool().get('party.party-party.category')
+                    pc = PartyCategory()
+                    pc.party = p.id
+                    pc.category = cat[0].id
+                    PartyCategory.save([pc])
             return p
     
\ No newline at end of file