decorator on class

MingMingWu · MingMingWu · commit 32241cb57b20 · 2017-11-19T00:32:14.000+08:00
diff --git a/.gitignore b/.gitignore
@@ -1 +1,2 @@
 /.idea
+*.pyc
diff --git a/decorator.py b/decorator.py
@@ -5,6 +5,7 @@
 
 import time
 import sys
+import functools
 # function decorator
 
 
@@ -63,7 +64,6 @@ class Foo(object):
     pass
 
 
-
 class Human(object):
     def __init__(self, func):
         self.name = None
@@ -84,6 +84,30 @@ def eat(self):
 def someone():
     pass
 
+
+def CatDecorator(cls):
+
+    @functools.wraps(cls)
+    class new_class:
+        def __init__(self, name):
+            cls.__init__(self, name)
+            self.sex = "male"
+
+        def set_sex(self, sex):
+            self.sex = sex
+
+        def get_set(self):
+            return self.sex
+
+    return new_class
+
+
+@CatDecorator
+class Cat(object):
+    def __init__(self, name):
+        self.name = name
+
+
 if __name__ == "__main__":
     # function decorator
     print "\n----------function decorator----------"
@@ -100,3 +124,7 @@ def someone():
     Tim.breath()
     Tim.eat()
     Bob = someone()
+
+    Jack = Cat("Jack")
+    Jack.set_sex("femal")
+    print Jack.get_set()
