Fix lint errors

This commit is contained in:
Erin 2024-03-19 15:19:44 -07:00
parent 166ccbf5ea
commit 601ded68ac
No known key found for this signature in database
GPG key ID: 1CAD275313C62460
3 changed files with 9 additions and 7 deletions

View file

@ -15,8 +15,7 @@ from django.contrib.contenttypes.models import ContentType
from django.urls import reverse
from dateutil.relativedelta import relativedelta # type: ignore
from epplibwrapper.errors import ErrorCode, RegistryError
from registrar.models import Contact, Domain, DomainRequest, DraftDomain, User, Website, FederalAgency
from registrar.utility import csv_export
from registrar.models import Contact, Domain, DomainRequest, DraftDomain, User, Website
from registrar.utility.errors import FSMApplicationError, FSMErrorCodes
from registrar.views.utility.mixins import OrderableFieldsMixin
from django.contrib.admin.views.main import ORDER_VAR
@ -1789,6 +1788,7 @@ class VerifiedByStaffAdmin(ListHeaderAdmin):
obj.requestor = request.user if request.user.is_authenticated else None
super().save_model(request, obj, form, change)
class FederalAgencyAdmin(ListHeaderAdmin):
list_display = ["agency"]
search_fields = ["agency"]

View file

@ -6,10 +6,12 @@ from django.db import migrations, models
from registrar.models import FederalAgency
from typing import Any
# For linting: RunPython expects a function reference.
def create_federal_agencies(apps, schema_editor) -> Any:
FederalAgency.create_federal_agencies(apps, schema_editor)
class Migration(migrations.Migration):
dependencies = [

View file

@ -18,7 +18,7 @@ class FederalAgency(TimeStampedModel):
)
def __str__(self) -> str:
return self.agency
return f"{self.agency}"
# TODO: Update parameters to put in
def create_federal_agencies(apps, schema_editor):