Fix some class names

thinkst · Jan 24, 2024 · 42679e3 · 42679e3
1 parent ec8e732
commit 42679e3
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 8 deletions.
diff --git a/canarytokens/models.py b/canarytokens/models.py
@@ -2184,7 +2184,7 @@ class DownloadCMDRequest(TokenDownloadRequest):
     fmt: Literal[DownloadFmtTypes.CMD] = DownloadFmtTypes.CMD
 
 
-class DownloadCSSClonedWebTokenRequest(TokenDownloadRequest):
+class DownloadCSSClonedWebRequest(TokenDownloadRequest):
     fmt: Literal[DownloadFmtTypes.CSSCLONEDSITE] = DownloadFmtTypes.CSSCLONEDSITE
 
 
@@ -2207,7 +2207,7 @@ class DownloadSlackAPIRequest(TokenDownloadRequest):
         DownloadAzureIDCertRequest,
         DownloadCCRequest,
         DownloadCMDRequest,
-        DownloadCSSClonedWebTokenRequest,
+        DownloadCSSClonedWebRequest,
         DownloadIncidentListCSVRequest,
         DownloadIncidentListJsonRequest,
         DownloadKubeconfigRequest,
@@ -2322,7 +2322,7 @@ class DownloadCMDResponse(TokenDownloadResponse):
     auth: str
 
 
-class DownloadCSSClonedWebTokenResponse(TokenDownloadResponse):
+class DownloadCSSClonedWebResponse(TokenDownloadResponse):
     contenttype: Literal[
         DownloadContentTypes.TEXTPLAIN
     ] = DownloadContentTypes.TEXTPLAIN

diff --git a/frontend/app.py b/frontend/app.py
@@ -55,8 +55,8 @@
     ClonedWebTokenResponse,
     CSSClonedWebTokenRequest,
     CSSClonedWebTokenResponse,
-    DownloadCSSClonedWebTokenRequest,
-    DownloadCSSClonedWebTokenResponse,
+    DownloadCSSClonedWebRequest,
+    DownloadCSSClonedWebResponse,
     CMDTokenRequest,
     CMDTokenResponse,
     CustomBinaryTokenRequest,
@@ -603,10 +603,10 @@ def _(
 
 @create_download_response.register
 def _(
-    download_request_details: DownloadCSSClonedWebTokenRequest, canarydrop: Canarydrop
-) -> DownloadCSSClonedWebTokenResponse:
+    download_request_details: DownloadCSSClonedWebRequest, canarydrop: Canarydrop
+) -> DownloadCSSClonedWebResponse:
     """"""
-    return DownloadCSSClonedWebTokenResponse(
+    return DownloadCSSClonedWebResponse(
         token=download_request_details.token,
         auth=download_request_details.auth,
         content=canarydrop.get_cloned_site_css(frontend_settings.CLOUDFRONT_URL),