adding count

This commit is contained in:
brent s 2020-07-14 17:30:47 -04:00
parent 8f3da5ee34
commit 743edf045b
Signed by: bts
GPG Key ID: 8C004C2F93481F6B

View File

@ -49,17 +49,21 @@ class InfoScraper(object):
def parseArgs(): def parseArgs():
args = argparse.ArgumentParser(description = 'Get quick information from a URL at a glance') args = argparse.ArgumentParser(description = 'Get quick information from a URL at a glance')
args.add_argument('-e', '--elem', args.add_argument('-e', '--elem',
dest = 'strip', dest = 'elem',
default = def_elem, default = def_elem,
help = ('The element(s) you want to scrape from the page. This is likely just going to be "{0}" (the default)').format(def_elem)) help = ('The element(s) you want to scrape from the page. This is likely just going to be "{0}" (the default)').format(def_elem))
args.add_argument('-s', '--strip', args.add_argument('-s', '--strip',
dest = 'strip', dest = 'strip',
action = 'store_true', action = 'store_true',
help = ('Whether to strip whitespace at the beginning/end of each element text')) help = ('If specified, strip whitespace at the beginning/end of each element text'))
args.add_argument('-d', '--delineate', args.add_argument('-d', '--delineate',
dest = 'delin', dest = 'delin',
action = 'store_true', action = 'store_true',
help = ('Whether to delineate each element instance')) help = ('If specified, delineate each element instance'))
args.add_argument('-c', '--count',
dest = 'count',
action = 'store_true',
help = ('If specified, provide a count of how many times -e/--elem was found'))
args.add_argument('url', args.add_argument('url',
metavar = 'URL', metavar = 'URL',
help = ('The URL to parse. It may need to be quoted or escaped depending on the URL and what shell you\'re using')) help = ('The URL to parse. It may need to be quoted or escaped depending on the URL and what shell you\'re using'))
@ -70,6 +74,8 @@ def main():
args = parseArgs().parse_args() args = parseArgs().parse_args()
i = InfoScraper(**vars(args)) i = InfoScraper(**vars(args))
rslts = i.find() rslts = i.find()
if args.count:
print('Element {0} was found {1} time(s) at {2}. Results follow:'.format(args.elem, len(rslts), args.url))
for i in rslts: for i in rslts:
t = i.text t = i.text
if args.strip: if args.strip: