Rooba
/
attrs
mirror of https://github.com/python-attrs/attrs.git
1
1
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Simplify attributes decorator

This commit is contained in:
Hynek Schlawack 2015-01-30 09:50:33 +01:00
parent 5753281838
commit 564fade925
1 changed files with 12 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
attr/_make.py
View File

@ -122,13 +122,6 @@ def attributes(maybe_cl=None, add_repr=True, add_cmp=True, add_hash=True,
C(_private=42) C(_private=42)
:type add_init: bool :type add_init: bool
""" """
# attrs_or class type depends on the usage of the decorator. It's a class
# if it's used as `@attributes` but ``None`` (or a value passed) if used
# as `@attributes()`.
if isinstance(maybe_cl, type):
_transform_attrs(maybe_cl)
return _add_init(_add_hash(_add_cmp(_add_repr(maybe_cl))))
else:
def wrap(cl): def wrap(cl):
_transform_attrs(cl) _transform_attrs(cl)
if add_repr is True: if add_repr is True:
@ -140,7 +133,12 @@ def attributes(maybe_cl=None, add_repr=True, add_cmp=True, add_hash=True,
if add_init is True: if add_init is True:
cl = _add_init(cl) cl = _add_init(cl)
return cl return cl
# attrs_or class type depends on the usage of the decorator. It's a class
# if it's used as `@attributes` but ``None`` (or a value passed) if used
# as `@attributes()`.
if isinstance(maybe_cl, type):
return wrap(maybe_cl)
else:
return wrap return wrap