handle_play_like_call decorator to use functools.wraps

this way, decorated methods - typically Scene.play - have a decent docstring
so first-time users can use help()
This commit is contained in:
Thierry Parmentelat 2020-08-16 11:59:07 +02:00
parent 8f52c7bda8
commit e7251bcc2f

View file

@ -3,6 +3,7 @@ import random
import platform import platform
import itertools as it import itertools as it
import logging import logging
from functools import wraps
from tqdm import tqdm as ProgressDisplay from tqdm import tqdm as ProgressDisplay
import numpy as np import numpy as np
@ -372,6 +373,7 @@ class Scene(object):
return animations return animations
def handle_play_like_call(func): def handle_play_like_call(func):
@wraps(func)
def wrapper(self, *args, **kwargs): def wrapper(self, *args, **kwargs):
self.update_skipping_status() self.update_skipping_status()
should_write = not self.skip_animations should_write = not self.skip_animations