penguin-wwy/magic-in-ten-mins-py

十分钟魔法练习 Python 版

magic-in-ten-mins-py's Issues

1 里面可以用 dataclass 简化代码

py 的枚举类比较弱。所以需要用 Union 或者 | 来表达和类型。（利用继承的思路学到了）

from __future__ import annotations

import enum
from dataclasses import dataclass
from typing import Union


@dataclass
class Student:
    name: str
    id: int


@dataclass
class Teacher:
    name: str
    office: str


SchoolPerson = Union[Student, Teacher]


class Bool(enum.Enum):
    TRUE = enum.auto()
    FALSE = enum.auto()


def test_bool():
    b = Bool.FALSE
    match b:
        case Bool.TRUE:
            ...
        case Bool.FALSE:
            print("oh my?")


test_bool()


@dataclass
class O:
    ...


@dataclass
class S:
    x: Nat

    def __repr__(self) -> str:
        def count(x: Nat):
            match x:
                case S(x):
                    return 1 + count(x)
                case O():
                    return 0

        return str(count(self))


Nat = S | O


print(S(S(S(S(O())))))