Monkeytype pylib/anki/db.py

This commit is contained in:
Alan Du 2020-02-26 20:15:49 -05:00
parent a96f1abb33
commit acf1b00cd2

View file

@ -5,7 +5,7 @@ import os
import time import time
from sqlite3 import Cursor from sqlite3 import Cursor
from sqlite3 import dbapi2 as sqlite from sqlite3 import dbapi2 as sqlite
from typing import Any, List from typing import Any, List, Type
DBError = sqlite.Error DBError = sqlite.Error
@ -110,5 +110,5 @@ class DB:
def _textFactory(self, data: bytes) -> str: def _textFactory(self, data: bytes) -> str:
return str(data, errors="ignore") return str(data, errors="ignore")
def cursor(self, factory=Cursor) -> Cursor: def cursor(self, factory: Type[Cursor] = Cursor) -> Cursor:
return self._db.cursor(factory) return self._db.cursor(factory)