a: str = "aa"
b: int = 1

# 參數(shù)和返回標(biāo)注了類(lèi)型，那么接下來(lái)調(diào)用時(shí)就能進(jìn)行提示
def example(a: str) -> str:
 return f"Hello {a}"

pirnt(example("world"))

# # 一些簡(jiǎn)單的標(biāo)注，看起來(lái)起不到效果，但如果換個(gè)有含義的名字呢
User = str
Age = int
Answer = str

def say_hello(u: User) -> Answer:
 return f"Hello {u}"

print(say_hello("Shadow"))

import typing

# 自定義類(lèi)型注解
User = str
Age = int

# 定義有多種類(lèi)型注解的類(lèi)型
AnyStr = typing.TypeVar('AnyStr', str, bytes)
a_str: AnyStr = "a"
a_bytes: AnyStr = b"a"



# 通用類(lèi)型, 接收通用的類(lèi)型，盡量少的去使用
def example_1(a: typing.Any):
  print(a)


"""
typing 模塊是允許使用下標(biāo)來(lái)輔助標(biāo)記類(lèi)型
"""

# 列表, 下標(biāo)為列表的屬性
def example_2(a_list: typing.List[User]) -> typing.List[str]:
  pass


# 字典，下標(biāo)第一個(gè)為key，第二個(gè)為value
def example_3(a_dict: typing.Dict[User, Age]) -> typing.Dict[str, int]:
  pass


# 元祖，下標(biāo)為元祖的屬性
def example_4(a_tuple: typing.Tuple[User] = None) -> typing.Tuple[User]:
  pass


# Union, 在一些場(chǎng)景下我們某些參數(shù)或返回值是不確定，至少給定一個(gè)參數(shù)類(lèi)型
def example_5(a_b: typing.Union[str, int]) -> typing.Union[str, int]:
  pass


# Optional, 與Union 有點(diǎn)類(lèi)似，但默認(rèn)多帶一個(gè)None，至少給定一個(gè)參數(shù)類(lèi)型
# 如：Optional[str] 等價(jià)于 Union[str, None]
def example_6(a: str) -> typing.Optional[str]:
  pass


# Tuple, 返回值有多個(gè)的時(shí)候, 如需要返回str, int, bool, float
def example_7() -> typing.Tuple[str, int, bool, float]:
  pass


# class, 類(lèi)本身也是一種類(lèi)型
class Action:

  up: str = "up"
  down: str = "down"

# 指定需求一個(gè)action對(duì)象的參數(shù)
def example_8(action_obj: Action) -> Action:
  pass


# 這樣在一些枚舉參數(shù)的場(chǎng)景下，我們也可以使用類(lèi)作為我們枚舉參數(shù)的歸類(lèi)
def example_9(action_cls: Action) -> Action:
  pass

# 如果上面的枚舉參數(shù)你覺(jué)得并不能很好的實(shí)現(xiàn)，那么還是可以使用自定義類(lèi)型注解的方式去實(shí)現(xiàn)
Action = str
up: Action = "up"
down: Action = "down"

# 在python3.9 中對(duì)枚舉參數(shù)類(lèi)型有更好的支持
MODE = type.Literal['r', 'rb', 'w', 'wb']
def open_file(file: str, mode: MODE) -> str:
  pass

open_file('/some/path', 'r') # 正常
open_file('/other/path', 'typo') # 會(huì)提示該類(lèi)型不合法


# Type, 在一些多態(tài)類(lèi)的場(chǎng)景下標(biāo)注同一個(gè)類(lèi)型的不同的形態(tài)
class User: ...

class BasicUser(User): ...

class ProUser(User): ...

class TeamUser(User): ...

# 相當(dāng)于 typing.Union[User, BasicUser, ProUser, TeamUser]
def make_new_user(user_class: typing.Type[User]) -> User:
  return user_class()