From 866538f7872fa3bfbd1fa171f1a18525e3f0d5af Mon Sep 17 00:00:00 2001
From: wilsongomez <gomezwilson.99@gmail.com>
Date: Mon, 4 Apr 2022 17:49:31 -0500
Subject: [PATCH] minor fix

---
 sale.py | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/sale.py b/sale.py
index 1eec04e..e23b861 100644
--- a/sale.py
+++ b/sale.py
@@ -221,6 +221,11 @@ class Sale(metaclass=PoolMeta):
         try:
             sale, = cls.create([to_create])
             sale.set_number([sale])
+            if sale.table_assigned:
+                ShopTable = Pool().get('sale.shop.table')
+                tables = ShopTable.search([('id', '=', sale.table_assigned.id)])
+                if tables:
+                    ShopTable.write(tables, {'sale': sale.id, 'state':'occupied'})
             return sale.id
         except Exception as e:
             print(e, 'error')