gcivil-nyu-org / team4-wed-fall25 / 55

Committed 03 Nov 2025 03:34PM UTC coverage: 56.582% (+7.8%) from 48.812%

Build # 55

Build Type

push

travis-pro

Committed by

web-flow

Commit Message

Syncing the develop and main branch

Syncing Prod and Dev

Run Details

201 of 388 new or added lines in 11 files covered. (51.8%)

5 existing lines in 1 file now uncovered.

447 of 790 relevant lines covered (56.58%)

0.57 hits per line

Source File
Press 'n' to go to next uncovered line, 'b' for previous

92.45

/note2webapp/models.py

from django.db import models
from django.contrib.auth.models import User
from django.db.models.signals import post_save
from django.dispatch import receiver


# -----------------------
# USER PROFILE WITH ROLE
# -----------------------
class Profile(models.Model):
    ROLE_CHOICES = [
        ("uploader", "Model Uploader"),
        ("reviewer", "Model Reviewer"),
    ]
    user = models.OneToOneField(User, on_delete=models.CASCADE)
    role = models.CharField(max_length=20, choices=ROLE_CHOICES, default="uploader")

    def __str__(self):
        return f"{self.user.username} ({self.role})"


@receiver(post_save, sender=User)
def create_or_update_profile(sender, instance, created, **kwargs):
    """Automatically create or update Profile when User is created/updated"""
    if created:
        Profile.objects.create(user=instance)
    else:
        instance.profile.save()


# -----------------------
# MODEL UPLOAD + VERSION
# -----------------------
class ModelUpload(models.Model):
    user = models.ForeignKey(User, on_delete=models.CASCADE, related_name="uploads")
    name = models.CharField(max_length=200)
    created_at = models.DateTimeField(auto_now_add=True)

    def __str__(self):
        return self.name


class ModelVersion(models.Model):
    upload = models.ForeignKey(
        ModelUpload, on_delete=models.CASCADE, related_name="versions"
    )
    model_file = models.FileField(upload_to="models/")
    predict_file = models.FileField(upload_to="predict/")
    schema_file = models.FileField(
        upload_to="schemas/", blank=True, null=True
    )  # for test data generation
    tag = models.CharField(max_length=100)
    # ADD THIS: Model information field
    information = models.TextField(blank=True, null=True)

    category = models.CharField(
        max_length=50,
        choices=[
            ("research", "Research"),
            ("production", "Production"),
            ("testing", "Testing"),
        ],
        default="research",
    )

    status = models.CharField(
        max_length=20,
        choices=[("PENDING", "Pending"), ("PASS", "Pass"), ("FAIL", "Fail")],
        default="PENDING",
    )
    is_active = models.BooleanField(default=False)
    log = models.TextField(blank=True, null=True)
    created_at = models.DateTimeField(auto_now_add=True)

    # Soft delete fields
    is_deleted = models.BooleanField(default=False)
    deleted_at = models.DateTimeField(null=True, blank=True)
    # Store version number permanently
    version_number = models.IntegerField(default=1)

    class Meta:
        ordering = ["-created_at"]

    def save(self, *args, **kwargs):
        # Auto-generate version number for new versions only
        if not self.pk:  # Only for new objects
            from django.db.models import Max

            # Get the highest version number for this upload
            last_version = ModelVersion.objects.filter(upload=self.upload).aggregate(
                Max("version_number")
            )["version_number__max"]
            self.version_number = (last_version or 0) + 1
        super().save(*args, **kwargs)

    def get_version_number(self):
        """Keep this method for backwards compatibility"""
        return self.version_number

    def __str__(self):
        try:
            version_str = f"v{self.version_number}"
            upload_name = getattr(self.upload, "name", "Unknown Upload")
            return f"{upload_name} - {version_str} ({self.status})"
        except Exception:
            return f"ModelVersion (unsaved) - {self.status}"

1	from django.db import models	1✔
2	from django.contrib.auth.models import User	1✔
3	from django.db.models.signals import post_save	1✔
4	from django.dispatch import receiver	1✔
5
6
7	# -----------------------
8	# USER PROFILE WITH ROLE
9	# -----------------------
10	class Profile(models.Model):	1✔
11	ROLE_CHOICES = [	1✔
12	("uploader", "Model Uploader"),
13	("reviewer", "Model Reviewer"),
14	]
15	user = models.OneToOneField(User, on_delete=models.CASCADE)	1✔
16	role = models.CharField(max_length=20, choices=ROLE_CHOICES, default="uploader")	1✔
17
18	def __str__(self):	1✔
19	return f"{self.user.username} ({self.role})"	×
20
21
22	@receiver(post_save, sender=User)	1✔
23	def create_or_update_profile(sender, instance, created, **kwargs):	1✔
24	"""Automatically create or update Profile when User is created/updated"""
25	if created:	1✔
26	Profile.objects.create(user=instance)	1✔
27	else:
28	instance.profile.save()	1✔
29
30
31	# -----------------------
32	# MODEL UPLOAD + VERSION
33	# -----------------------
34	class ModelUpload(models.Model):	1✔
35	user = models.ForeignKey(User, on_delete=models.CASCADE, related_name="uploads")	1✔
36	name = models.CharField(max_length=200)	1✔
37	created_at = models.DateTimeField(auto_now_add=True)	1✔
38
39	def __str__(self):	1✔
40	return self.name	1✔
41
42
43	class ModelVersion(models.Model):	1✔
44	upload = models.ForeignKey(	1✔
45	ModelUpload, on_delete=models.CASCADE, related_name="versions"
46	)
47	model_file = models.FileField(upload_to="models/")	1✔
48	predict_file = models.FileField(upload_to="predict/")	1✔
49	schema_file = models.FileField(	1✔
50	upload_to="schemas/", blank=True, null=True
51	) # for test data generation
52	tag = models.CharField(max_length=100)	1✔
53	# ADD THIS: Model information field
54	information = models.TextField(blank=True, null=True)	1✔
55
56	category = models.CharField(	1✔
57	max_length=50,
58	choices=[
59	("research", "Research"),
60	("production", "Production"),
61	("testing", "Testing"),
62	],
63	default="research",
64	)
65
66	status = models.CharField(	1✔
67	max_length=20,
68	choices=[("PENDING", "Pending"), ("PASS", "Pass"), ("FAIL", "Fail")],
69	default="PENDING",
70	)
71	is_active = models.BooleanField(default=False)	1✔
72	log = models.TextField(blank=True, null=True)	1✔
73	created_at = models.DateTimeField(auto_now_add=True)	1✔
74
75	# Soft delete fields
76	is_deleted = models.BooleanField(default=False)	1✔
77	deleted_at = models.DateTimeField(null=True, blank=True)	1✔
78	# Store version number permanently
79	version_number = models.IntegerField(default=1)	1✔
80
81	class Meta:	1✔
82	ordering = ["-created_at"]	1✔
83
84	def save(self, args, *kwargs):	1✔
85	# Auto-generate version number for new versions only
86	if not self.pk: # Only for new objects	1✔
87	from django.db.models import Max	1✔
88
89	# Get the highest version number for this upload
90	last_version = ModelVersion.objects.filter(upload=self.upload).aggregate(	1✔
91	Max("version_number")
92	)["version_number__max"]
93	self.version_number = (last_version or 0) + 1	1✔
94	super().save(args, *kwargs)	1✔
95
96	def get_version_number(self):	1✔
97	"""Keep this method for backwards compatibility"""
NEW 98	return self.version_number	×
99
100	def __str__(self):	1✔
101	try:	1✔
102	version_str = f"v{self.version_number}"	1✔
103	upload_name = getattr(self.upload, "name", "Unknown Upload")	1✔
104	return f"{upload_name} - {version_str} ({self.status})"	1✔
NEW 105	except Exception:	×
NEW 106	return f"ModelVersion (unsaved) - {self.status}"	×

gcivil-nyu-org / team4-wed-fall25 / 55

Source File Press 'n' to go to next uncovered line, 'b' for previous

Source File
Press 'n' to go to next uncovered line, 'b' for previous