pub/aerich
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Checking cache action

Browse Source
This commit is contained in:
Waket Zheng 2024-06-06 17:21:46 +08:00
parent c42fdab74d
commit bceeb236c2
2 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
aerich/utils.py
View File

@ -4,7 +4,7 @@ import re
import sys
from pathlib import Path
from types import ModuleType
from typing import Dict, Optional
from typing import Dict, Optional, Union
from click import BadOptionUsage, ClickException, Context
from tortoise import BaseDBAsyncClient, Tortoise
@ -95,7 +95,7 @@ def is_default_function(string: str) -> Optional[re.Match]:
return re.match(r"^<function.+>$", str(string or ""))
def import_py_file(file: Path) -> ModuleType:
def import_py_file(file: Union[str, Path]) -> ModuleType:
module_name, file_ext = os.path.splitext(os.path.split(file)[-1])
spec = importlib.util.spec_from_file_location(module_name, file)
module = importlib.util.module_from_spec(spec) # type:ignore[arg-type]

2
tests/test_utils.py
View File

@ -1,6 +1,6 @@
from aerich.utils import import_py_file
def test_import_py_file():
def test_import_py_file() -> None:
m = import_py_file("aerich/utils.py")
assert getattr(m, "import_py_file")