mirror of
https://github.com/cisagov/manage.get.gov.git
synced 2025-06-30 08:13:27 +02:00
Add documentaiton
This commit is contained in:
parent
cdaf4afd2a
commit
9e57be5aba
3 changed files with 145 additions and 17 deletions
108
docs/developer/management_script_helpers.md
Normal file
108
docs/developer/management_script_helpers.md
Normal file
|
@ -0,0 +1,108 @@
|
|||
# Terminal Helper Functions
|
||||
`terminal_helper.py` contains utility functions to assist with common terminal and script operations.
|
||||
|
||||
## TerminalColors
|
||||
`TerminalColors` provides ANSI color codes as variables to style terminal output. Example usage:
|
||||
|
||||
print(f"{TerminalColors.OKGREEN}Success!{TerminalColors.ENDC}")
|
||||
|
||||
## ScriptDataHelper
|
||||
### bulk_update_fields
|
||||
|
||||
`bulk_update_fields` performs a memory-efficient bulk update on a Django model in batches using a Paginator.
|
||||
|
||||
Usage:
|
||||
bulk_update_fields(Domain, page.object_list, ["first_ready"])
|
||||
|
||||
## PopulateScriptTemplate
|
||||
|
||||
`PopulateScriptTemplate` is an abstract base class that provides a template for creating generic populate scripts. It handles logging and bulk updating for repetitive scripts that update a few fields.
|
||||
|
||||
**Disclaimer:** This template is intended as a shorthand for simple scripts. It is not recommended for complex operations. See `transfer_federal_agency.py` for a straightforward example of how to use this template.
|
||||
|
||||
To use `PopulateScriptTemplate`, create a new class that inherits from it and implement the `update_record` method. This method defines how each record should be updated.
|
||||
|
||||
The class provides the following optional configuration variables:
|
||||
- `prompt_title`: The header displayed by `prompt_for_execution` when the script starts (default: "Do you wish to proceed?")
|
||||
- `display_run_summary_items_as_str`: If True, runs `str(item)` on each item when printing the run summary for prettier output (default: False)
|
||||
- `run_summary_header`: The header for the script run summary printed after the script finishes (default: None)
|
||||
|
||||
The main method provided by `PopulateScriptTemplate` is `mass_update_records`. This method loops through each valid object (specified by `filter_conditions`) and updates the fields defined in `fields_to_update` using the `update_record` method.
|
||||
|
||||
Before updating, `mass_update_records` prompts the user to confirm the proposed changes. If the user does not proceed, the script will exit.
|
||||
|
||||
After processing the records, `mass_update_records` performs a bulk update on the specified fields using `ScriptDataHelper.bulk_update_fields` and logs a summary of the script run using `TerminalHelper.log_script_run_summary`.
|
||||
|
||||
The class also provides helper methods:
|
||||
- `get_class_name`: Returns a display-friendly class name for the terminal prompt
|
||||
- `get_failure_message`: Returns the message to display if a record fails to update
|
||||
- `should_skip_record`: Defines the condition for skipping a record (by default, no records are skipped)
|
||||
|
||||
To create a script using `PopulateScriptTemplate`:
|
||||
1. Create a new class that inherits from `PopulateScriptTemplate`
|
||||
2. Implement the `update_record` method to define how each record should be updated
|
||||
3. Optionally, override the configuration variables and helper methods as needed
|
||||
4. Call `mass_update_records` within `handle` and run the script
|
||||
|
||||
## TerminalHelper
|
||||
### log_script_run_summary
|
||||
|
||||
`log_script_run_summary` logs a summary of a script run, including counts of updated, skipped, and failed records.
|
||||
|
||||
### print_conditional
|
||||
|
||||
`print_conditional` conditionally logs a statement at a specified severity if a condition is met.
|
||||
|
||||
### prompt_for_execution
|
||||
|
||||
`prompt_for_execution` prompts the user to inspect a string and confirm if they wish to proceed. Returns True if proceeding, False if skipping, or exits the script.
|
||||
|
||||
### get_file_line_count
|
||||
|
||||
`get_file_line_count` returns the number of lines in a file.
|
||||
|
||||
### print_to_file_conditional
|
||||
|
||||
`print_to_file_conditional` conditionally writes content to a file if a condition is met.
|
||||
|
||||
Refer to the source code for full function signatures and additional details.
|
||||
|
||||
### query_yes_no
|
||||
|
||||
`query_yes_no` prompts the user with a yes/no question and returns True for "yes" or False for "no".
|
||||
|
||||
Usage:
|
||||
```python
|
||||
if query_yes_no("Do you want to proceed?"):
|
||||
print("Proceeding...")
|
||||
else:
|
||||
print("Aborting.")
|
||||
```
|
||||
|
||||
### query_yes_no_exit
|
||||
|
||||
`query_yes_no_exit` is similar to `query_yes_no` but includes an "exit" option to terminate the script.
|
||||
|
||||
Usage:
|
||||
if query_yes_no_exit("Continue, abort, or exit?"):
|
||||
print("Continuing...")
|
||||
else:
|
||||
print("Aborting.")
|
||||
# Script will exit if user selected "e" for exit
|
||||
|
||||
### array_as_string
|
||||
|
||||
`array_as_string` converts a list of strings into a single string with each element on a new line.
|
||||
|
||||
Usage:
|
||||
```python
|
||||
my_list = ["apple", "banana", "cherry"]
|
||||
print(array_as_string(my_list))
|
||||
```
|
||||
|
||||
Output:
|
||||
```
|
||||
apple
|
||||
banana
|
||||
cherry
|
||||
```
|
Loading…
Add table
Add a link
Reference in a new issue