forked from pythoninside/europython2018
-
Notifications
You must be signed in to change notification settings - Fork 0
/
09-meta_dataclasses.py
63 lines (42 loc) · 1.36 KB
/
09-meta_dataclasses.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
from dataclasses import dataclass
from math import pi
class TypeChecker:
required_type = object
def __init__(self, name=None):
self.name = f'_{name}'
def __get__(self, instance, owner=None):
return instance.__dict__[self.name]
def __set__(self, instance, value):
assert isinstance(value, self.required_type), \
f'Booooo! Expecting a {self.required_type.__name__}'
instance.__dict__[self.name] = value
def typed_dataclass(cls):
cls = dataclass(cls)
for var_name, var_type in cls.__annotations__.items():
class Checker(TypeChecker):
required_type = var_type
setattr(cls, var_name, Checker(var_name))
return cls
class TypeCheckMeta(type):
def __new__(meta, name, bases, dct):
cls = super().__new__(meta, name, bases, dct)
return typed_dataclass(cls)
class Base(metaclass=TypeCheckMeta):
__annotations__ = {}
class Point(Base):
x: int
y: int
def move_by(self, dx, dy):
self.x += dx
self.y += dy
def __str__(self):
return f'A Point at {self.x}, {self.y}'
class Circle(Base):
center: Point
radius: int
@property
def area(self):
return pi * self.radius ** 2
def __str__(self):
return f'A Circle at {self.center.x}, {self.center.y} and ' + \
f'radius {self.radius}'