Skip to content

Commit

Permalink
issue 449: feature(audit): timerange can be input as command args
Browse files Browse the repository at this point in the history
  • Loading branch information
hhlee445 committed Mar 31, 2023
1 parent c1868ad commit 0edcdf2
Showing 1 changed file with 20 additions and 4 deletions.
24 changes: 20 additions & 4 deletions tools/audit.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,7 @@
import argparse
import logging
import os
import sys
from io import StringIO
from pathlib import PurePath
from pprint import pprint
Expand All @@ -8,8 +11,6 @@
from dotenv import dotenv_values
from elasticsearch import RequestsHttpConnection, helpers

import logging

logging.getLogger("elasticsearch").setLevel(level=logging.WARNING)
logging.basicConfig(
# format="%(levelname)s: %(relativeCreated)7d %(name)s:%(filename)s:%(funcName)s:%(lineno)s - %(message)s",
Expand Down Expand Up @@ -55,11 +56,26 @@ def get_body() -> dict:
"aggs": {}
}

argparser = argparse.ArgumentParser(add_help=True)
argparser.add_argument(
"--start-datetime",
default="1970-01-01T00:00:00.000000",
help=f'ISO formatted datetime string. Must be compatible with Python Elasticsearch Client. Defaults to "%(default)s".'
)
argparser.add_argument(
"--end-datetime",
default="9999-01-01T00:00:00.000000",
help=f'ISO formatted datetime string. Must be compatible with Python Elasticsearch Client. Defaults to "%(default)s".'
)

logging.info(f'{sys.argv=}')
args = argparser.parse_args(sys.argv[1:])


def get_range(
datetime_fieldname="creation_timestamp",
start_dt_iso="1970-01-01T00:00:00.000000",
end_dt_iso="9999-01-01T00:00:00.000000"
start_dt_iso=args.start_datetime,
end_dt_iso=args.end_datetime
) -> dict:
return {
"range": {
Expand Down

0 comments on commit 0edcdf2

Please sign in to comment.