chore: fmt

Author: somehowchris

src/riskmatrix/layouts/steps.py

@@ -41,7 +41,11 @@ def steps(context: 'Organization', request: 'IRequest') -> 'RenderData':
                 '#',
                 disabled=True
             ),
-            Step(_("Finish Assessment"), request.route_url('finish_assessment'), disabled=False),
+            Step(
+                _("Finish Assessment"),
+                request.route_url('finish_assessment'),
+                disabled=False
+            ),
         ]
     }
 

src/riskmatrix/mail/mailer.py

@@ -547,8 +547,12 @@ def create_or_update_template(
             name = f'[{self.stream}] {organization.name}: {template.name}'
         if len(name) > 100:
             name = name[:97] + '...'
-        html_content: str = markdown_to_html(template.email_content)
-        plain_content = markdown_to_plaintext(template.email_content)
+        html_content: str = markdown_to_html(  # noqa: F821
+            template.email_content
+        )
+        plain_content = markdown_to_plaintext(  # noqa: F821
+            template.email_content
+        )
 
         # replace logos with appropriate placeholders.
         html_content = html_content.replace(

src/riskmatrix/mail/types.py

@@ -64,4 +64,4 @@ class TemplateMailParams(
     RequiredTemplateMailParams,
     OptionalTemplateMailParams
 ):
-    pass
+    pass

src/riskmatrix/models/asset.py

@@ -51,7 +51,7 @@ class Asset(Base, SoftDeleteMixin, SerializerMixin):
     assessments: Mapped[list['RiskAssessment']] = relationship(
         back_populates='asset'
     )
-    
+
     organization: Mapped['Organization'] = relationship(
         back_populates='assets'
     )

src/riskmatrix/models/organization.py

@@ -33,7 +33,7 @@ class Organization(Base):
     risks: Mapped[list['Risk']] = relationship(
         back_populates='organization',
     )
-    risk_catalogs:  Mapped[list['RiskCatalog']] = relationship(
+    risk_catalogs:  Mapped[list[RiskCatalog]] = relationship(
         back_populates='organization',
     )
     users: Mapped[list['User']] = relationship(

src/riskmatrix/models/password_change_token.py

@@ -98,4 +98,4 @@ def expired(self) -> bool:
         expiring_time = self.time_requested + timedelta(hours=48)
         if datetime.utcnow() > expiring_time:
             return True
-        return False
+        return False

src/riskmatrix/models/risk.py

@@ -28,6 +28,7 @@
 
 from sqlalchemy_serializer import SerializerMixin
 
+
 class Risk(SoftDeleteMixin, Base, SerializerMixin):
 
     __tablename__ = 'risk'

src/riskmatrix/models/risk_assessment.py

@@ -18,7 +18,6 @@
 from riskmatrix.orm.meta import UUIDStr
 from riskmatrix.orm.meta import UUIDStrPK
 from sqlalchemy.types import JSON
-from dataclasses import dataclass
 from sqlalchemy_serializer import SerializerMixin
 
 from typing import Any, ClassVar
@@ -82,7 +81,7 @@ def __init__(
         self,
         asset:  Asset,
         risk:   Risk,
-        info:  'RiskAssessmentInfo',
+        info:  RiskAssessmentInfo,
         **meta: Any
     ):
         self.id = str(uuid4())

src/riskmatrix/models/risk_assessment_info.py

@@ -1,4 +1,3 @@
-from dataclasses import dataclass
 from sedate import utcnow
 from riskmatrix.orm.meta import Base
 import enum
@@ -8,16 +7,13 @@
 from sqlalchemy.orm import Mapped
 from riskmatrix.orm.meta import UUIDStr
 from riskmatrix.orm.meta import UUIDStrPK
-from sqlalchemy import UniqueConstraint
-from sqlalchemy import Column
+from sqlalchemy_serializer import SerializerMixin
 from datetime import datetime
 
 from typing import TYPE_CHECKING
 
 if TYPE_CHECKING:
     from riskmatrix.models import RiskAssessment
-    
-
 
 
 class RiskAssessmentState(enum.Enum):
@@ -31,8 +27,6 @@ def __str__(self) -> str:
         }
         return names[self]
 
-from sqlalchemy_serializer import SerializerMixin
-
 
 class RiskAssessmentInfo(Base, SerializerMixin):
 
@@ -43,7 +37,7 @@ class RiskAssessmentInfo(Base, SerializerMixin):
         ForeignKey('organization.id', ondelete='CASCADE'),
         index=True,
     )
-    
+
     name: Mapped[str] = mapped_column(nullable=True)
 
     state: Mapped[RiskAssessmentState] = mapped_column(

src/riskmatrix/models/risk_catalog.py

@@ -58,7 +58,9 @@ class RiskCatalog(Base, SoftDeleteMixin, SerializerMixin):
     modified: Mapped[datetime | None] = mapped_column(onupdate=utcnow)
 
     risks: Mapped[list['Risk']] = relationship(back_populates='catalog')
-    organization: Mapped['Organization'] = relationship(back_populates='risk_catalogs')
+    organization: Mapped['Organization'] = relationship(
+        back_populates='risk_catalogs'
+    )
 
     def __init__(
         self,