mirror of
https://github.com/apache/superset.git
synced 2026-04-19 16:14:52 +00:00
[mypy] Enforcing typing for superset.examples (#9469)
Co-authored-by: John Bodley <john.bodley@airbnb.com>
This commit is contained in:
@@ -37,7 +37,18 @@ from email.mime.text import MIMEText
|
||||
from email.utils import formatdate
|
||||
from enum import Enum
|
||||
from time import struct_time
|
||||
from typing import Any, Dict, Iterator, List, NamedTuple, Optional, Set, Tuple, Union
|
||||
from typing import (
|
||||
Any,
|
||||
Dict,
|
||||
Iterator,
|
||||
List,
|
||||
NamedTuple,
|
||||
Optional,
|
||||
Set,
|
||||
Tuple,
|
||||
TYPE_CHECKING,
|
||||
Union,
|
||||
)
|
||||
from urllib.parse import unquote_plus
|
||||
|
||||
import bleach
|
||||
@@ -72,6 +83,9 @@ try:
|
||||
except ImportError:
|
||||
pass
|
||||
|
||||
if TYPE_CHECKING:
|
||||
from superset.models.core import Database
|
||||
|
||||
|
||||
logging.getLogger("MARKDOWN").setLevel(logging.INFO)
|
||||
logger = logging.getLogger(__name__)
|
||||
@@ -944,7 +958,7 @@ def get_or_create_db(database_name, sqlalchemy_uri, *args, **kwargs):
|
||||
return database
|
||||
|
||||
|
||||
def get_example_database():
|
||||
def get_example_database() -> "Database":
|
||||
from superset import conf
|
||||
|
||||
db_uri = conf.get("SQLALCHEMY_EXAMPLES_URI") or conf.get("SQLALCHEMY_DATABASE_URI")
|
||||
@@ -1057,11 +1071,15 @@ def get_since_until(
|
||||
else:
|
||||
rel, num, grain = time_range.split()
|
||||
if rel == "Last":
|
||||
since = relative_start - relativedelta(**{grain: int(num)}) # type: ignore
|
||||
since = relative_start - relativedelta( # type: ignore
|
||||
**{grain: int(num)} # type: ignore
|
||||
)
|
||||
until = relative_end
|
||||
else: # rel == 'Next'
|
||||
since = relative_start
|
||||
until = relative_end + relativedelta(**{grain: int(num)}) # type: ignore
|
||||
until = relative_end + relativedelta( # type: ignore
|
||||
**{grain: int(num)} # type: ignore
|
||||
)
|
||||
else:
|
||||
since = since or ""
|
||||
if since:
|
||||
|
||||
Reference in New Issue
Block a user