133 lines
4.6 KiB
Python
133 lines
4.6 KiB
Python
import re
|
|
|
|
with open('expenses/models.py', 'r', encoding='utf-8') as f:
|
|
lines = f.readlines()
|
|
|
|
new_lines = []
|
|
in_project = False
|
|
in_category = False
|
|
in_vendor = False
|
|
in_expense = False
|
|
in_projectphase = False
|
|
|
|
for line in lines:
|
|
if line.startswith('class Project(models.Model):'):
|
|
in_project = True
|
|
abstracts = """from django.core.exceptions import ValidationError
|
|
|
|
class TimeStampedModel(models.Model):
|
|
created_at = models.DateTimeField(auto_now_add=True, verbose_name="Fecha de Creación")
|
|
updated_at = models.DateTimeField(auto_now=True, verbose_name="Última Modificación")
|
|
|
|
class Meta:
|
|
abstract = True
|
|
|
|
class CatalogModel(TimeStampedModel):
|
|
code = models.CharField(max_length=50, unique=True, verbose_name="Código")
|
|
name = models.CharField(max_length=200, verbose_name="Nombre")
|
|
description = models.TextField(blank=True, verbose_name="Descripción")
|
|
|
|
class Meta:
|
|
abstract = True
|
|
|
|
"""
|
|
new_lines.append(abstracts)
|
|
new_lines.append('class Project(CatalogModel):\n')
|
|
continue
|
|
|
|
if line.startswith('class Category(models.Model):'):
|
|
in_project = False
|
|
in_category = True
|
|
new_lines.append('class Category(CatalogModel):\n')
|
|
continue
|
|
|
|
if line.startswith('class Vendor(models.Model):'):
|
|
in_category = False
|
|
in_vendor = True
|
|
new_lines.append('class Vendor(TimeStampedModel):\n')
|
|
continue
|
|
|
|
if line.startswith('class ProjectPhase(models.Model):'):
|
|
in_projectphase = True
|
|
new_lines.append('class ProjectPhase(TimeStampedModel):\n')
|
|
continue
|
|
|
|
if line.startswith('class ExpenseReport(models.Model):'):
|
|
in_vendor = False
|
|
in_projectphase = False
|
|
|
|
if line.startswith('class Expense(models.Model):'):
|
|
in_expense = True
|
|
new_lines.append('class Expense(TimeStampedModel):\n')
|
|
continue
|
|
|
|
if line.startswith('class VehicleLog(models.Model):'):
|
|
in_expense = False
|
|
|
|
if in_project:
|
|
if 'code =' in line and 'default=\'PROY-000\'' in line: continue
|
|
if 'name = models.CharField(max_length=200' in line: continue
|
|
if 'description = models.TextField(blank=True' in line: continue
|
|
if 'created_at = models.DateTimeField(auto_now_add=True' in line: continue
|
|
if 'updated_at = models.DateTimeField(auto_now=True' in line: continue
|
|
|
|
if in_category:
|
|
if 'code = models.CharField(max_length=10' in line: continue
|
|
if 'name = models.CharField(max_length=100' in line: continue
|
|
if 'description = models.TextField(blank=True' in line: continue
|
|
|
|
if in_vendor:
|
|
if 'created_at = models.DateTimeField(auto_now_add=True)' in line: continue
|
|
if 'updated_at = models.DateTimeField(auto_now=True)' in line: continue
|
|
|
|
if in_projectphase:
|
|
if 'created_at = models.DateTimeField(auto_now_add=True)' in line: continue
|
|
if 'updated_at = models.DateTimeField(auto_now=True)' in line: continue
|
|
|
|
if in_expense:
|
|
if 'created_at = models.DateTimeField(auto_now_add=True)' in line: continue
|
|
if 'updated_at = models.DateTimeField(auto_now=True)' in line: continue
|
|
if ' def save(self, *args, **kwargs):' in line:
|
|
encap = """
|
|
def clean(self):
|
|
super().clean()
|
|
if self.amount_clp and self.amount_clp < 0:
|
|
raise ValidationError({'amount_clp': 'El monto neto no puede ser negativo.'})
|
|
|
|
amount = Decimal(str(self.amount_clp or '0'))
|
|
tax = Decimal(str(self.tax_amount_clp or '0'))
|
|
expected = amount + tax
|
|
|
|
if self.total_amount_clp and abs(self.total_amount_clp - expected) > Decimal('1.0'):
|
|
raise ValidationError({'total_amount_clp': f'Discrepancia. Total esperado: {expected}'})
|
|
|
|
def approve_expense(self, user):
|
|
if self.approval_status == ApprovalStatusEnum.EN_REVISION:
|
|
self.approval_status = ApprovalStatusEnum.SOLICITADO_FINANZAS
|
|
self.elevated_by = user
|
|
self.save()
|
|
|
|
def reject_expense(self, reason):
|
|
self.approval_status = ApprovalStatusEnum.OBSERVADO_RECHAZADO
|
|
self.rejection_reason = reason
|
|
self.save()
|
|
"""
|
|
new_lines.append(encap + "\n")
|
|
|
|
new_lines.append(line)
|
|
|
|
proxy = """
|
|
class Reimbursement(Expense):
|
|
class Meta:
|
|
proxy = True
|
|
verbose_name = "Rendición Especializada"
|
|
verbose_name_plural = "Rendiciones Especializadas"
|
|
|
|
def get_approval_flow(self):
|
|
return "Flujo Acelerado de Reembolso"
|
|
"""
|
|
new_lines.append(proxy)
|
|
|
|
with open('expenses/models.py', 'w', encoding='utf-8') as f:
|
|
f.writelines(new_lines)
|