From 24e7156991d244f9fa10a558c009f0e9b6eceb52 Mon Sep 17 00:00:00 2001
From: Matt Krump <1036969+mkrump@users.noreply.github.com>
Date: Tue, 28 Jul 2020 16:19:16 -0600
Subject: [PATCH] Turn on check_untyped_defs for aqt.editor

* Turn on check_untyped_defs for aqt.browser
* Add type hints
---
 qt/aqt/editor.py | 4 +++-
 qt/mypy.ini      | 2 ++
 2 files changed, 5 insertions(+), 1 deletion(-)

diff --git a/qt/aqt/editor.py b/qt/aqt/editor.py
index e07060616..a367616b0 100644
--- a/qt/aqt/editor.py
+++ b/qt/aqt/editor.py
@@ -13,6 +13,7 @@ import urllib.request
 import warnings
 from typing import Callable, List, Optional, Tuple
 
+import bs4
 import requests
 from bs4 import BeautifulSoup
 
@@ -859,7 +860,7 @@ to a cloze type first, via 'Notes>Change Note Type'"""
 
     removeTags = ["script", "iframe", "object", "style"]
 
-    def _pastePreFilter(self, html, internal):
+    def _pastePreFilter(self, html: str, internal: bool) -> str:
         # https://anki.tenderapp.com/discussions/ankidesktop/39543-anki-is-replacing-the-character-by-when-i-exit-the-html-edit-mode-ctrlshiftx
         if html.find(">") < 0:
             return html
@@ -868,6 +869,7 @@ to a cloze type first, via 'Notes>Change Note Type'"""
             warnings.simplefilter("ignore", UserWarning)
             doc = BeautifulSoup(html, "html.parser")
 
+        tag: bs4.element.Tag
         if not internal:
             for tag in self.removeTags:
                 for node in doc(tag):
diff --git a/qt/mypy.ini b/qt/mypy.ini
index 4d4b6a50a..8489500df 100644
--- a/qt/mypy.ini
+++ b/qt/mypy.ini
@@ -82,3 +82,5 @@ check_untyped_defs=true
 check_untyped_defs=true
 [mypy-aqt.browser]
 check_untyped_defs=true
+[mypy-aqt.editor]
+check_untyped_defs=true