58 lines
1.5 KiB
Python
Executable File
58 lines
1.5 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
#
|
|
# http://github.com/mitchweaver/metal-archives
|
|
#
|
|
# get the location of a given band from metal-archives
|
|
#
|
|
# Usage: ./location 'black sabbath'
|
|
#
|
|
|
|
from bs4 import BeautifulSoup
|
|
from bs4.element import Comment
|
|
import urllib.request
|
|
import sys
|
|
|
|
def tag_visible(element):
|
|
if element.parent.name in ['style', 'script', 'head', 'title', 'meta', '[document]']:
|
|
return False
|
|
if isinstance(element, Comment):
|
|
return False
|
|
return True
|
|
|
|
def text_from_html(body):
|
|
soup = BeautifulSoup(body, 'html.parser')
|
|
texts = soup.findAll(text=True)
|
|
visible_texts = filter(tag_visible, texts)
|
|
lines = []
|
|
for line in visible_texts:
|
|
lines.append(line)
|
|
return lines
|
|
|
|
def parse(url):
|
|
try:
|
|
html = urllib.request.urlopen(url.replace(' ', '_').lower()).read()
|
|
|
|
location = []
|
|
found_count = 0
|
|
for line in text_from_html(html):
|
|
if len(line.strip()) < 2: continue
|
|
if 'Country of origin:' in line:
|
|
location.append(line)
|
|
found_count = 1
|
|
elif found_count > 0:
|
|
location.append(line)
|
|
found_count += 1
|
|
if found_count == 4:
|
|
break
|
|
|
|
print(location[0] + ' ' + location[1])
|
|
print(location[2] + ' ' + location[3])
|
|
except:
|
|
print("Can't find it?")
|
|
|
|
def main():
|
|
parse(url = 'https://www.metal-archives.com/bands/' + sys.argv[1])
|
|
|
|
if __name__ == "__main__":
|
|
main()
|