From a0d98d6eb6b789c78b3509f59fc12d2f5356ee79 Mon Sep 17 00:00:00 2001 From: Michael DM Dryden Date: Sat, 30 May 2020 00:30:20 -0400 Subject: [PATCH] stats: Set argument defaults in method signatures --- stats.py | 9 ++++----- 1 file changed, 4 insertions(+), 5 deletions(-) diff --git a/stats.py b/stats.py index bb53b0f..f945132 100644 --- a/stats.py +++ b/stats.py @@ -11,6 +11,8 @@ import seaborn as sns from matplotlib.figure import Figure from sqlalchemy.engine import Engine +from utils import escape_markdown + sns.set_context('paper') sns.set_style('whitegrid') sns.set_palette("Set2") @@ -92,7 +94,7 @@ class StatsRunner(object): with self.engine.connect() as con: con.execute(query, sql_dict) - def get_chat_counts(self, n: int = None, start: str = None, end: str = None) -> Tuple[str, None]: + def get_chat_counts(self, n: int = 20, start: str = None, end: str = None) -> Tuple[str, None]: """ Get top chat users :param n: Number of users to show @@ -104,11 +106,8 @@ class StatsRunner(object): sql_dict = {} query_conditions = [] - if n is not None: - if n <= 0: + if n <= 0: raise HelpException(f'n must be greater than 0, got: {n}') - else: - n = 20 if start: sql_dict['start_dt'] = pd.to_datetime(start)