xeals/holy-heck-i-really-like-stats
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

configure output file

Browse Source
This commit is contained in:
xeals 2023-03-28 00:24:21 +11:00
parent 89508608f0
commit 872abf2e6c
Signed by: xeals
GPG Key ID: A498C7AF27EC6B5C
1 changed files with 12 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
main.py
View File

@ -261,7 +261,9 @@ def fetch(replay: str, cache: bool = True) -> Replay:
def main(): def main():
parser = argparse.ArgumentParser( parser = argparse.ArgumentParser(
prog=APP, description="extracts stats from a Showdown replay" prog=APP,
description="extracts stats from a Showdown replay",
formatter_class=argparse.ArgumentDefaultsHelpFormatter,
) )
parser.add_argument("-v", "--verbose", action="count", help="add debugging info") parser.add_argument("-v", "--verbose", action="count", help="add debugging info")
parser.add_argument( parser.add_argument(
@ -270,6 +272,14 @@ def main():
action="store_true", action="store_true",
help="fetch replays instead of using cache", help="fetch replays instead of using cache",
) )
parser.add_argument(
"-o",
"--output",
action="store",
metavar="FILE",
default="data.db",
help="output data file",
)
parser.add_argument("replay", nargs="+", help="replay ID or URL") parser.add_argument("replay", nargs="+", help="replay ID or URL")
args = parser.parse_args(sys.argv[1:]) args = parser.parse_args(sys.argv[1:])
@ -279,7 +289,7 @@ def main():
LOG.setLevel(logging.DEBUG) LOG.setLevel(logging.DEBUG)
try: try:
db = sqlite3.connect("data.db") db = sqlite3.connect(args.output)
_init_db(db) _init_db(db)
for r in args.replay: for r in args.replay: