From 930c12d33e1093f874a4beb4d6376621e464e8c0 Mon Sep 17 00:00:00 2001 From: Terry Truong Date: Sat, 20 Aug 2022 13:16:21 +1000 Subject: Use argparse in python scripts --- backend/tolData/enwiki/genDescData.py | 15 ++++++--------- 1 file changed, 6 insertions(+), 9 deletions(-) (limited to 'backend/tolData/enwiki/genDescData.py') diff --git a/backend/tolData/enwiki/genDescData.py b/backend/tolData/enwiki/genDescData.py index b0ca272..0085d70 100755 --- a/backend/tolData/enwiki/genDescData.py +++ b/backend/tolData/enwiki/genDescData.py @@ -5,15 +5,12 @@ import bz2 import html, mwxml, mwparserfromhell import sqlite3 -usageInfo = f""" -Usage: {sys.argv[0]} - -Reads through the wiki dump, and attempts to -parse short-descriptions, and add them to a database. -""" -if len(sys.argv) > 1: - print(usageInfo, file=sys.stderr) - sys.exit(1) +import argparse +parser = argparse.ArgumentParser(description=""" +Reads through the wiki dump, and attempts to parse short-descriptions, +and add them to a database +""", formatter_class=argparse.RawDescriptionHelpFormatter) +parser.parse_args() dumpFile = "enwiki-20220501-pages-articles-multistream.xml.bz2" # Had about 22e6 pages enwikiDb = "descData.db" -- cgit v1.2.3