Revisions on model

This commit is contained in:
Rachid Mrad 2024-01-23 18:15:56 -05:00
parent d1a80d3307
commit 3f8e5ce204
No known key found for this signature in database
GPG key ID: EF38E4CEC4A8F3CF
5 changed files with 16 additions and 20 deletions

View file

@ -82,12 +82,12 @@ def login_callback(request):
if user: if user:
login(request, user) login(request, user)
logger.info("Successfully logged in user %s" % user) logger.info("Successfully logged in user %s" % user)
# Double login bug? # Double login bug (1507)?
return redirect(request.session.get("next", "/")) return redirect(request.session.get("next", "/"))
else: else:
raise o_e.BannedUser() raise o_e.BannedUser()
except o_e.NoStateDefined as nsd_err: except o_e.NoStateDefined as nsd_err:
logger.debug(f"No State Defined: {nsd_err}") logger.warning(f"No State Defined: {nsd_err}")
return redirect(request.session.get("next", "/")) return redirect(request.session.get("next", "/"))
except Exception as err: except Exception as err:
return error_page(request, err) return error_page(request, err)

View file

@ -1240,14 +1240,14 @@ class DraftDomainAdmin(ListHeaderAdmin):
class VeryImportantPersonAdmin(ListHeaderAdmin): class VeryImportantPersonAdmin(ListHeaderAdmin):
list_display = ("email", "user", "truncated_notes", "created_at") list_display = ("email", "requestor", "truncated_notes", "created_at")
search_fields = ["email"] search_fields = ["email"]
search_help_text = "Search by email." search_help_text = "Search by email."
list_filter = [ list_filter = [
"user", "requestor",
] ]
readonly_fields = [ readonly_fields = [
"user", "requestor",
] ]
def truncated_notes(self, obj): def truncated_notes(self, obj):
@ -1258,7 +1258,7 @@ class VeryImportantPersonAdmin(ListHeaderAdmin):
def save_model(self, request, obj, form, change): def save_model(self, request, obj, form, change):
# Set the user field to the current admin user # Set the user field to the current admin user
obj.user = request.user if request.user.is_authenticated else None obj.requestor = request.requestor if request.requestor.is_authenticated else None
super().save_model(request, obj, form, change) super().save_model(request, obj, form, change)

View file

@ -1,4 +1,4 @@
# Generated by Django 4.2.7 on 2024-01-19 00:18 # Generated by Django 4.2.7 on 2024-01-23 21:02
from django.conf import settings from django.conf import settings
from django.db import migrations, models from django.db import migrations, models
@ -17,10 +17,10 @@ class Migration(migrations.Migration):
("id", models.BigAutoField(auto_created=True, primary_key=True, serialize=False, verbose_name="ID")), ("id", models.BigAutoField(auto_created=True, primary_key=True, serialize=False, verbose_name="ID")),
("created_at", models.DateTimeField(auto_now_add=True)), ("created_at", models.DateTimeField(auto_now_add=True)),
("updated_at", models.DateTimeField(auto_now=True)), ("updated_at", models.DateTimeField(auto_now=True)),
("email", models.EmailField(blank=True, db_index=True, help_text="Email", max_length=254, null=True)), ("email", models.EmailField(blank=True, db_index=True, help_text="Email", max_length=254)),
("notes", models.TextField(blank=True, help_text="Notes", null=True)), ("notes", models.TextField(blank=True, help_text="Notes")),
( (
"user", "requestor",
models.ForeignKey( models.ForeignKey(
blank=True, blank=True,
null=True, null=True,

View file

@ -8,13 +8,13 @@ class VeryImportantPerson(TimeStampedModel):
"""""" """"""
email = models.EmailField( email = models.EmailField(
null=True, null=False,
blank=True, blank=True,
help_text="Email", help_text="Email",
db_index=True, db_index=True,
) )
user = models.ForeignKey( requestor = models.ForeignKey(
"registrar.User", "registrar.User",
null=True, null=True,
blank=True, blank=True,
@ -23,14 +23,10 @@ class VeryImportantPerson(TimeStampedModel):
) )
notes = models.TextField( notes = models.TextField(
null=True, null=False,
blank=True, blank=True,
help_text="Notes", help_text="Notes",
) )
def __str__(self): def __str__(self):
try:
if self.email:
return self.email return self.email
except Exception:
return ""

View file

@ -1755,10 +1755,10 @@ class VeryImportantPersonAdminTestCase(TestCase):
# Create a request object # Create a request object
request = self.factory.post("/admin/yourapp/veryimportantperson/add/") request = self.factory.post("/admin/yourapp/veryimportantperson/add/")
request.user = self.superuser request.requestor = self.superuser
# Call the save_model method # Call the save_model method
admin_instance.save_model(request, vip_instance, None, None) admin_instance.save_model(request, vip_instance, None, None)
# Check that the user field is set to the request.user # Check that the user field is set to the request.user
self.assertEqual(vip_instance.user, self.superuser) self.assertEqual(vip_instance.requestor, self.superuser)