From c3e0cc869bcc4434363480dc04db0323fc6f5682 2021-03-09 21:17:30
From: Brett Smith <brettcsmith@brettcsmith.org>
Date: 2021-03-09 21:17:30
Subject: [PATCH] query: Improve help text in interactive shell.

---

diff --git a/conservancy_beancount/reports/query.py b/conservancy_beancount/reports/query.py
index 24eab377a8dc7286a67f078308ae23d44c7a98ac..20ebaf173cc0c619d504ba1edf30b18647f24ab8 100644
--- a/conservancy_beancount/reports/query.py
+++ b/conservancy_beancount/reports/query.py
@@ -9,6 +9,7 @@ import argparse
 import contextlib
 import datetime
 import enum
+import functools
 import itertools
 import logging
 import sys
@@ -122,6 +123,7 @@ class StrMeta(bc_query_env.AnyMeta):
 
 
 class AggregateSet(bc_query_compile.EvalAggregator):
+    """Filter argument values that aren't unique."""
     __intypes__ = [object]
 
     def __init__(self, operands: List[str]) -> None:
@@ -223,6 +225,7 @@ class BQLShell(bc_query_shell.BQLShell):
         self.last_line_parsed = line
         super().run_parser(line, default_close_date)
 
+    @functools.wraps(bc_query_shell.BQLShell.on_Select, ('__doc__',))
     def on_Select(self, statement: QueryStatement) -> None:
         output_format: str = self.vars['format']
         try: