Notifier le contenu de la tâche avant et après l'exécution de la tâche avec Fabric

Étant donné que j'utilisais Fabric et que j'avais beaucoup de tâches pour mes propres besoins, je voulais afficher les tâches à effectuer et les tâches exécutées avant et après l'exécution des tâches, j'ai donc créé une classe CustomTask.

report.py

# -*- coding: utf-8 -*-

import logging
import time

from fabric.tasks import Task
from fabric.operations import *
from fabric.colors import *

class ReportStatus(Task):
    """
Classe CustomTask qui vous indique que la tâche est sur le point d'être exécutée et qu'elle a été exécutée avant et après l'exécution de la tâche.

    @task(task_class=ReportStatus, task_name="Afficher le test avant et après l'exécution")
    def something_task():
        print green("Exécution des tâches")
    """
    def __init__(self, func, task_name, *args, **kwargs):
        super(ReportStatus, self).__init__(*args, **kwargs)
        self.func = func
        self.task_name = task_name

        if hasattr(func, '__doc__') and func.__doc__:
            self.__doc__ = func.__doc__
        else:
            self.__doc__ = task_name

        if hasattr(callable, '__module__'):
            self.__module__ = callable.__module__

    def run(self, *args, **kwargs):
        print cyan("[%s] [%s]Courir" % (self.task_name, self.func.__name__))
        logging.info("[%s][%s] executing" % (self.func.__name__, self.task_name))

        result = self.func(*args, **kwargs)

        print green("[%s] [%s]Terminé" % (self.task_name, self.func.__name__), bold=True)
        logging.info("[%s][%s] done" % (self.func.__name__, self.task_name))
        print "\n"

        return result

    def __call__(self, *args, **kwargs):
        return self.run(*args, **kwargs)

    def __getattr__(self, k):
        return getattr(self.func, k)

    def __details__(self):
        return get_task_details(self.func)

J'utilise également la journalisation pour cracher le journal, donc si vous le définissez correctement, il produira également des journaux

fabfile.py

# -*- coding: utf-8 -*-

import datetime
import logging
from fabric.colors import *
from fabric.api import *

from report import ReportStatus

log_file_name = "%s.log" % (datetime.datetime.today().strftime("%Y%m%d"))
logging.basicConfig(
    format='[%(asctime)s][%(levelname)s] %(message)s',
    filename=log_file_name, datefmt='%Y/%m/%d %I:%M:%S',
    level=logging.INFO
)

@task(task_class=ReportStatus, task_name="Afficher le test avant et après l'exécution")
def something_task():
    print green("Exécution des tâches")

Une fois exécuté, cela ressemble à ceci.

le journal est émis dans le format spécifié

[2017/01/19 01:20:18][INFO] [something_task][Afficher le test avant et après l'exécution] executing
[2017/01/19 01:20:18][INFO] [something_task][Afficher le test avant et après l'exécution] done