zulip/docs/subsystems/management-commands.md

68 lines
3.4 KiB
Markdown
Raw Normal View History

# Management commands
Zulip has a number of [Django management commands][django-docs] that
live under `{zerver,zilencer,analytics}/management/commands/`.
If you need some Python code to run with a Zulip context (access to
the database, etc.) in a script, it should probably go in a management
command. The key thing distinguishing these from production scripts
(`scripts/`) and development scripts (`tools/`) is that management
commands can access the database.
While Zulip takes advantage of built-in Django management commands for
things like managing Django migrations, we also have dozens that we've
written for a range of purposes:
- Cron jobs to do regular updates, e.g., `update_analytics_counts.py`,
`sync_ldap_user_data`, etc.
- Useful parts of provisioning or upgrading a Zulip development
environment or server, e.g., `makemessages`, `compilemessages`,
`populate_db`, `fill_memcached_caches`, etc.
- The actual scripts run by supervisord to run the persistent
processes in a Zulip server, e.g., `runtornado` and `process_queue`.
- For a sysadmin to verify a Zulip server's configuration during
installation, e.g., `checkconfig`, `send_test_email`.
- As the interface for doing those rare operations that don't have a
UI yet, e.g., `deactivate_realm`, `reactivate_realm`,
`change_user_email` (for the case where the user doesn't control the
old email address).
- For a sysadmin to easily interact with and script common possible
changes they might want to make to the database on a Zulip server.
E.g., `send_password_reset_email`, `export`, `purge_queue`.
## Writing management commands
It's generally pretty easy to template off an existing management
command to write a new one. Some good examples are
`change_user_email` and `deactivate_realm`. The Django documentation
is good, but we have a few pieces advice specific to the Zulip
project.
- Inherit from the `ZulipBaseCommand` class in
`zerver/lib/management.py`; this will add some helpful general
flags, as well as tools for adding and parsing `--realm` and
`--user` flags, so you don't need to write the tedious code of
looking those objects up. This is especially important for users,
since the library handles the issues around looking up users by
email well (if there's a unique user with that email, just modify it
without requiring the user to specify the realm as well, but if
there's a collision, throw a nice error).
- Avoid writing a lot of code in management commands; management
commands are annoying to unit test, and thus easier to maintain if
all the interesting logic is in a nice function that is unit tested
(and ideally, also used in Zulip's existing code). Look for code in
`zerver/lib/` that already does what you need. For most actions,
you can just call a `do_change_foo` type function from
`zerver/actions/` to do all the work; this is usually far
better than manipulating the database directly, since the library
functions used by the UI are maintained to correctly live-update the
UI if needed.
Management commands are essentially independent Python scripts with
access to the Zulip server's database and libraries; so you don't need
to do anything special like restart the server when iteratively
testing one, even if testing in a Zulip production environment where
the server doesn't normally restart whenever a file is edited.
[django-docs]: https://docs.djangoproject.com/en/5.0/howto/custom-management-commands/