one more fix to bookmedia tests

[wolnelektury.git] / src / wolnelektury / management / profile.py

diff --git a/src/wolnelektury/management/profile.py b/src/wolnelektury/management/profile.py
index 90e5ec5..6978ea8 100644 (file)
--- a/src/wolnelektury/management/profile.py
+++ b/src/wolnelektury/management/profile.py
@@ -1,17 +1,20 @@
-
+# -*- coding: utf-8 -*-
+# This file is part of Wolnelektury, licensed under GNU Affero GPLv3 or later.
+# Copyright © Fundacja Nowoczesna Polska. See NOTICE for more information.
+#
 import cProfile
 import functools
 import os
 
 _object = None
 
+
 def profile(meth):
     def _wrapper(self, *args, **kwargs):
-        object = self
-        setattr(object, "__%s" % meth.__name__, meth)
-        cProfile.runctx('object.__%s(object, *args, **kwargs)' % (meth.__name__, ), globals(), locals(),
-            "profile.%d" % os.getpid())
+        setattr(self, "__%s" % meth.__name__, meth)
+        cProfile.runctx(
+            'object.__%s(object, *args, **kwargs)' % (meth.__name__, ), globals(), locals(),
+            'profile.%d' % os.getpid())
 
     functools.update_wrapper(_wrapper, meth)
     return _wrapper
-