Skip to content

Commit defb26c

Browse files
committed
type: add type annotations to asset module
Add type annotations to save_image_to_asset_dir() function. Changes: - Add parameter types for mdrender and rnode using TYPE_CHECKING - Add return type Optional[List[str]]
1 parent 731735b commit defb26c

File tree

1 file changed

+9
-2
lines changed

1 file changed

+9
-2
lines changed

md2zhihu/asset/__init__.py

Lines changed: 9 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -2,21 +2,28 @@
22
import os
33
import re
44
import shutil
5+
from typing import TYPE_CHECKING
6+
from typing import List
7+
from typing import Optional
58

69
import urllib3
710
from k3handy import pjoin
811
from k3handy import to_bytes
912

13+
if TYPE_CHECKING:
14+
from ..renderer.md_render import MDRender
15+
from ..renderer.render_node import RenderNode
1016

11-
def save_image_to_asset_dir(mdrender, rnode):
17+
18+
def save_image_to_asset_dir(mdrender: "MDRender", rnode: "RenderNode") -> Optional[List[str]]:
1219
# {'alt': 'openacid',
1320
# 'src': 'https://...',
1421
# 'title': None,
1522
# 'type': 'image'},
1623

1724
n = rnode.node
1825

19-
src = n["src"]
26+
src: str = n["src"]
2027
if re.match(r"https?://", src):
2128
if not mdrender.conf.download:
2229
return None

0 commit comments

Comments
 (0)