adding count
This commit is contained in:
parent
8f3da5ee34
commit
743edf045b
@ -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:
|
||||||
|
Loading…
Reference in New Issue
Block a user