Files
django-q2/django_q/management/commands/qmemory.py
T
Kenny Heinonen 728a1b85d0 Add "qmemory" command (#553)
* Add qmemory command

* Update README, add test

* Redesign qmemory

* Add optional argument to show each worker's memory usage

* Add --workers argument to test_qmemory
2021-05-14 13:44:40 +02:00

32 lines
888 B
Python

from django.core.management.base import BaseCommand
from django.utils.translation import gettext as _
from django_q.monitor import memory
class Command(BaseCommand):
# Translators: help text for qmemory management command
help = _("Monitors Q Cluster memory usage")
def add_arguments(self, parser):
parser.add_argument(
"--run-once",
action="store_true",
dest="run_once",
default=False,
help="Run once and then stop.",
)
parser.add_argument(
"--workers",
action="store_true",
dest="workers",
default=False,
help="Show each worker's memory usage.",
)
def handle(self, *args, **options):
memory(
run_once=options.get("run_once", False),
workers=options.get("workers", False)
)