Merge pull request #49 from usermicrodevices/develop

add on_doc_post_save

Author: usermicrodevices

api/views.py

@@ -248,6 +248,8 @@ def post(self, request, *args, **kwargs):
         if records and sum_final and registered_at:
             id_owner = data.get('owner', request.user.default_company.id if request.user.default_company else 1)
             default_contractor = Company.objects.filter(extinfo__default_cash_contractor=True).first()
+            if not default_contractor and request.user.companies.count():
+                default_contractor = request.user.companies.first()
             id_contractor = data.get('contractor', default_contractor.id if default_contractor else 2)
             dtype = data.get('type', 'sale')
             doc_type, created = DocType.objects.get_or_create(alias=dtype, defaults={'alias':dtype, 'name':dtype.title()})

core/models.py

@@ -123,6 +123,29 @@ def on_doc_pre_save(sender, **kwargs):
             if value:
                 instance.sum_final = value
 
+@receiver(post_save, sender=Doc)
+def on_doc_post_save(sender, **kwargs):
+    instance: Doc = kwargs['instance']
+    if not kwargs.get('created', False) and not instance.extinfo.get('not_use_post_save_auto_register', False):
+        recs = get_model('core.Record').objects.filter(doc=instance)
+        recs_count = recs.count()
+        regs = get_model('core.Register').objects.filter(rec__in=recs.values_list('id', flat=True))
+        if instance.type.auto_register:
+            if recs_count and recs_count != regs.count():
+                regs_new = []
+                for r in recs:
+                    regs_new.append(get_model('core.Register')(rec=r))
+                try:
+                    rgs = get_model('core.Register').objects.bulk_create(regs_new)
+                except Exception as e:
+                    instance.loge(e)
+        elif recs_count:
+            if regs.count():
+                try:
+                    regs.delete()
+                except Exception as e:
+                    instance.loge(e)
+
 
 class Record(CustomAbstractModel):
     count = models.DecimalField(max_digits=15, decimal_places=3, default=0, null=False, blank=False, verbose_name=_('count'), help_text=_('count of products'))