Nuevos cambios para config del repo y creacion del servicio de creacion comercial

Back_comercial_iko/services/Comercial_service.py

@@ -1,7 +1,11 @@
 from Back_comercial_iko.database.Database import Database
 from Back_comercial_iko.models.Societe_model import SocieteModel
+from Back_comercial_iko.models.Projet_model import ProjetModel
+from Back_comercial_iko.models.Projet_extrafields_model import ProjetExtrafieldsModel
 from Back_comercial_iko.core.validators.Field_validator import FieldValidator
-from Back_comercial_iko.core.SocieteNotFoundException import SocieteNotFoundException
+from Back_comercial_iko.core.MessageException import MessageException
+from datetime import datetime
+
 
 class ComercialService:
 
@@ -9,18 +13,72 @@ class ComercialService:
         self.conn = Database()
         self.db = self.conn.setConnection()
 
-    def create_project_commercial(self, data: dict, token_data: dict):
+   
+    def new_ref(self):
+        ALIAS_REF = "PCom"
+        year = datetime.now().strftime("%y")
+        
+        result = self.db.query(ProjetModel.REF).filter(
+            ProjetModel.REF.like(f"%{ALIAS_REF}0%")
+        ).order_by(ProjetModel.REF.desc()).first()
+        
+        if result and result.REF:
+            ref_str = result.REF
+            start = ref_str.find(ALIAS_REF) + len(ALIAS_REF)
+            numero_str = ref_str[start:start + 4]
+            new_ref_num = int(numero_str) + 1 if numero_str.isdigit() else 1
+        else:
+            new_ref_num = 1
+        
+        return f"{ALIAS_REF}{str(new_ref_num).zfill(4)}-{year}"
 
+
+    def create_project_commercial(self, data: dict, token_data: dict):
+        """Service to create a new commercial project"""
         data_session_login = token_data['data']
         FieldValidator.required(data.get("ex_societe"), "ex_societe")
-        FieldValidator.required(data.get("nomUser"), "nomUser")
-        nombre_soc = data.get("ex_societe")
-        result_soc = (self.db.query(SocieteModel.ROWID, SocieteModel.NOM).filter(SocieteModel.NOM == nombre_soc).first())
-        if not result_soc:
-            raise SocieteNotFoundException()
+        FieldValidator.required(data.get("description"), "description")
+        FieldValidator.required(data.get("date_o"), "date_o")
+        FieldValidator.required(data.get("date_e"), "date_e")
+        name_soc = data.get("ex_societe")
+        date_c = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
+        result_soc = (self.db.query(SocieteModel.ROWID, SocieteModel.NOM).filter(SocieteModel.NOM == name_soc).first())
+        fk_soc = result_soc.ROWID if result_soc else None
+        id_user = data_session_login['id']
+        description = data.get("description")
+        dateo = data.get("date_o")
+        datee = data.get("date_e")
+        last_contact = data.get("last_contact"),
+        agency = data.get("agency"),
+        advertiser = data.get("advertiser"),
+        status = data.get("status")
+
+        if fk_soc is None:
+            new_societe = SocieteModel(NOM=name_soc, NAME_ALIAS=name_soc, ENTITY=1, CLIENT=1, FOURNISSEUR=0, DATEC=date_c)
+            self.db.add(new_societe)
+            self.db.flush()
+            self.db.commit()
+            fk_soc = new_societe.ROWID
+
+        new_project = ProjetModel(TMS=date_c, DESCRIPTION=description, TITLE=description, DATEO=dateo, DATEE=datee, FK_STATUT=1,
+        FK_USER_CREAT=id_user, REF=self.new_ref(), DATEC=date_c, FK_SOC=fk_soc)
+        self.db.add(new_project)
+        self.db.flush()
+        self.db.commit()
+        rowid_project = new_project.ROWID if new_project.ROWID else MessageException.raise_it(extra_msg="No se pudo crea el proyecto")
+
+        if rowid_project:
+                extrafields_project = ProjetExtrafieldsModel(FK_OBJECT=rowid_project, TIPO=2, LAST_CONTACT=last_contact,
+                AGENCIA=agency, ANUNCIANTE=advertiser, ESTADO=status)
+                self.db.add(extrafields_project)
+                self.db.flush()
+                self.db.commit()
+                rowid_extrafields = extrafields_project.ROWID if extrafields_project.ROWID else MessageException.raise_it(extra_msg="No se pudo crear el proyecto")
 
         return {
-            "fk_soc": result_soc.ROWID,
-            "nombre_soc": nombre_soc
+            "fk_soc": fk_soc,
+            "nombre_soc": name_soc,
+            "rowid_project": rowid_project,
+            "rowid_extrafields": rowid_extrafields,
         }